backend dev done

Author: rahul-aot

backend/app/api/schema.py

@@ -0,0 +1,10 @@
+from fastapi import APIRouter
+from app.services.schema_loader import load_database_schema
+
+router = APIRouter(tags=["Schema"])
+
+
+@router.get("/schema")
+def get_schema():
+    schema = load_database_schema()
+    return {"tables": schema}

backend/app/api/schema_create.py

@@ -0,0 +1,13 @@
+from fastapi import APIRouter, HTTPException
+from app.services.schema_creator import create_table
+
+router = APIRouter(tags=["Schema"])
+
+
+@router.post("/schema/create")
+def create_schema(payload: dict):
+    try:
+        create_table(payload["table_name"], payload["columns"])
+        return {"status": "table created"}
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))

backend/app/services/schema_creator.py

@@ -0,0 +1,43 @@
+import psycopg2
+from app.core.config import settings
+
+ALLOWED_TYPES = {
+    "text": "TEXT",
+    "int": "INTEGER",
+    "float": "REAL",
+    "timestamp": "TIMESTAMP",
+}
+
+
+def create_table(table_name: str, columns: list[dict]):
+    cols_sql = []
+
+    for col in columns:
+        if col["type"] not in ALLOWED_TYPES:
+            raise ValueError(f"Unsupported column type: {col['type']}")
+
+        cols_sql.append(
+            f"{col['name']} {ALLOWED_TYPES[col['type']]}"
+        )
+
+    sql = f"""
+    CREATE TABLE IF NOT EXISTS {table_name} (
+        id SERIAL PRIMARY KEY,
+        {", ".join(cols_sql)}
+    );
+    """
+
+    conn = psycopg2.connect(
+        host=settings.DB_HOST,
+        port=settings.DB_PORT,
+        dbname=settings.DB_NAME,
+        user=settings.DB_USER,
+        password=settings.DB_PASSWORD,
+    )
+
+    try:
+        with conn.cursor() as cursor:
+            cursor.execute(sql)
+            conn.commit()
+    finally:
+        conn.close()

backend/main.py

@@ -2,6 +2,8 @@
 from fastapi.middleware.cors import CORSMiddleware
 
 from app.api.query import router as query_router
+from app.api.schema import router as schema_router
+from app.api.schema_create import router as schema_create_router
 from app.core.config import settings
 
 app = FastAPI(
@@ -20,7 +22,10 @@
 )
 
 # Routes
+
+app.include_router(schema_router, prefix="/api")
 app.include_router(query_router, prefix="/api")
+app.include_router(schema_create_router, prefix="/api")
 
 
 @app.get("/health")