from typing import List
from fastapi import HTTPException
from sqlalchemy.orm import Session
from app.api.margin_and_sales_contribution_analysis import schema
from app.dependency.authantication import JWTPayloadSchema
from app.models.main.margin_and_sales_contribution_analysis import MarginandSalesContributionAnalysisBase, TblMarginandSalesContributionAnalysis
from app.utils.schemas_utils import CustomResponse
from app.locale.messages import Messages

class MarginandSalesContributionAnalysisService:
    def __init__(self, db:Session, token:JWTPayloadSchema):
        self.db = db
        self.token = token
        
    async def created(self, request:schema.MarginandSalesContributionAnalysisCreate):
        created_data = MarginandSalesContributionAnalysisBase.model_validate(request.model_dump())
        TblMarginandSalesContributionAnalysis.create(created_data, self.db)
        self.db.commit()
        return CustomResponse(status="1", message=Messages.MARGIN_AND_SALES)
    
    async def geted(self, group_id:int):
        geted_data = TblMarginandSalesContributionAnalysis.get(group_id, self.db)
        if not geted_data:
            raise HTTPException(status_code=404, detail="Get group ID not found")
        return [schema.MarginandSalesContributionAnalysisResponse.model_validate(get_group) for get_group in geted_data]
    
    async def updated(self, request:List[schema.MarginandSalesContributionAnalysisUpdate]):
        for req in request:
            updated_data = MarginandSalesContributionAnalysisBase.model_validate(req.model_dump())
            if updated_data.margin_id is None:
                return CustomResponse(status="-1", message=Messages.MARGIN_AND_SALES_NOT)
            TblMarginandSalesContributionAnalysis.update(updated_data.margin_id, updated_data, self.db)
        self.db.commit()
        return CustomResponse(status="1", message=Messages.MARGIN_AND_SALES_UPDATE)