added add_callsigns function in utils to add the callsigns on app startup

Author: Syzygicality

gs/backend/api/lifespan.py

@@ -6,7 +6,7 @@
 from fastapi_cache.backends.inmemory import InMemoryBackend
 
 from gs.backend.data.database.engine import get_db_session, setup_database
-from gs.backend.data.resources.utils import add_main_commands
+from gs.backend.data.resources.utils import add_callsigns, add_main_commands
 
 
 @asynccontextmanager
@@ -19,4 +19,5 @@ async def lifespan(_: FastAPI) -> AsyncGenerator[None, None]:
     # Otherwise, will get transaction is inactive error
     setup_database(get_db_session())
     add_main_commands(get_db_session())
+    add_callsigns(get_db_session())
     yield

gs/backend/data/resources/utils.py

@@ -1,5 +1,6 @@
 from sqlmodel import Session, select
 
+from gs.backend.data.resources.callsigns import callsigns
 from gs.backend.data.resources.main_commands import main_commands
 from gs.backend.data.tables.main_tables import MainCommand
 
@@ -10,6 +11,17 @@ def add_main_commands(session: Session) -> None:
     """
     query = select(MainCommand).limit(1)  # Check if the db is empty
     result = session.exec(query).first()
-    if result is None:
+    if not result:
         session.add_all(main_commands())
         session.commit()
+
+
+def add_callsigns(session: Session) -> None:
+    """
+    Setup the valid callsigns to the database
+    """
+    query = select(MainCommand).limit(1)
+    result = session.exec(query).first()
+    if not result:
+        session.add_all(callsigns())
+        session.commit()