76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
from typing import List, Optional
|
|
from app.db.database import get_db
|
|
from app.models.models import Contractor
|
|
from app.security import get_current_active_user
|
|
import app.schemas
|
|
|
|
router = APIRouter(
|
|
prefix="/contractors",
|
|
tags=["Contractors"],
|
|
dependencies=[Depends(get_current_active_user)]
|
|
)
|
|
|
|
@router.post("/", response_model=app.schemas.Contractor)
|
|
def create_contractor(contractor: app.schemas.ContractorCreate, db: Session = Depends(get_db)):
|
|
db_contractor = Contractor(**contractor.dict())
|
|
db.add(db_contractor)
|
|
db.commit()
|
|
db.refresh(db_contractor)
|
|
return db_contractor
|
|
|
|
@router.get("/", response_model=List[app.schemas.Contractor])
|
|
def read_contractors(
|
|
parent_id: Optional[int] = None,
|
|
only_parents: bool = False,
|
|
is_active: Optional[bool] = None,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
query = db.query(Contractor)
|
|
if only_parents:
|
|
query = query.filter(Contractor.parent_id == None)
|
|
elif parent_id is not None:
|
|
query = query.filter(Contractor.parent_id == parent_id)
|
|
|
|
if is_active is not None:
|
|
query = query.filter(Contractor.is_active == is_active)
|
|
return query.all()
|
|
|
|
@router.get("/{contractor_id}", response_model=app.schemas.Contractor)
|
|
def read_contractor(contractor_id: int, db: Session = Depends(get_db)):
|
|
db_contractor = db.query(Contractor).filter(Contractor.id == contractor_id).first()
|
|
if not db_contractor:
|
|
raise HTTPException(status_code=404, detail="Contractor not found")
|
|
return db_contractor
|
|
|
|
@router.put("/{contractor_id}", response_model=app.schemas.Contractor)
|
|
@router.patch("/{contractor_id}", response_model=app.schemas.Contractor)
|
|
def update_contractor(
|
|
contractor_id: int,
|
|
contractor: app.schemas.ContractorUpdate,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
db_contractor = db.query(Contractor).filter(Contractor.id == contractor_id).first()
|
|
if not db_contractor:
|
|
raise HTTPException(status_code=404, detail="Contractor not found")
|
|
|
|
update_data = contractor.dict(exclude_unset=True)
|
|
for key, value in update_data.items():
|
|
setattr(db_contractor, key, value)
|
|
|
|
db.commit()
|
|
db.refresh(db_contractor)
|
|
return db_contractor
|
|
|
|
@router.delete("/{contractor_id}", status_code=status.HTTP_204_NO_CONTENT)
|
|
def delete_contractor(contractor_id: int, db: Session = Depends(get_db)):
|
|
db_contractor = db.query(Contractor).filter(Contractor.id == contractor_id).first()
|
|
if not db_contractor:
|
|
raise HTTPException(status_code=404, detail="Contractor not found")
|
|
|
|
# Optional: instead of hard delete, maybe just deactivate
|
|
db.delete(db_contractor)
|
|
db.commit()
|
|
return None
|