38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from fastapi import APIRouter, HTTPException, Depends
|
|
from sqlmodel import Session, select
|
|
from typing import List
|
|
|
|
from ..model.models import GroupDB, GroupResponse, GroupCreate
|
|
from ..services.database import engine
|
|
|
|
group_router = APIRouter(prefix="/groups", tags=["Group"])
|
|
|
|
def get_session():
|
|
with Session(engine) as db:
|
|
yield db
|
|
|
|
def add_and_refresh(db: Session, obj):
|
|
db.add(obj)
|
|
db.commit()
|
|
db.refresh(obj)
|
|
return obj
|
|
|
|
@group_router.get("/", response_model=List[GroupResponse])
|
|
def get_groups(*, db: Session = Depends(get_session)):
|
|
groups = db.exec(select(GroupDB)).all()
|
|
return groups
|
|
|
|
@group_router.post("/", response_model=GroupResponse)
|
|
def create_group(*, db: Session = Depends(get_session), group: GroupCreate):
|
|
db_group = GroupDB.model_validate(group)
|
|
add_and_refresh(db, db_group)
|
|
return db_group
|
|
|
|
@group_router.delete("/{group_id}")
|
|
def delete_group(*, db: Session = Depends(get_session), group_id: int):
|
|
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"} |