pre-commit

samples/apps/cap/py/autogencap/ActorConnector.py

@@ -9,6 +9,7 @@
 from .Config import xsub_url, xpub_url, router_url
 from typing import Any, Dict
 
+
 class ActorConnector:
     def __init__(self, context, topic):
         self._context = context
@@ -29,10 +30,10 @@ def _send_recv_router_msg(self):
         req_socket = self._context.socket(zmq.REQ)
         req_socket.connect(router_url)
         try:
-            Debug("ActorConnector","Broker Check Request Sent")
+            Debug("ActorConnector", "Broker Check Request Sent")
             req_socket.send_string("Request")
             _ = req_socket.recv_string()
-            Debug("ActorConnector","Broker Check Response Received")
+            Debug("ActorConnector", "Broker Check Response Received")
         finally:
             req_socket.close()
 
@@ -46,8 +47,8 @@ def _connect_pub_socket(self):
             evt: Dict[str, Any] = {}
             mon_evt = recv_monitor_message(monitor)
             evt.update(mon_evt)
-            if evt['event'] == zmq.EVENT_MONITOR_STOPPED or evt['event'] == zmq.EVENT_HANDSHAKE_SUCCEEDED:
-                Debug("ActorConnector","Handshake received (Or Monitor stopped)")
+            if evt["event"] == zmq.EVENT_MONITOR_STOPPED or evt["event"] == zmq.EVENT_HANDSHAKE_SUCCEEDED:
+                Debug("ActorConnector", "Handshake received (Or Monitor stopped)")
                 break
         monitor.close()
         self._send_recv_router_msg()

samples/apps/cap/py/autogencap/Broker.py

@@ -4,6 +4,7 @@
 from autogencap.DebugLog import Debug, Info, Warn
 from autogencap.Config import xsub_url, xpub_url, router_url
 
+
 class Broker:
     def __init__(self, context: zmq.Context = zmq.Context()):
         self._context: zmq.Context = context
@@ -37,17 +38,18 @@ def _init_sockets(self):
             if self._router:
                 self._router.close()
             return False
-                
+
     def start(self) -> bool:
-        Debug("BROKER", f"Trying to start broker.")
+        Debug("BROKER", "Trying to start broker.")
         self._run = True
         self._broker_thread: threading.Thread = threading.Thread(target=self.thread_fn)
         self._broker_thread.start()
         time.sleep(0.01)
         return True
 
     def stop(self):
-        if not self._run: return
+        if not self._run:
+            return
         # Error("BROKER_ERR", "fix cleanup self._context.term()")
         Debug("BROKER", "stopped")
         self._run = False
@@ -64,9 +66,9 @@ def thread_fn(self):
         try:
             if not self._init_sockets():
                 Debug("BROKER", "Receive thread not started since sockets were not initialized")
-                self._run  = False
+                self._run = False
                 return
-            
+
             # Poll sockets for events
             self._poller: zmq.Poller = zmq.Poller()
             self._poller.register(self._xpub, zmq.POLLIN)
@@ -81,27 +83,28 @@ def thread_fn(self):
                     message = self._xpub.recv_multipart()
                     Debug("BROKER", f"subscription message: {message[0]}")
                     self._xsub.send_multipart(message)
-                
+
                 if self._xsub in events:
                     message = self._xsub.recv_multipart()
                     Debug("BROKER", f"publishing message: {message}")
                     self._xpub.send_multipart(message)
-                    
+
                 if self._router in events:
                     message = self._router.recv_multipart()
                     Debug("BROKER", f"router message: {message}")
                     # Mirror it back for now to confirm connectivity
-                    # More interesting reserved point to point 
+                    # More interesting reserved point to point
                     # routing coming in the the future
                     self._router.send_multipart(message)
-                    
+
         except Exception as e:
             Debug("BROKER", f"thread encountered an error: {e}")
         finally:
             self._run = False
             Debug("BROKER", "thread ended")
         return
 
+
 # Run a standalone broker that all other Actors can connect to.
 # This can also run inproc with the other actors.
 def main():
@@ -133,5 +136,6 @@ def main():
             Info("BROKER", "KeyboardInterrupt.  Stopping the broker.")
             broker.stop()
 
+
 if __name__ == "__main__":
     main()

samples/apps/cap/py/autogencap/DirectorySvc.py

@@ -21,6 +21,7 @@
 # TODO (Future DirectorySv PR) use actor description, network_id, other properties to make directory
 # service more generic and powerful
 
+
 class DirectoryActor(Actor):
     def __init__(self, topic: str, name: str):
         super().__init__(topic, name)
@@ -86,6 +87,7 @@ def _actor_lookup_msg_handler(self, topic: str, msg_type: str, msg: bytes, sende
         serialized_msg = actor_lookup_resp.SerializeToString()
         sender_connection.send_bin_msg(ActorLookupResponse.__name__, serialized_msg)
 
+
 class DirectorySvc:
     def __init__(self, context: zmq.Context = zmq.Context()):
         self._context: zmq.Context = context
@@ -151,6 +153,7 @@ def lookup_actor_info_by_name(self, actor_name: str) -> ActorInfoCollection:
                 return actor_lookup_resp.actor
         return None
 
+
 # Run a standalone directory service
 def main():
     context: zmq.Context = zmq.Context()
@@ -192,5 +195,6 @@ def main():
     context.term()
     Info("main", "Done.")
 
+
 if __name__ == "__main__":
     main()

samples/apps/cap/py/demo/zmq_tests.py

@@ -6,6 +6,7 @@
 from zmq.utils.monitor import recv_monitor_message
 from autogencap.Config import xsub_url, xpub_url, router_url, dealer_url
 
+
 def zmq_sub_test():
     context = zmq.Context()
     sub_socket = context.socket(zmq.SUB)
@@ -29,18 +30,20 @@ def zmq_sub_test():
             print(f"No message received in {elapsed_time:.2f} seconds")
     sub_socket.close()
 
+
 def event_monitor(pub_socket: zmq.Socket) -> None:
     monitor = pub_socket.get_monitor_socket()
     while monitor.poll():
         evt: Dict[str, Any] = {}
         mon_evt = recv_monitor_message(monitor)
         evt.update(mon_evt)
         print(evt)
-        if evt['event'] == zmq.EVENT_MONITOR_STOPPED or evt['event'] == zmq.EVENT_HANDSHAKE_SUCCEEDED:
+        if evt["event"] == zmq.EVENT_MONITOR_STOPPED or evt["event"] == zmq.EVENT_HANDSHAKE_SUCCEEDED:
             break
     monitor.close()
 
-def zmq_pub_test():    
+
+def zmq_pub_test():
     context = zmq.Context()
     pub_socket = context.socket(zmq.PUB)
     pub_socket.setsockopt(zmq.XPUB_VERBOSE, 1)
@@ -52,6 +55,7 @@ def zmq_pub_test():
         pub_socket.send_string(str(i))
     pub_socket.close()
 
+
 def zmq_router_dealer_test():
     context = zmq.Context()
     router_socket = context.socket(zmq.ROUTER)
@@ -85,13 +89,14 @@ def zmq_router_dealer_test():
                 message = dealer_socket.recv_multipart()
                 print("BROKER", f"publishing message: {message[0]}")
                 router_socket.send_multipart(message)
-                
+
     except Exception as e:
         print("BROKER", f"thread encountered an error: {e}")
     finally:
         print("BROKER", "thread ended")
     return
 
+
 def zmq_req_test(context: zmq.Context = None):
     if context is None:
         context = zmq.Context()
@@ -106,6 +111,7 @@ def zmq_req_test(context: zmq.Context = None):
     finally:
         req_socket.close()
 
+
 def zmq_rep_test():
     context = zmq.Context()
     rep_socket = context.socket(zmq.REP)
@@ -120,19 +126,20 @@ def zmq_rep_test():
     finally:
         rep_socket.close()
 
+
 if __name__ == "__main__":
     if len(sys.argv) > 1:
-        if sys.argv[1] == 'pub':
+        if sys.argv[1] == "pub":
             zmq_pub_test()
-        elif sys.argv[1] == 'sub':
+        elif sys.argv[1] == "sub":
             zmq_sub_test()
-        elif sys.argv[1] == 'router':
+        elif sys.argv[1] == "router":
             zmq_router_dealer_test()
-        elif sys.argv[1] == 'req':
+        elif sys.argv[1] == "req":
             zmq_req_test()
-        elif sys.argv[1] == 'rep':
+        elif sys.argv[1] == "rep":
             zmq_rep_test()
         else:
             print("Invalid argument. Please use 'pub', 'sub' 'router', 'req', 'rep'")
     else:
-        print("Please provide an argument. Please use 'pub', 'sub' 'router', 'req', 'rep'")
+        print("Please provide an argument. Please use 'pub', 'sub' 'router', 'req', 'rep'")