that was annoying

eive_tmtc/pus_tc/system/controllers.py

@@ -0,0 +1,94 @@
+from typing import Union
+
+from tmtccmd.tc import DefaultPusQueueHelper
+from tmtccmd.tc.pus_200_fsfw_modes import Modes
+from tmtccmd.util import ObjectIdU32, ObjectIdBase
+
+from eive_tmtc.tmtc.common import pack_mode_cmd_with_info
+import eive_tmtc.config.object_ids as obj_ids
+
+from eive_tmtc.pus_tc.prompt_parameters import (
+    prompt_parameters_cli,
+    prompt_parameters_gui,
+)
+
+
+class OpCodes:
+    THERMAL_CONTROLLER = [obj_ids.THERMAL_CONTROLLER_ID.hex(), "ctrl-th"]
+    CORE_CONTROLLER = [obj_ids.CORE_CONTROLLER_ID.hex(), "ctrl-core"]
+
+
+class Info:
+    THERMAL_CONTROLLER = "Thermal controller"
+    CORE_CONTROLLER = "ACS controller"
+
+
+def pack_cmd_ctrl_to_prompted_mode(
+    q: DefaultPusQueueHelper, object_id: ObjectIdU32, gui: bool
+):
+    param_list = [
+        {"name": "Mode", "defaultValue": "2"},
+        {"name": "Submode", "defaultValue": "0"},
+    ]
+    if gui:
+        parameters = prompt_parameters_gui(param_list)
+    else:
+        parameters = prompt_parameters_cli(param_list)
+    mode = int(parameters["Mode"])
+    if mode < 0 or mode > 2:
+        print("Invalid Mode, defaulting to OFF")
+        mode = 0
+    submode = int(parameters["Submode"])
+    pack_mode_cmd_with_info(
+        object_id=object_id.as_bytes,
+        mode=mode,
+        submode=submode,
+        q=q,
+        info=f"Commanding {object_id} to {mode}, {submode}",
+    )
+
+
+def pack_cmd_ctrl_to_off(
+    q: DefaultPusQueueHelper, object_id: Union[ObjectIdBase, ObjectIdU32]
+):
+    pack_mode_cmd_with_info(
+        object_id=object_id.as_bytes,
+        mode=Modes.OFF,
+        submode=0,
+        q=q,
+        info=f"Commanding {object_id} OFF",
+    )
+
+
+def pack_cmd_ctrl_to_on(q: DefaultPusQueueHelper, object_id: ObjectIdU32):
+    pack_mode_cmd_with_info(
+        object_id=object_id.as_bytes,
+        mode=Modes.ON,
+        submode=0,
+        q=q,
+        info=f"Commanding {object_id} ON",
+    )
+
+
+def pack_cmd_ctrl_to_nml(
+    q: DefaultPusQueueHelper, object_id: Union[ObjectIdBase, ObjectIdU32]
+):
+    pack_mode_cmd_with_info(
+        object_id=object_id.as_bytes,
+        mode=Modes.NORMAL,
+        submode=0,
+        q=q,
+        info=f"Commanding {object_id} NORMAL",
+    )
+
+
+def get_object_from_op_code(op_code: str):
+    try:
+        return bytes.fromhex(op_code)
+    except:
+        pass
+
+    if op_code in OpCodes.THERMAL_CONTROLLER:
+        return obj_ids.THERMAL_CONTROLLER_ID
+    if op_code in OpCodes.CORE_CONTROLLER:
+        return obj_ids.CORE_CONTROLLER_ID