Pouzor
974e782057
- Add dict[str, Any] / list[Any] type params throughout (fingerprint, models, schemas, scanner, status_checker) - Add return type annotations to all route functions (nodes, edges, canvas, scan, auth, status, main) - Fix no-any-return in security.py: cast pwd/jwt results to bool/str explicitly - Fix canvas.py: use model_validate() for NodeResponse/EdgeResponse, rename db_node/db_edge upsert vars - Fix scheduler.py: rename 'result' → 'check_result' to avoid type collision - Fix get_db() return type: AsyncGenerator[AsyncSession, None] - Add types-PyYAML for yaml import stubs - Fix scanner.py: remove unnecessary type: ignore comment (nmap has stubs) - Fix scan.py: wrap scalars().all() with list() for Sequence→list compatibility
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.api.deps import get_current_user
|
|
from app.db.database import get_db
|
|
from app.db.models import Edge
|
|
from app.schemas.edges import EdgeCreate, EdgeResponse, EdgeUpdate
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("", response_model=list[EdgeResponse])
|
|
async def list_edges(db: AsyncSession = Depends(get_db), _: str = Depends(get_current_user)) -> list[Edge]:
|
|
result = await db.execute(select(Edge))
|
|
return list(result.scalars().all())
|
|
|
|
|
|
@router.post("", response_model=EdgeResponse, status_code=status.HTTP_201_CREATED)
|
|
async def create_edge(body: EdgeCreate, db: AsyncSession = Depends(get_db), _: str = Depends(get_current_user)) -> Edge:
|
|
edge = Edge(**body.model_dump())
|
|
db.add(edge)
|
|
await db.commit()
|
|
await db.refresh(edge)
|
|
return edge
|
|
|
|
|
|
@router.delete("/{edge_id}", status_code=status.HTTP_204_NO_CONTENT)
|
|
async def delete_edge(edge_id: str, db: AsyncSession = Depends(get_db), _: str = Depends(get_current_user)) -> None:
|
|
edge = await db.get(Edge, edge_id)
|
|
if not edge:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Edge not found")
|
|
await db.delete(edge)
|
|
await db.commit()
|
|
|
|
|
|
@router.patch("/{edge_id}", response_model=EdgeResponse)
|
|
async def update_edge(
|
|
edge_id: str, body: EdgeUpdate, db: AsyncSession = Depends(get_db), _: str = Depends(get_current_user)
|
|
) -> Edge:
|
|
edge = await db.get(Edge, edge_id)
|
|
if not edge:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Edge not found")
|
|
for field, value in body.model_dump(exclude_unset=True).items():
|
|
setattr(edge, field, value)
|
|
await db.commit()
|
|
await db.refresh(edge)
|
|
return edge
|