Circuits handling

This commit is contained in:
2025-12-16 12:59:43 -05:00
parent 65eb3272f2
commit 2d4f1467de
7 changed files with 337 additions and 2 deletions

View File

@@ -1,5 +1,5 @@
from fastapi import APIRouter
from app.api.endpoints import auth, pprs, public, aircraft, airport, local_flights, departures, arrivals
from app.api.endpoints import auth, pprs, public, aircraft, airport, local_flights, departures, arrivals, circuits
api_router = APIRouter()
@@ -8,6 +8,7 @@ api_router.include_router(pprs.router, prefix="/pprs", tags=["pprs"])
api_router.include_router(local_flights.router, prefix="/local-flights", tags=["local_flights"])
api_router.include_router(departures.router, prefix="/departures", tags=["departures"])
api_router.include_router(arrivals.router, prefix="/arrivals", tags=["arrivals"])
api_router.include_router(circuits.router, prefix="/circuits", tags=["circuits"])
api_router.include_router(public.router, prefix="/public", tags=["public"])
api_router.include_router(aircraft.router, prefix="/aircraft", tags=["aircraft"])
api_router.include_router(airport.router, prefix="/airport", tags=["airport"])

View File

@@ -0,0 +1,108 @@
from typing import List
from fastapi import APIRouter, Depends, HTTPException, status, Request
from sqlalchemy.orm import Session
from app.api.deps import get_db, get_current_read_user, get_current_operator_user
from app.crud.crud_circuit import crud_circuit
from app.schemas.circuit import Circuit, CircuitCreate, CircuitUpdate
from app.models.ppr import User
from app.core.utils import get_client_ip
router = APIRouter()
@router.get("/", response_model=List[Circuit])
async def get_circuits(
skip: int = 0,
limit: int = 100,
db: Session = Depends(get_db),
current_user: User = Depends(get_current_read_user)
):
"""Get circuit records"""
circuits = crud_circuit.get_multi(db, skip=skip, limit=limit)
return circuits
@router.get("/flight/{local_flight_id}", response_model=List[Circuit])
async def get_circuits_by_flight(
local_flight_id: int,
db: Session = Depends(get_db),
current_user: User = Depends(get_current_read_user)
):
"""Get all circuits for a specific local flight"""
circuits = crud_circuit.get_by_local_flight(db, local_flight_id=local_flight_id)
return circuits
@router.post("/", response_model=Circuit)
async def create_circuit(
request: Request,
circuit_in: CircuitCreate,
db: Session = Depends(get_db),
current_user: User = Depends(get_current_operator_user)
):
"""Record a new circuit (touch and go) for a local flight"""
circuit = crud_circuit.create(db, obj_in=circuit_in)
# Send real-time update via WebSocket
if hasattr(request.app.state, 'connection_manager'):
await request.app.state.connection_manager.broadcast({
"type": "circuit_recorded",
"data": {
"id": circuit.id,
"local_flight_id": circuit.local_flight_id,
"circuit_timestamp": circuit.circuit_timestamp.isoformat()
}
})
return circuit
@router.get("/{circuit_id}", response_model=Circuit)
async def get_circuit(
circuit_id: int,
db: Session = Depends(get_db),
current_user: User = Depends(get_current_read_user)
):
"""Get a specific circuit record"""
circuit = crud_circuit.get(db, circuit_id=circuit_id)
if not circuit:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail="Circuit record not found"
)
return circuit
@router.put("/{circuit_id}", response_model=Circuit)
async def update_circuit(
circuit_id: int,
circuit_in: CircuitUpdate,
db: Session = Depends(get_db),
current_user: User = Depends(get_current_operator_user)
):
"""Update a circuit record"""
circuit = crud_circuit.get(db, circuit_id=circuit_id)
if not circuit:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail="Circuit record not found"
)
circuit = crud_circuit.update(db, db_obj=circuit, obj_in=circuit_in)
return circuit
@router.delete("/{circuit_id}")
async def delete_circuit(
circuit_id: int,
db: Session = Depends(get_db),
current_user: User = Depends(get_current_operator_user)
):
"""Delete a circuit record"""
circuit = crud_circuit.get(db, circuit_id=circuit_id)
if not circuit:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail="Circuit record not found"
)
crud_circuit.delete(db, circuit_id=circuit_id)
return {"detail": "Circuit record deleted"}

View File

@@ -0,0 +1,55 @@
from typing import List, Optional
from sqlalchemy.orm import Session
from sqlalchemy import desc
from datetime import datetime
from app.models.circuit import Circuit
from app.schemas.circuit import CircuitCreate, CircuitUpdate
class CRUDCircuit:
def get(self, db: Session, circuit_id: int) -> Optional[Circuit]:
return db.query(Circuit).filter(Circuit.id == circuit_id).first()
def get_by_local_flight(self, db: Session, local_flight_id: int) -> List[Circuit]:
"""Get all circuits for a specific local flight"""
return db.query(Circuit).filter(
Circuit.local_flight_id == local_flight_id
).order_by(Circuit.circuit_timestamp).all()
def get_multi(
self,
db: Session,
skip: int = 0,
limit: int = 100
) -> List[Circuit]:
return db.query(Circuit).order_by(desc(Circuit.created_at)).offset(skip).limit(limit).all()
def create(self, db: Session, obj_in: CircuitCreate) -> Circuit:
db_obj = Circuit(
local_flight_id=obj_in.local_flight_id,
circuit_timestamp=obj_in.circuit_timestamp
)
db.add(db_obj)
db.commit()
db.refresh(db_obj)
return db_obj
def update(self, db: Session, db_obj: Circuit, obj_in: CircuitUpdate) -> Circuit:
obj_data = obj_in.dict(exclude_unset=True)
for field, value in obj_data.items():
setattr(db_obj, field, value)
db.add(db_obj)
db.commit()
db.refresh(db_obj)
return db_obj
def delete(self, db: Session, circuit_id: int) -> bool:
circuit = self.get(db, circuit_id)
if circuit:
db.delete(circuit)
db.commit()
return True
return False
crud_circuit = CRUDCircuit()

View File

@@ -0,0 +1,12 @@
from sqlalchemy import Column, DateTime, BigInteger, ForeignKey
from sqlalchemy.sql import func
from app.db.session import Base
class Circuit(Base):
__tablename__ = "circuits"
id = Column(BigInteger, primary_key=True, autoincrement=True)
local_flight_id = Column(BigInteger, ForeignKey("local_flights.id", ondelete="CASCADE"), nullable=False, index=True)
circuit_timestamp = Column(DateTime, nullable=False, index=True)
created_at = Column(DateTime, nullable=False, server_default=func.current_timestamp())

View File

@@ -0,0 +1,24 @@
from pydantic import BaseModel
from datetime import datetime
from typing import Optional
class CircuitBase(BaseModel):
local_flight_id: int
circuit_timestamp: datetime
class CircuitCreate(CircuitBase):
pass
class CircuitUpdate(BaseModel):
circuit_timestamp: Optional[datetime] = None
class Circuit(CircuitBase):
id: int
created_at: datetime
class Config:
from_attributes = True