eive-obsw/generators/objects/objects.py

113 lines
3.6 KiB
Python
Raw Normal View History

2021-05-17 18:56:38 +02:00
#! /usr/bin/env python3
"""
@file objects.py
@brief Part of the Mission Information Base Exporter for the SOURCE project by KSat.
@details
Object exporter.
To use MySQLdb, run pip install mysqlclient or install in IDE.
On Windows, Build Tools installation might be necessary
@data 21.11.2019
"""
import datetime
2021-06-08 17:10:47 +02:00
from fsfwgen.objects.objects import sql_object_exporter, ObjectDefinitionParser, write_translation_file, \
2021-05-17 19:12:25 +02:00
export_object_file, write_translation_header_file
2021-06-08 17:10:47 +02:00
from fsfwgen.utility.printer import PrettyPrinter
from fsfwgen.utility.file_management import copy_file, move_file
2021-05-17 18:56:38 +02:00
2021-05-18 16:14:11 +02:00
from definitions import BspType, DATABASE_NAME
2021-05-17 18:56:38 +02:00
DATE_TODAY = datetime.datetime.now()
DATE_STRING_FULL = DATE_TODAY.strftime("%Y-%m-%d %H:%M:%S")
GENERATE_CSV = True
MOVE_CSV = True
GENERATE_CPP = True
COPY_CPP = True
2021-05-17 19:12:25 +02:00
GENERATE_HEADER = True
2021-05-17 18:56:38 +02:00
BSP_SELECT = BspType.BSP_Q7S
BSP_DIR_NAME = BSP_SELECT.value
2021-05-17 19:03:37 +02:00
if BSP_SELECT == BspType.BSP_Q7S or BSP_SELECT == BspType.BSP_LINUX_BOARD:
2021-05-17 20:03:56 +02:00
FSFW_CONFIG_ROOT = f"../../linux/fsfwconfig"
2021-05-17 19:03:37 +02:00
else:
2021-05-17 20:03:56 +02:00
FSFW_CONFIG_ROOT = f"../../{BSP_DIR_NAME}/fsfwconfig"
2021-05-17 19:03:37 +02:00
2021-05-17 20:03:56 +02:00
EXPORT_TO_SQL = True
CPP_COPY_DESTINATION = f"{FSFW_CONFIG_ROOT}/objects/"
2021-05-17 18:56:38 +02:00
CSV_MOVE_DESTINATION = "../"
CPP_FILENAME = "translateObjects.cpp"
2021-05-17 19:12:25 +02:00
CPP_H_FILENAME = "translateObjects.h"
2021-05-18 16:16:02 +02:00
CSV_OBJECT_FILENAME = f"{BSP_SELECT.value}_objects.csv"
2021-05-17 18:56:38 +02:00
FILE_SEPARATOR = ";"
2021-05-18 16:49:24 +02:00
OBJECTS_PATH = f"{FSFW_CONFIG_ROOT}/objects/systemObjectList.h"
FRAMEWORK_OBJECT_PATH = "../../fsfw/objectmanager/frameworkObjects.h"
COMMON_OBJECTS_PATH = "../../common/config/commonObjects.h"
OBJECTS_DEFINITIONS = [OBJECTS_PATH, FRAMEWORK_OBJECT_PATH, COMMON_OBJECTS_PATH]
2021-05-17 18:56:38 +02:00
SQL_DELETE_OBJECTS_CMD = """
DROP TABLE IF EXISTS Objects
"""
SQL_CREATE_OBJECTS_CMD = """
CREATE TABLE IF NOT EXISTS Objects(
id INTEGER PRIMARY KEY,
objectid TEXT,
name TEXT
)
"""
SQL_INSERT_INTO_OBJECTS_CMD = """
INSERT INTO Objects(objectid, name)
VALUES(?,?)
"""
def main():
print("Parsing objects: ")
list_items = parse_objects()
handle_file_export(list_items)
if EXPORT_TO_SQL:
print("ObjectParser: Exporting to SQL")
sql_object_exporter(
object_table=list_items, delete_cmd=SQL_DELETE_OBJECTS_CMD, insert_cmd=SQL_INSERT_INTO_OBJECTS_CMD,
2021-05-18 16:14:11 +02:00
create_cmd=SQL_CREATE_OBJECTS_CMD, db_filename=f"../{DATABASE_NAME}"
2021-05-17 18:56:38 +02:00
)
def parse_objects():
# fetch objects
object_parser = ObjectDefinitionParser(OBJECTS_DEFINITIONS)
subsystem_definitions = object_parser.parse_files()
# id_subsystem_definitions.update(framework_subsystem_definitions)
list_items = sorted(subsystem_definitions.items())
PrettyPrinter.pprint(list_items)
print("ObjectParser: Number of objects: ", len(list_items))
return list_items
def handle_file_export(list_items):
if GENERATE_CPP:
print("ObjectParser: Generating translation C++ file.")
write_translation_file(filename=CPP_FILENAME, list_of_entries=list_items, date_string_full=DATE_STRING_FULL)
if COPY_CPP:
print("ObjectParser: Copying object file to " + CPP_COPY_DESTINATION)
copy_file(CPP_FILENAME, CPP_COPY_DESTINATION)
2021-05-17 19:12:25 +02:00
if GENERATE_HEADER:
write_translation_header_file(filename=CPP_H_FILENAME)
copy_file(filename=CPP_H_FILENAME, destination=CPP_COPY_DESTINATION)
2021-05-17 18:56:38 +02:00
if GENERATE_CSV:
print("ObjectParser: Generating text export.")
export_object_file(filename=CSV_OBJECT_FILENAME, object_list=list_items, file_separator=FILE_SEPARATOR)
if MOVE_CSV:
2021-05-17 19:03:37 +02:00
move_file(file_name=CSV_OBJECT_FILENAME, destination=CSV_MOVE_DESTINATION)
2021-05-17 18:56:38 +02:00
if __name__ == "__main__":
main()