add TCS obj prompt

eive_tmtc/tmtc/obj_prompt.py

@@ -15,6 +15,7 @@ from eive_tmtc.config.object_ids import (
     SUS_6_R_LOC_XFYBZM_PT_XF,
     RW1_ID,
     RW2_ID,
+    RTD_0_PLOC_HSPD,
 )
 
 SUBSYSTEM_DICT = {
@@ -40,6 +41,20 @@ ACS_OBJ_DICT = {
     13: ("RW 2", RW2_ID),
 }
 
+TCS_OBJ_DICT = {
+    0: ("RTD 0", RTD_0_PLOC_HSPD),
+}
+
+
+def get_obj_if_from_dict(lut: dict) -> bytes:
+    for k, v in lut.items():
+        print(f"{k}: {v[0]}")
+    obj_key = int(input("Please specify target object by key: "))
+    acs_obj = lut[obj_key]
+    if acs_obj is None:
+        raise ValueError("invalid key")
+    return lut[1]
+
 
 def prompt_object() -> bytes:
     for k, v in SUBSYSTEM_DICT.items():
@@ -49,10 +64,9 @@ def prompt_object() -> bytes:
     if subsystem is None:
         raise ValueError("invalid key")
     if subsystem == "acs":
-        for k, v in ACS_OBJ_DICT.items():
+        return get_obj_if_from_dict(ACS_OBJ_DICT)
-            print(f"{k}: {v[0]}")
+    elif subsystem == "tcs":
-        obj_key = int(input("Please specify target object by key: "))
+        return get_obj_if_from_dict(TCS_OBJ_DICT)
-        acs_obj = ACS_OBJ_DICT[obj_key]
+    else:
-        if acs_obj is None:
+        print(f"No object for subsystem {subsystem}")
-            raise ValueError("invalid key")
+        return bytes()
-        return acs_obj[1]