71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
from eive_tmtc.config.definitions import CustomServiceList
|
|
from eive_tmtc.tmtc.obj_prompt import prompt_object
|
|
from spacepackets.ecss import PusTelecommand
|
|
from tmtccmd.config.tmtc import (
|
|
tmtc_definitions_provider,
|
|
TmtcDefinitionWrapper,
|
|
OpCodeEntry,
|
|
)
|
|
from tmtccmd.tmtc import service_provider
|
|
from tmtccmd.pus.s201_fsfw_health import Subservice, FsfwHealth
|
|
from tmtccmd.tmtc.decorator import ServiceProviderParams
|
|
|
|
|
|
class OpCode:
|
|
SET_HEALTH = "set_health"
|
|
ANNOUNCE_HEALTH_ALL = "read_health_all"
|
|
ANNOUNCE_HEALTH = "read_health"
|
|
|
|
|
|
class Info:
|
|
SET_HEALTH = "Set health of specific object"
|
|
ANNOUNCE_HEALTH_ALL = "Read all health states"
|
|
ANNOUNCE_HEALTH = "Read health state of one object"
|
|
|
|
|
|
def prompt_health() -> FsfwHealth:
|
|
for item in FsfwHealth:
|
|
print(f"{item}: {item.name}")
|
|
health_idx = int(input("Please enter health by index: "))
|
|
return FsfwHealth(health_idx)
|
|
|
|
|
|
@service_provider(CustomServiceList.HEALTH)
|
|
def pack_test_command(p: ServiceProviderParams):
|
|
o = p.op_code
|
|
q = p.queue_helper
|
|
if o == OpCode.SET_HEALTH:
|
|
app_data = bytearray(prompt_object())
|
|
health = prompt_health()
|
|
app_data.append(health)
|
|
q.add_log_cmd(Info.SET_HEALTH)
|
|
q.add_pus_tc(
|
|
PusTelecommand(
|
|
service=201, subservice=Subservice.TC_SET_HEALTH, app_data=app_data
|
|
)
|
|
)
|
|
elif o == OpCode.ANNOUNCE_HEALTH:
|
|
app_data = bytearray(prompt_object())
|
|
q.add_log_cmd(Info.ANNOUNCE_HEALTH)
|
|
q.add_pus_tc(
|
|
PusTelecommand(
|
|
service=201, subservice=Subservice.TC_ANNOUNCE_HEALTH, app_data=app_data
|
|
)
|
|
)
|
|
elif o == OpCode.ANNOUNCE_HEALTH_ALL:
|
|
q.add_log_cmd(Info.ANNOUNCE_HEALTH_ALL)
|
|
q.add_pus_tc(
|
|
PusTelecommand(service=201, subservice=Subservice.TC_ANNOUNCE_HEALTH_ALL)
|
|
)
|
|
else:
|
|
raise ValueError(f"unknown op code {o} for service {CustomServiceList.HEALTH}")
|
|
|
|
|
|
@tmtc_definitions_provider
|
|
def add_health_cmd_defs(defs: TmtcDefinitionWrapper):
|
|
oce = OpCodeEntry()
|
|
oce.add(OpCode.ANNOUNCE_HEALTH_ALL, Info.ANNOUNCE_HEALTH_ALL)
|
|
oce.add(OpCode.ANNOUNCE_HEALTH, Info.ANNOUNCE_HEALTH)
|
|
oce.add(OpCode.SET_HEALTH, Info.SET_HEALTH)
|
|
defs.add_service(CustomServiceList.HEALTH, info="Health Service", op_code_entry=oce)
|