cmd fn

1 files changed, 11 insertions, 2 deletions

diff --git a/rpc/rpc.py b/rpc/rpc.py
index e8a3f9a..1d928c7 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -10,10 +10,11 @@ from PIL import Image
 load_dotenv(find_dotenv())
 
 class CortexRPC(object):
-  def __init__(self, get_fn, set_fn, ready_fn):
+  def __init__(self, get_fn, set_fn, ready_fn, cmd_fn):
     super().__init__() 
     self._get_fn = get_fn
     self._set_fn = set_fn
+    self._cmd_fn = cmd_fn
     self.working = False
     self.connect(ready_fn)
 
@@ -66,12 +67,20 @@ class CortexRPC(object):
     return 'ok'
   """
 
+  def send_command(self, cmd, payload):
+    print('your command, {}: {}'.format(cmd, payload))
+    if self._cmd_fn is not None:
+      return self._cmd_fn(key, value)
+    else:
+      print('No command listener provided to CortexRPC')
+    return 'ok'
+
   def set_param(self, key, value):
     print('aloha. {}: {}'.format(key, value))
     if self._set_fn is not None:
       self._set_fn(key, value)
     else:
-      print('No listener provided to CortexRPC')
+      print('No set param listener provided to CortexRPC')
     return 'ok'
 
   def get_params(self):