From 3d836c372d7eff1e2b507888f1ff652a30187c54 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sat, 26 May 2018 18:25:21 +0200
Subject: checking on RPC status and sending it thru

---
 rpc/rpc.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'rpc')

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:
-- 
cgit v1.2.3-70-g09d2