add first service registering decorator

tmtccmd/tc/__init__.py

@@ -21,3 +21,4 @@
 )
 
 from .handler import FeedWrapper, TcHandlerBase, SendCbParams
+from .decorator import service_provider, route_to_registered_service_handlers

tmtccmd/tc/decorator.py

@@ -0,0 +1,36 @@
+import functools
+
+from tmtccmd.tc.procedure import DefaultProcedureInfo
+from tmtccmd.tc.queue import DefaultPusQueueHelper
+
+SERVICE_HANDLER_DICT = dict()
+
+
+def service_provider(service: str):
+    """Decorator. TODO: Documentation"""
+    global SERVICE_HANDLER_DICT
+
+    def actual_service_decorator(handler):
+        global SERVICE_HANDLER_DICT
+
+        @functools.wraps(handler)
+        def service_handler_wrapper(
+            info: DefaultProcedureInfo,
+            queue_helper: DefaultPusQueueHelper,
+            op_code: str,
+        ):
+            return handler(info, queue_helper, op_code)
+
+        SERVICE_HANDLER_DICT.update({service: service_handler_wrapper})
+        return service_handler_wrapper
+
+    return actual_service_decorator
+
+
+def route_to_registered_service_handlers(
+    info: DefaultProcedureInfo, queue_helper: DefaultPusQueueHelper, op_code: str
+) -> bool:
+    if info.service in SERVICE_HANDLER_DICT:
+        SERVICE_HANDLER_DICT[info.service](info, queue_helper, op_code)
+        return True
+    return False

tmtccmd/tc/pus_5_event.py

@@ -4,7 +4,7 @@
 from spacepackets.ecss.conf import get_default_tc_apid
 from spacepackets.ecss.pus_5_event import Subservices
 
-from tmtccmd.tc.queue import QueueHelperBase
+from tmtccmd.tc.queue import DefaultPusQueueHelper
 
 
 def pack_enable_event_reporting_command(ssc: int, apid: int = -1):
@@ -29,7 +29,7 @@ def pack_disable_event_reporting_command(ssc: int, apid: int = -1):
     )
 
 
-def pack_generic_service_5_test_into(q: QueueHelperBase):
+def pack_generic_service_5_test_into(q: DefaultPusQueueHelper):
     q.add_log_cmd("Testing Service 5")
     # invalid subservice
     q.add_log_cmd("Testing Service 5: Invalid subservice")