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

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)