初始化

app/api/database.py

@@ -0,0 +1,56 @@
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.orm import Session
+from typing import List
+from ..core.database import get_db
+from ..schemas.database import (
+    SQLExecuteRequest, SQLExecuteResponse, TableDataRequest,
+    TableDataResponse, CreateTableRequest, ImportDataRequest
+)
+from ..services.database import DatabaseService
+
+router = APIRouter(prefix="/database", tags=["数据库管理"])
+
+@router.post("/execute", response_model=SQLExecuteResponse)
+def execute_sql(request: SQLExecuteRequest, db: Session = Depends(get_db)):
+    """执行SQL语句"""
+    result = DatabaseService.execute_sql(db, request.sql)
+    return SQLExecuteResponse(**result)
+
+@router.post("/table-data", response_model=TableDataResponse)
+def get_table_data(request: TableDataRequest, db: Session = Depends(get_db)):
+    """获取表数据"""
+    result = DatabaseService.get_table_data(
+        db,
+        request.table_name,
+        request.limit or 100,
+        request.offset or 0
+    )
+    return TableDataResponse(**result)
+
+@router.post("/create-table", response_model=SQLExecuteResponse)
+def create_table(request: CreateTableRequest, db: Session = Depends(get_db)):
+    """创建表"""
+    result = DatabaseService.create_table(
+        db,
+        request.table_name,
+        request.columns,
+        request.primary_key
+    )
+    return SQLExecuteResponse(**result)
+
+@router.delete("/drop-table/{table_name}", response_model=SQLExecuteResponse)
+def drop_table(table_name: str, db: Session = Depends(get_db)):
+    """删除表"""
+    result = DatabaseService.drop_table(db, table_name)
+    return SQLExecuteResponse(**result)
+
+@router.post("/import-data", response_model=SQLExecuteResponse)
+def import_data(request: ImportDataRequest, db: Session = Depends(get_db)):
+    """导入数据"""
+    result = DatabaseService.import_data(db, request.table_name, request.data)
+    return SQLExecuteResponse(**result)
+
+@router.get("/tables", response_model=List[str])
+def get_table_list():
+    """获取所有表名"""
+    return DatabaseService.get_table_list()