ship customization: renaming

1 files changed, 4 insertions, 2 deletions

diff --git a/__main__.py b/__main__.py
index 36cbc28..5372651 100644
--- a/__main__.py
+++ b/__main__.py
@@ -3,7 +3,8 @@ from typing import Annotated
 
 from fastapi import FastAPI, Depends
 
-from .apirouters import agents, tasks
+from .apirouters import agents, tasks, customize_ship
+from .apirouters import tasks, customize_ship
 from .modules.database import cursor, conn
 
 
@@ -12,7 +13,7 @@ async def lifespan(app: FastAPI):
     cursor.execute(
         "CREATE TABLE IF NOT EXISTS agents(primary_key INTEGER PRIMARY KEY, callsign TEXT NOT NULL UNIQUE, token TEXT NOT NULL)")
     cursor.execute(
-        "CREATE TABLE IF NOT EXISTS ships(primary_key INTEGER PRIMARY KEY, symbol TEXT NOT NULL UNIQUE, task TEXT NOT NULL, params TEXT)")
+        "CREATE TABLE IF NOT EXISTS ships(primary_key INTEGER PRIMARY KEY, symbol TEXT NOT NULL UNIQUE, task TEXT NOT NULL, params TEXT, name TEXT)")
     conn.commit()
     agents.load_agents_from_database()
     yield
@@ -21,4 +22,5 @@ async def lifespan(app: FastAPI):
 app = FastAPI(lifespan=lifespan)
 app.include_router(agents.router)
 app.include_router(tasks.router)
+app.include_router(customize_ship.router)