eive-tmtc/pus_tc/system/controllers.py

from tmtccmd.tc.definitions import TcQueueT
from tmtccmd.config import QueueCommands

from .common import command_mode
import config.object_ids as obj_ids

from pus_tc.prompt_parameters import prompt_parameters


class OpCodes:
    THERMAL_CONTROLLER = [obj_ids.THERMAL_CONTROLLER_ID.hex(), "thermal_controller"]
    CORE_CONTROLLER = [obj_ids.CORE_CONTROLLER_ID.hex(), "core_controller"]


class Info:
    THERMAL_CONTROLLER = "Thermal controller"
    CORE_CONTROLLER = "ACS controller"


def pack_controller_commands(tc_queue: TcQueueT, op_code: str):
    parameters = prompt_parameters(
        [
            {"name": "Mode", "defaultValue": "2"},
            {"name": "Submode", "defaultValue": "0"},
        ]
    )
    mode = int(parameters["Mode"])
    if mode < 0 or mode > 2:
        print("Invalid Mode, defaulting to OFF")
        mode = 0
    submode = int(parameters["Submode"])
    command_mode(
        object_id=get_object_from_op_code(op_code),
        mode=mode,
        submode=submode,
        tc_queue=tc_queue,
        info=op_code + " to " + str(mode) + "," + str(submode),
    )

    tc_queue.appendleft((QueueCommands.WAIT, 20))

    command_mode(
        object_id=get_object_from_op_code(op_code),
        mode=0,
        submode=0,
        tc_queue=tc_queue,
        info=op_code + " to 0,0",
    )


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