from fastapi import APIRouter, Depends
from typing import Annotated, List
from sqlalchemy.orm import Session

from app.api.commercial_equipment import schemas
from app.api.commercial_equipment.schemas import CommercialEquipmentCreate
from app.api.commercial_equipment.service import CommercialEquipmentService
from app.database.main.mysql import get_db
from app.dependency.authantication import JWTPayloadSchema, get_current_student

commercial_equipment_router = APIRouter()

@commercial_equipment_router.post("/commercial_equipment_create", response_model_exclude_none=True)
async def create_commercial_equipment(
    request: CommercialEquipmentCreate,
    db: Annotated[Session, Depends(get_db)],
    token: Annotated[JWTPayloadSchema, Depends(get_current_student)],
):
    return await CommercialEquipmentService(db, token).create_commercial_equipment(request)

@commercial_equipment_router.get("/commercial_equipment_response/{group_id}/total_potential")
async def get_by_group(group_id: int,db: Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student),):
    return await CommercialEquipmentService(db, token).get_commercial_equipment(group_id)

@commercial_equipment_router.put("/update_commercial_equipment", response_model_exclude_none=True)
async def update_commercial_equipment(request:List[schemas.CommercialEquipmentUpdate], db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await CommercialEquipmentService(db,token).updated(request)

@commercial_equipment_router.delete("/delete_commercial_equipment", response_model_exclude_none=True)
async def delete_commercial_equipment(equipment_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await CommercialEquipmentService(db,token).deleted(equipment_id)