checking on RPC status and sending it thru

1 files changed, 5 insertions, 0 deletions

diff --git a/rpc/rpc.py b/rpc/rpc.py
index a00fd2c..4b634a9 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -6,6 +6,7 @@ import time
 from io import BytesIO
 import base64
 from PIL import Image
+import atexit
 
 load_dotenv(find_dotenv())
 
@@ -33,11 +34,15 @@ class CortexRPC(object):
     val = None
     try:
       val = self.relay.connected('rpc')
+      atexit.register(lambda: self.relay.disconnecting())
       print('RPC connected...'.format(val))
     except zerorpc.TimeoutExpired:
       print('RPC timeout :(')
     return val
 
+  def ping(self):
+    return 'pong'
+
   def send_command(self, cmd, payload):
     print('your command, {}: {}'.format(cmd, payload))
     if self._cmd_fn is not None: