agents are authenticated on space traders api

1 files changed, 5 insertions, 3 deletions

diff --git a/__main__.py b/__main__.py
index 9dba645..9b23484 100644
--- a/__main__.py
+++ b/__main__.py
@@ -4,13 +4,15 @@ from typing import Annotated
 from fastapi import FastAPI, Depends
 
 from .apirouters import agents
-from .modules import database
+from .modules.database import cursor, conn
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    database.cursor.execute(
-        "CREATE TABLE IF NOT EXISTS agents(primary_key INTEGER PRIMARY KEY, callsign TEXT NOT NULL UNIQUE, token_hash TEXT NOT NULL)")
+    cursor.execute(
+        "CREATE TABLE IF NOT EXISTS agents(primary_key INTEGER PRIMARY KEY, callsign TEXT NOT NULL UNIQUE, token TEXT NOT NULL)")
+    conn.commit()
+    agents.load_agents_from_database()
     yield