import enum from typing import Dict, Tuple from config.definitions import CustomServiceList from config.object_ids import PL_SUBSYSTEM_ID from spacepackets.ecss import PusTelecommand from tmtc.common import pack_mode_cmd_with_info from tmtccmd.config import TmtcDefinitionWrapper, OpCodeEntry from tmtccmd.config.tmtc import tmtc_definitions_provider from tmtccmd.tc import service_provider from tmtccmd.tc.decorator import ServiceProviderParams from tmtccmd.tc.pus_200_fsfw_modes import Subservices as ModeSubservices class OpCodes(str, enum.Enum): OFF = "off" REPORT_ALL_MODES = "report_modes" class PayloadModes(enum.IntEnum): OFF = 0 class Info(str, enum.Enum): OFF = "Off Command" REPORT_ALL_MODES = "Report all modes" HANDLER_LIST: Dict[str, Tuple[int, str]] = { OpCodes.OFF: (PayloadModes.OFF, Info.OFF), } @service_provider(CustomServiceList.PL_SS) def build_acs_subsystem_cmd(p: ServiceProviderParams): op_code = p.op_code q = p.queue_helper info_prefix = "ACS Subsystem" if op_code in OpCodes.REPORT_ALL_MODES: q.add_log_cmd(f"{info_prefix}: {Info.REPORT_ALL_MODES}") q.add_pus_tc( PusTelecommand( service=200, subservice=ModeSubservices.TC_MODE_ANNOUNCE_RECURSIVE, app_data=PL_SUBSYSTEM_ID, ) ) mode_info_tup = HANDLER_LIST.get(op_code) if mode_info_tup is None: return pack_mode_cmd_with_info( object_id=PL_SUBSYSTEM_ID, info=f"{info_prefix}: {mode_info_tup[1]}", submode=0, mode=mode_info_tup[0], q=q, ) @tmtc_definitions_provider def add_payload_subsystem_cmds(defs: TmtcDefinitionWrapper): oce = OpCodeEntry() oce.add(OpCodes.OFF, Info.OFF) oce.add(OpCodes.REPORT_ALL_MODES, Info.REPORT_ALL_MODES) defs.add_service(CustomServiceList.PL_SS, "Payload Subsystem", oce)