from uuid import UUID

from app.config import TaskEnv
from app.models.bill_delay import BillDelayModel
from core.repositories.base import BaseRepository


class BillDelayRepository(BaseRepository):
    def __init__(self):
        super().__init__(collection=TaskEnv.CLN_BILL_DELAY, model=BillDelayModel)

    def create_sum(self, coding: BillDelayModel) -> BillDelayModel:
        all_total = 0
        if coding.financial:
            total = 0
            if coding.financial.status:
                total += coding.financial.status
            if coding.financial.prepayment:
                total += coding.financial.prepayment
            if coding.financial.modification:
                total += coding.financial.modification
            if coding.financial.good_job:
                total += coding.financial.good_job
            coding.financial.total = total
            all_total += total
        if coding.force_major:
            total = 0
            if coding.force_major.force_event:
                total += coding.force_major.force_event
            if coding.force_major.material:
                total += coding.force_major.material
            if coding.force_major.rules:
                total += coding.force_major.rules
            if coding.force_major.define_indicators:
                total += coding.force_major.define_indicators
            coding.force_major.total = total
            all_total += total

        if coding.technical:
            total = 0
            if coding.technical.map_technical:
                total += coding.technical.map_technical
            if coding.technical.permit:
                total += coding.technical.permit
            if coding.technical.purchase:
                total += coding.technical.purchase
            if coding.technical.suspense:
                total += coding.technical.suspense
            coding.technical.total = total
            all_total += total
        if coding.extra_new_work:
            all_total += coding.extra_new_work
        if coding.other:
            all_total += coding.other
        if all_total:
            coding.total_day = all_total
        return coding
    async def create(self, coding: BillDelayModel):
        coding = self.create_sum(coding)
        return await super().create(coding)
    async def update(self, _id: UUID, update: BillDelayModel):
        update = self.create_sum(update)
        return await super().update(_id, update)