eive-tmtc/pus_tc/tmtcc_tc_packer_hook.py

"""
@brief      This file transfers control of TC packing to the user
@details    Template configuration file. Copy this folder to the TMTC commander root and adapt
            it to your needs.
"""

import os
from collections import deque
from typing import Union

from config.tmtcc_definitions import ServiceList
from tmtc_core.utility.tmtcc_logger import get_logger
from tmtc_core.pus_tc.tmtcc_pus_tc_base import TcQueueT
from tmtc_core.pus_tc.tmtcc_tc_service5_event import pack_service5_test_into
from tmtc_core.pus_tc.tmtcc_tc_service17_test import pack_service17_ping_command
from pus_tc.tmtcc_tc_p60dock import pack_p60dock_test_into
from pus_tc.tmtcc_tc_pdu2 import pack_pdu2_test_into
from pus_tc.tmtcc_tc_pdu1 import pack_pdu1_test_into
from pus_tc.tmtcc_tc_acu import pack_acu_test_into
from tmtc_core.core.tmtcc_object_id_manager import get_object_id
from config.tmtcc_object_ids import ObjectIds

LOGGER = get_logger()


def pack_service_queue_user(service: Union[int, str], op_code: int, service_queue: TcQueueT):
    if service == ServiceList.SERVICE_5:
        return pack_service5_test_into(service_queue)
    if service == ServiceList.SERVICE_17:
        return service_queue.appendleft(pack_service17_ping_command(ssc=1700).pack_command_tuple())
    if service == ServiceList.P60DOCK:
        object_id = get_object_id(ObjectIds.P60DOCK_HANDLER_ID)
        return pack_p60dock_test_into(object_id, service_queue)
    if service == ServiceList.PDU1:
        object_id = get_object_id(ObjectIds.PDU1_HANDLER_ID)
        return pack_pdu1_test_into(object_id, service_queue)
    if service == ServiceList.PDU2:
        object_id = get_object_id(ObjectIds.PDU2_HANDLER_ID)
        return pack_pdu2_test_into(object_id, service_queue)
    if service == ServiceList.ACU:
        object_id = get_object_id(ObjectIds.ACU_HANDLER_ID)
        return pack_acu_test_into(object_id, service_queue)
    LOGGER.warning("Invalid Service !")


def create_total_tc_queue_user() -> TcQueueT:
    if not os.path.exists("log"):
        os.mkdir("log")
    tc_queue = deque()
    pack_service5_test_into(tc_queue)
    tc_queue.appendleft(pack_service17_ping_command(ssc=1700).pack_command_tuple())
    return tc_queue