eive-tmtc/pus_tc/devs/gps.py

import enum

from config.definitions import CustomServiceList
from tmtccmd.config import add_op_code_entry, add_service_op_code_entry
from tmtccmd.tc.pus_3_fsfw_hk import make_sid, generate_one_hk_command
from tmtccmd.config.definitions import QueueCommands, ServiceOpCodeDictT
from tmtccmd.logging import get_console_logger
from tmtccmd.tc.definitions import TcQueueT
from tmtccmd.tc.pus_8_funccmd import generate_action_command


LOGGER = get_console_logger()


class OpCodes:
    REQ_OS_HK = ["0", "hk-os"]
    RESET_GNSS = ["5", "reset"]


class Info:
    REQ_OS_HK = "Request One-Shot HK"
    RESET_GNSS = "Reset GNSS using reset pin"


class SetIds:
    HK = 0


def add_gps_cmds(cmd_dict: ServiceOpCodeDictT):
    op_code_dict = dict()
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=OpCodes.RESET_GNSS, info=Info.RESET_GNSS
    )
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=OpCodes.REQ_OS_HK, info=Info.REQ_OS_HK
    )
    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        op_code_entry=op_code_dict,
        name=CustomServiceList.GPS_CTRL.value,
        info="GPS/GNSS Controller",
    )


class SetIds:
    HK = 0


def pack_gps_command(object_id: bytes, tc_queue: TcQueueT, op_code: str):
    if op_code in OpCodes.RESET_GNSS:
        # TODO: This needs to be re-implemented
        LOGGER.warning("Reset pin handling needs to be re-implemented")
    if op_code in OpCodes.REQ_OS_HK:
        tc_queue.appendleft((QueueCommands.PRINT, f"GMSS: {Info.REQ_OS_HK}"))
        cmd = generate_one_hk_command(
            sid=make_sid(object_id=object_id, set_id=SetIds.HK), ssc=0
        )
        tc_queue.appendleft(cmd.pack_command_tuple())