from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from typing import Annotated
from app.api.plumbing.schemas import plumbingCreate
from app.api.plumbing.service import PlumlingService
from app.database.main.mysql import get_db
from app.dependency.authantication import JWTPayloadSchema, get_current_student

plumbing_router = APIRouter()

@plumbing_router.post("/Plumbing_creat", response_model_exclude_none=True)
async def create_plumbing(request: plumbingCreate,db: Annotated[Session, Depends(get_db)],token: Annotated[JWTPayloadSchema, Depends(get_current_student)]):
    return await PlumlingService(db, token).create_plumbing(request)

@plumbing_router.get("/Plumbing_creat/{group_id}")
async def get_total_consumption_by_group(group_id: int,db: Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student),):
    return await PlumlingService(db, token).get_plumbing(group_id)