from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from app.api.utilitys import schema
from app.api.utilitys.service import UtilityService
from app.database.main.mysql import get_db
from app.dependency.authantication import JWTPayloadSchema, get_current_student

utility_router = APIRouter()

@utility_router.post("/create_utility", response_model_exclude_none=True)
async def create_utility(request:schema.UtilityCreate, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await UtilityService(db,token).create_utility(request)


@utility_router.get("/get_utility{group_id}", response_model_exclude_none=True)
async def get_utility(group_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await UtilityService(db,token).get_utility(group_id)

@utility_router.get("/get_total_cost", response_model_exclude_none=True)
async def get_sum_total_cost(db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await UtilityService(db,token).get_total_cost()

@utility_router.put("/update_utility", response_model_exclude_none=True)
async def update_utility(request:schema.UtilityUpdate, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await UtilityService(db,token).update_utility(request)

@utility_router.delete("/delete_utility", response_model_exclude_none=True)
async def delete_utility(group_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await UtilityService(db,token).delete_utility(group_id)