31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
from fastapi import APIRouter, HTTPException, Depends, status
|
|
from sqlmodel import Session, select
|
|
from typing import List
|
|
|
|
from ..model.models import GroupDB, GroupResponse, GroupCreate
|
|
from ..services.database import engine, get_session, add_and_refresh
|
|
from ..services.auth import auth_is_admin
|
|
|
|
group_router = APIRouter(prefix="/groups", tags=["Group"])
|
|
|
|
@group_router.get("/", response_model=List[GroupResponse])
|
|
def get_groups(*, db: Session = Depends(get_session), admin: bool = Depends(auth_is_admin)):
|
|
groups = db.exec(select(GroupDB)).all()
|
|
return groups
|
|
|
|
@group_router.post("/", response_model=GroupResponse)
|
|
def create_group(*, db: Session = Depends(get_session), group: GroupCreate, admin: bool = Depends(auth_is_admin)):
|
|
db_group = GroupDB.model_validate(group)
|
|
group = db.exec(select(GroupDB).where(GroupDB.name == db_group.name)).first()
|
|
if group is not None:
|
|
raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Group already exists!")
|
|
return add_and_refresh(db, db_group)
|
|
|
|
@group_router.delete("/{group_id}")
|
|
def delete_group(*, db: Session = Depends(get_session), group_id: int, admin: bool = Depends(auth_is_admin)):
|
|
db_group = db.get(GroupDB, group_id)
|
|
if db_group is None:
|
|
raise HTTPException(status_code=404, detail="Group not found")
|
|
db.delete(db_group)
|
|
db.commit()
|
|
return {"message": "Group deleted successfully"} |