loganandClaude Opus 4.7
439155c92e
FastAPI backend with auth, user, instance, ticket, query, dashboard APIs. React + TypeScript frontend with mock data UI (10 pages). SQLite + async SQLAlchemy, JWT auth, role-based access control. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
146 lines
4.7 KiB
Python
146 lines
4.7 KiB
Python
from typing import Optional, List
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.database import get_db
|
|
from app.core.security import get_current_user, require_role
|
|
from app.models.user import User
|
|
from app.models.instance import Instance
|
|
from app.schemas.instance import InstanceCreate, InstanceUpdate, InstanceResponse, ConnectionTestResult
|
|
from app.services.instance_service import InstanceService
|
|
|
|
router = APIRouter(prefix="/instances", tags=["数据源管理"])
|
|
|
|
|
|
@router.get("", response_model=List[InstanceResponse])
|
|
async def list_instances(
|
|
db_type: Optional[str] = None,
|
|
env: Optional[str] = None,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
stmt = select(Instance).order_by(Instance.id)
|
|
if db_type:
|
|
stmt = stmt.where(Instance.db_type == db_type)
|
|
if env:
|
|
stmt = stmt.where(Instance.env == env)
|
|
result = await db.execute(stmt)
|
|
return result.scalars().all()
|
|
|
|
|
|
@router.post("", response_model=InstanceResponse, status_code=201)
|
|
async def create_instance(
|
|
data: InstanceCreate,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role("admin", "dba")),
|
|
):
|
|
instance = Instance(
|
|
name=data.name,
|
|
db_type=data.db_type,
|
|
host=data.host,
|
|
port=data.port,
|
|
username=data.username,
|
|
password=data.password,
|
|
database=data.database,
|
|
env=data.env,
|
|
description=data.description,
|
|
)
|
|
db.add(instance)
|
|
await db.commit()
|
|
await db.refresh(instance)
|
|
return instance
|
|
|
|
|
|
@router.get("/{instance_id}", response_model=InstanceResponse)
|
|
async def get_instance(
|
|
instance_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
result = await db.execute(select(Instance).where(Instance.id == instance_id))
|
|
instance = result.scalar_one_or_none()
|
|
if not instance:
|
|
raise HTTPException(status_code=404, detail="数据源不存在")
|
|
return instance
|
|
|
|
|
|
@router.put("/{instance_id}", response_model=InstanceResponse)
|
|
async def update_instance(
|
|
instance_id: int,
|
|
data: InstanceUpdate,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role("admin", "dba")),
|
|
):
|
|
result = await db.execute(select(Instance).where(Instance.id == instance_id))
|
|
instance = result.scalar_one_or_none()
|
|
if not instance:
|
|
raise HTTPException(status_code=404, detail="数据源不存在")
|
|
|
|
update_data = data.model_dump(exclude_unset=True)
|
|
for key, value in update_data.items():
|
|
setattr(instance, key, value)
|
|
await db.commit()
|
|
await db.refresh(instance)
|
|
return instance
|
|
|
|
|
|
@router.delete("/{instance_id}")
|
|
async def delete_instance(
|
|
instance_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role("admin", "dba")),
|
|
):
|
|
result = await db.execute(select(Instance).where(Instance.id == instance_id))
|
|
instance = result.scalar_one_or_none()
|
|
if not instance:
|
|
raise HTTPException(status_code=404, detail="数据源不存在")
|
|
|
|
await db.delete(instance)
|
|
await db.commit()
|
|
return {"message": "数据源已删除"}
|
|
|
|
|
|
@router.post("/{instance_id}/test", response_model=ConnectionTestResult)
|
|
async def test_connection(
|
|
instance_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role("admin", "dba")),
|
|
):
|
|
result = await db.execute(select(Instance).where(Instance.id == instance_id))
|
|
instance = result.scalar_one_or_none()
|
|
if not instance:
|
|
raise HTTPException(status_code=404, detail="数据源不存在")
|
|
|
|
return await InstanceService.test_connection(instance)
|
|
|
|
|
|
@router.get("/{instance_id}/databases")
|
|
async def list_databases(
|
|
instance_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
result = await db.execute(select(Instance).where(Instance.id == instance_id))
|
|
instance = result.scalar_one_or_none()
|
|
if not instance:
|
|
raise HTTPException(status_code=404, detail="数据源不存在")
|
|
|
|
return await InstanceService.list_databases(instance)
|
|
|
|
|
|
@router.get("/{instance_id}/databases/{database}/tables")
|
|
async def list_tables(
|
|
instance_id: int,
|
|
database: str,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
result = await db.execute(select(Instance).where(Instance.id == instance_id))
|
|
instance = result.scalar_one_or_none()
|
|
if not instance:
|
|
raise HTTPException(status_code=404, detail="数据源不存在")
|
|
|
|
return await InstanceService.list_tables(instance, database)
|