224 lines
6.7 KiB
Python
224 lines
6.7 KiB
Python
from typing import List, Optional
|
|
from fastapi import APIRouter, Depends, HTTPException, status, Request
|
|
from sqlalchemy.orm import Session
|
|
from datetime import date
|
|
from app.api.deps import get_db, get_current_read_user, get_current_operator_user
|
|
from app.crud.crud_ppr import ppr as crud_ppr
|
|
from app.crud.crud_journal import journal as crud_journal
|
|
from app.schemas.ppr import PPR, PPRCreate, PPRUpdate, PPRStatus, PPRStatusUpdate, Journal
|
|
from app.models.ppr import User
|
|
from app.core.utils import get_client_ip
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/", response_model=List[PPR])
|
|
async def get_pprs(
|
|
request: Request,
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
status: Optional[PPRStatus] = None,
|
|
date_from: Optional[date] = None,
|
|
date_to: Optional[date] = None,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_read_user)
|
|
):
|
|
"""Get PPR records with optional filtering"""
|
|
pprs = crud_ppr.get_multi(
|
|
db, skip=skip, limit=limit, status=status,
|
|
date_from=date_from, date_to=date_to
|
|
)
|
|
return pprs
|
|
|
|
|
|
@router.post("/", response_model=PPR)
|
|
async def create_ppr(
|
|
request: Request,
|
|
ppr_in: PPRCreate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_operator_user)
|
|
):
|
|
"""Create a new PPR record"""
|
|
client_ip = get_client_ip(request)
|
|
ppr = crud_ppr.create(db, obj_in=ppr_in, created_by=current_user.username, user_ip=client_ip)
|
|
|
|
# Send real-time update via WebSocket
|
|
if hasattr(request.app.state, 'connection_manager'):
|
|
await request.app.state.connection_manager.broadcast({
|
|
"type": "ppr_created",
|
|
"data": {
|
|
"id": ppr.id,
|
|
"ac_reg": ppr.ac_reg,
|
|
"status": ppr.status.value
|
|
}
|
|
})
|
|
|
|
return ppr
|
|
|
|
|
|
@router.get("/{ppr_id}", response_model=PPR)
|
|
async def get_ppr(
|
|
ppr_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_read_user)
|
|
):
|
|
"""Get a specific PPR record"""
|
|
ppr = crud_ppr.get(db, ppr_id=ppr_id)
|
|
if not ppr:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="PPR record not found"
|
|
)
|
|
return ppr
|
|
|
|
|
|
@router.put("/{ppr_id}", response_model=PPR)
|
|
async def update_ppr(
|
|
request: Request,
|
|
ppr_id: int,
|
|
ppr_in: PPRUpdate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_operator_user)
|
|
):
|
|
"""Update a PPR record"""
|
|
db_ppr = crud_ppr.get(db, ppr_id=ppr_id)
|
|
if not db_ppr:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="PPR record not found"
|
|
)
|
|
|
|
client_ip = get_client_ip(request)
|
|
ppr = crud_ppr.update(db, db_obj=db_ppr, obj_in=ppr_in, user=current_user.username, user_ip=client_ip)
|
|
|
|
# Send real-time update
|
|
if hasattr(request.app.state, 'connection_manager'):
|
|
await request.app.state.connection_manager.broadcast({
|
|
"type": "ppr_updated",
|
|
"data": {
|
|
"id": ppr.id,
|
|
"ac_reg": ppr.ac_reg,
|
|
"status": ppr.status.value
|
|
}
|
|
})
|
|
|
|
return ppr
|
|
|
|
|
|
@router.patch("/{ppr_id}", response_model=PPR)
|
|
async def patch_ppr(
|
|
request: Request,
|
|
ppr_id: int,
|
|
ppr_in: PPRUpdate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_operator_user)
|
|
):
|
|
"""Partially update a PPR record (only provided fields will be updated)"""
|
|
db_ppr = crud_ppr.get(db, ppr_id=ppr_id)
|
|
if not db_ppr:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="PPR record not found"
|
|
)
|
|
|
|
client_ip = get_client_ip(request)
|
|
ppr = crud_ppr.update(db, db_obj=db_ppr, obj_in=ppr_in, user=current_user.username, user_ip=client_ip)
|
|
|
|
# Send real-time update
|
|
if hasattr(request.app.state, 'connection_manager'):
|
|
await request.app.state.connection_manager.broadcast({
|
|
"type": "ppr_updated",
|
|
"data": {
|
|
"id": ppr.id,
|
|
"ac_reg": ppr.ac_reg,
|
|
"status": ppr.status.value
|
|
}
|
|
})
|
|
|
|
return ppr
|
|
|
|
|
|
@router.patch("/{ppr_id}/status", response_model=PPR)
|
|
async def update_ppr_status(
|
|
request: Request,
|
|
ppr_id: int,
|
|
status_update: PPRStatusUpdate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_operator_user)
|
|
):
|
|
"""Update PPR status (LANDED, DEPARTED, etc.)"""
|
|
client_ip = get_client_ip(request)
|
|
ppr = crud_ppr.update_status(
|
|
db,
|
|
ppr_id=ppr_id,
|
|
status=status_update.status,
|
|
timestamp=status_update.timestamp,
|
|
user=current_user.username,
|
|
user_ip=client_ip
|
|
)
|
|
if not ppr:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="PPR record not found"
|
|
)
|
|
|
|
# Send real-time update
|
|
if hasattr(request.app.state, 'connection_manager'):
|
|
await request.app.state.connection_manager.broadcast({
|
|
"type": "status_update",
|
|
"data": {
|
|
"id": ppr.id,
|
|
"ac_reg": ppr.ac_reg,
|
|
"status": ppr.status.value,
|
|
"timestamp": ppr.landed_dt.isoformat() if ppr.landed_dt else (ppr.departed_dt.isoformat() if ppr.departed_dt else None)
|
|
}
|
|
})
|
|
|
|
return ppr
|
|
|
|
|
|
@router.delete("/{ppr_id}", response_model=PPR)
|
|
async def delete_ppr(
|
|
request: Request,
|
|
ppr_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_operator_user)
|
|
):
|
|
"""Delete (soft delete) a PPR record"""
|
|
client_ip = get_client_ip(request)
|
|
ppr = crud_ppr.delete(db, ppr_id=ppr_id, user=current_user.username, user_ip=client_ip)
|
|
if not ppr:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="PPR record not found"
|
|
)
|
|
|
|
# Send real-time update
|
|
if hasattr(request.app.state, 'connection_manager'):
|
|
await request.app.state.connection_manager.broadcast({
|
|
"type": "ppr_deleted",
|
|
"data": {
|
|
"id": ppr.id,
|
|
"ac_reg": ppr.ac_reg
|
|
}
|
|
})
|
|
|
|
return ppr
|
|
|
|
|
|
@router.get("/{ppr_id}/journal", response_model=List[Journal])
|
|
async def get_ppr_journal(
|
|
ppr_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_read_user)
|
|
):
|
|
"""Get journal entries for a specific PPR"""
|
|
# Verify PPR exists
|
|
ppr = crud_ppr.get(db, ppr_id=ppr_id)
|
|
if not ppr:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="PPR record not found"
|
|
)
|
|
|
|
return crud_journal.get_by_ppr_id(db, ppr_id=ppr_id) |