from typing import List
from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from app.api.gross_margin_contribution import service
from app.api.gross_margin_contribution.schema import GrossMarginContributionCreate, GrossMarginContributionUpdate
from app.database.main.mysql import get_db
from app.dependency.authantication import JWTPayloadSchema, get_current_student
from app.models.main.gross_margin_contributions import GrossMarginContributionBase

gross_margin_contribution_router = APIRouter()

@gross_margin_contribution_router.post("/create_gross_margin_contribution", response_model_exclude_none=True)
async def create_gross_margin_contribution(request:GrossMarginContributionCreate, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await service.GrossMarginContributionService(db,token).create_gross_margin(request)

@gross_margin_contribution_router.get("/get_gross_margin_contribution", response_model_exclude_none=True)
async def get_gross_margin_contribution(gross_margin_get_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await service.GrossMarginContributionService(db,token).get_gross_margin_contribution(gross_margin_get_id)

@gross_margin_contribution_router.get("/get_group_gross_margin_contribution", response_model_exclude_none=True)
async def get_group_gross_margin_contribution(group_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await service.GrossMarginContributionService(db,token).get_group_gross_margin_contribution(group_id)

@gross_margin_contribution_router.put("/update_gross_margin_contribution", response_model_exclude_none=True)
async def update_gross_margin_contribution(request:List[GrossMarginContributionUpdate], db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await service.GrossMarginContributionService(db,token).update_gross_margin_contribution_id(request)

@gross_margin_contribution_router.delete("/delete_gross_margin_contribution", response_model_exclude_none=True)
async def delete_gross_margin_contribution(gross_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await service.GrossMarginContributionService(db,token).delete_gross_margin_contribution_id(gross_id)

@gross_margin_contribution_router.get("get_excel", response_model_exclude_none=True)
async def get_excel(group_id:int, db:Session = Depends(get_db),token: JWTPayloadSchema = Depends(get_current_student)):
    return await service.GrossMarginContributionService(db,token).get_excel(group_id)