115 lines
3.9 KiB
Python
115 lines
3.9 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
import uuid
|
|
|
|
from bd import Settings
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from bd.tables.organization import Organization
|
|
from bd.tables.group import Group
|
|
from route.auth_utils import get_current_user
|
|
|
|
router = APIRouter(tags=["organizations"], prefix="/organizations")
|
|
|
|
|
|
class OrganizationCreate(BaseModel):
|
|
name_organization: str
|
|
group_id: str
|
|
|
|
|
|
class OrganizationUpdate(BaseModel):
|
|
name_organization: Optional[str]
|
|
group_id: Optional[str]
|
|
|
|
|
|
def get_session():
|
|
settings = Settings()
|
|
engine = create_engine(settings.DATABASE_URL_syncpg, future=True)
|
|
return sessionmaker(bind=engine, future=True)
|
|
|
|
|
|
@router.post("/", status_code=201)
|
|
def create_organization(payload: OrganizationCreate, _: object = Depends(get_current_user)):
|
|
Session = get_session()
|
|
try:
|
|
gid = uuid.UUID(payload.group_id)
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid group_id UUID")
|
|
with Session() as session:
|
|
grp = session.get(Group, gid)
|
|
if not grp:
|
|
raise HTTPException(status_code=404, detail="Group not found")
|
|
org = Organization(name_organization=payload.name_organization, group_id=gid)
|
|
session.add(org)
|
|
session.commit()
|
|
session.refresh(org)
|
|
return {"id": str(org.id), "name_organization": org.name_organization, "group_id": str(org.group_id)}
|
|
|
|
|
|
@router.get("/")
|
|
def list_organizations():
|
|
Session = get_session()
|
|
with Session() as session:
|
|
rows = session.query(Organization).all()
|
|
return [{"id": str(o.id), "name_organization": o.name_organization, "group_id": str(o.group_id)} for o in rows]
|
|
|
|
|
|
@router.get("/{org_id}")
|
|
def get_organization(org_id: str):
|
|
Session = get_session()
|
|
try:
|
|
oid = uuid.UUID(org_id)
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid UUID")
|
|
with Session() as session:
|
|
org = session.get(Organization, oid)
|
|
if not org:
|
|
raise HTTPException(status_code=404, detail="Organization not found")
|
|
return {"id": str(org.id), "name_organization": org.name_organization, "group_id": str(org.group_id)}
|
|
|
|
|
|
@router.put("/{org_id}")
|
|
def update_organization(org_id: str, payload: OrganizationUpdate, _: object = Depends(get_current_user)):
|
|
Session = get_session()
|
|
try:
|
|
oid = uuid.UUID(org_id)
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid UUID")
|
|
with Session() as session:
|
|
org = session.get(Organization, oid)
|
|
if not org:
|
|
raise HTTPException(status_code=404, detail="Organization not found")
|
|
if payload.name_organization is not None:
|
|
org.name_organization = payload.name_organization
|
|
if payload.group_id is not None:
|
|
try:
|
|
new_gid = uuid.UUID(payload.group_id)
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid group_id UUID")
|
|
grp = session.get(Group, new_gid)
|
|
if not grp:
|
|
raise HTTPException(status_code=404, detail="Group not found")
|
|
org.group_id = new_gid
|
|
session.add(org)
|
|
session.commit()
|
|
session.refresh(org)
|
|
return {"id": str(org.id), "name_organization": org.name_organization, "group_id": str(org.group_id)}
|
|
|
|
|
|
@router.delete("/{org_id}", status_code=204)
|
|
def delete_organization(org_id: str, _: object = Depends(get_current_user)):
|
|
Session = get_session()
|
|
try:
|
|
oid = uuid.UUID(org_id)
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid UUID")
|
|
with Session() as session:
|
|
org = session.get(Organization, oid)
|
|
if not org:
|
|
raise HTTPException(status_code=404, detail="Organization not found")
|
|
session.delete(org)
|
|
session.commit()
|
|
return {}
|