function typing for some of the grpc code

daemon/core/api/grpc/events.py

@@ -1,5 +1,6 @@
 import logging
 from queue import Empty, Queue
+from typing import List
 
 from core.api.grpc import core_pb2
 from core.api.grpc.grpcutils import convert_value
@@ -11,9 +12,10 @@ from core.emulator.data import (
     LinkData,
     NodeData,
 )
+from core.emulator.session import Session
 
 
-def handle_node_event(event):
+def handle_node_event(event: NodeData) -> core_pb2.NodeEvent:
     """
     Handle node event when there is a node event
 
@@ -34,7 +36,7 @@ def handle_node_event(event):
     return core_pb2.NodeEvent(node=node_proto, source=event.source)
 
 
-def handle_link_event(event):
+def handle_link_event(event: LinkData) -> core_pb2.LinkEvent:
     """
     Handle link event when there is a link event
 
@@ -90,7 +92,7 @@ def handle_link_event(event):
     return core_pb2.LinkEvent(message_type=event.message_type, link=link)
 
 
-def handle_session_event(event):
+def handle_session_event(event: EventData) -> core_pb2.SessionEvent:
     """
     Handle session event when there is a session event
 
@@ -110,7 +112,7 @@ def handle_session_event(event):
     )
 
 
-def handle_config_event(event):
+def handle_config_event(event: ConfigData) -> core_pb2.ConfigEvent:
     """
     Handle configuration event when there is configuration event
 
@@ -135,7 +137,7 @@ def handle_config_event(event):
     )
 
 
-def handle_exception_event(event):
+def handle_exception_event(event: ExceptionData) -> core_pb2.ExceptionEvent:
     """
     Handle exception event when there is exception event
 
@@ -153,7 +155,7 @@ def handle_exception_event(event):
     )
 
 
-def handle_file_event(event):
+def handle_file_event(event: FileData) -> core_pb2.FileEvent:
     """
     Handle file event
 
@@ -179,7 +181,7 @@ class EventStreamer:
     Processes session events to generate grpc events.
     """
 
-    def __init__(self, session, event_types):
+    def __init__(self, session: Session, event_types: List[core_pb2.EventType]) -> None:
         """
         Create a EventStreamer instance.
 
@@ -191,7 +193,7 @@ class EventStreamer:
         self.queue = Queue()
         self.add_handlers()
 
-    def add_handlers(self):
+    def add_handlers(self) -> None:
         """
         Add a session event handler for desired event types.
 
@@ -210,7 +212,7 @@ class EventStreamer:
         if core_pb2.EventType.SESSION in self.event_types:
             self.session.event_handlers.append(self.queue.put)
 
-    def process(self):
+    def process(self) -> core_pb2.Event:
         """
         Process the next event in the queue.
 
@@ -239,7 +241,7 @@ class EventStreamer:
             event = None
         return event
 
-    def remove_handlers(self):
+    def remove_handlers(self) -> None:
         """
         Remove session event handlers for events being watched.