eive/eive-tmtc
eive/eive-tmtc
14
0
Fork 0
Code Issues Pull Requests 2 Projects Releases 97 Wiki Activity

added store enum

Browse Source
This commit is contained in:
Robin Müller 2023-02-17 19:22:56 +01:00
parent ab9f2ca05d
commit 676f851659
No known key found for this signature in database
GPG Key ID: 71B58F8A3CDFA9AC
1 changed files with 22 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
eive_tmtc/tmtc/tm_store.py
View File

@ -1,4 +1,5 @@
import datetime
import enum
import logging
import math
import struct
@ -73,12 +74,20 @@ def add_persistent_tm_store_cmd_defs(defs: TmtcDefinitionWrapper):
)
class TmStoreSelect(enum.IntEnum):
OK_TM_STORE = 0
NOT_OK_TM_STORE = 1
MISC_TM_STORE = 2
HK_TM_STORE = 3
CFDP_TM_STORE = 4
STORE_DICT = {
OK_TM_STORE: "OK Store (Verification)",
NOT_OK_TM_STORE: "NOT OK Store (Events, Verification Failures..)",
MISC_TM_STORE: "Miscellaneous Store",
HK_TM_STORE: "HK TM Store",
CFDP_TM_STORE: "CFDP TM Store",
TmStoreSelect.OK_TM_STORE: (OK_TM_STORE, "OK Store (Verification)"),
TmStoreSelect.NOT_OK_TM_STORE: (NOT_OK_TM_STORE, "NOT OK Store (Events, Verification Failures..)"),
TmStoreSelect.MISC_TM_STORE: (MISC_TM_STORE, "Miscellaneous Store"),
TmStoreSelect.HK_TM_STORE: (HK_TM_STORE, "HK TM Store"),
TmStoreSelect.CFDP_TM_STORE: (CFDP_TM_STORE, "CFDP TM Store"),
}
@ -135,17 +144,18 @@ def store_select_prompt() -> (ObjectIdU32, str):
obj_id_dict = get_object_ids()
print("Available TM stores:")
idx_to_obj_id = dict()
for idx, (k, v) in enumerate(STORE_DICT.items()):
idx_to_obj_id.update({idx: (k, v)})
print(f" {idx}: {v}")
for k, v in STORE_DICT.items():
print(f" {k}: {v[1]}")
while True:
target_index = int(
input("Please enter the target store for the TM store transaction: ")
)
obj_id_and_store_str = idx_to_obj_id.get(target_index)
if obj_id_and_store_str is None:
desc_and_obj_id = STORE_DICT.get(TmStoreSelect(target_index))
if desc_and_obj_id is None:
_LOGGER.warning("Invalid index. Try again")
continue
break
obj_id_raw = obj_id_and_store_str[0]
return obj_id_dict.get(obj_id_raw), obj_id_and_store_str[1]
obj_id_raw = desc_and_obj_id[0]
obj_id = obj_id_dict.get(obj_id_raw)
print(f"Selected store: {obj_id} ({desc_and_obj_id[1]})")
return obj_id_dict.get(obj_id_raw), desc_and_obj_id[1]