from app import db
from datetime import datetime
from sqlalchemy import event
from app.utils.regex_utils import RegularExpression
from decimal import Decimal


class TrenchExcavationClient(db.Model):
    __tablename__ = "tr_ex_client"

    id = db.Column(db.Integer, primary_key=True)

    # Basic Fields
    RA_Bill_No=db.Column(db.String(500))
    Location = db.Column(db.String(500))
    MH_NO = db.Column(db.String(100))
    CC_length = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Actual_Trench_Length = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Ground_Level = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Invert_Level = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Excavated_level = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Cutting_Depth = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Avg_Depth = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Pipe_Dia_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
 
    # width
    Width_0_to_1_5_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Width_1_5_to_3_0_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Width_3_0_to_4_5_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Width_4_5_to_6_0_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Width_6_0_to_7_5_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    # Excavation categories
    Marshi_Muddy_Slushy_0_to_1_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Marshi_Muddy_Slushy_1_5_to_3_0 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Marshi_Muddy_Slushy_3_0_to_4_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Soft_Murum_0_to_1_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Murum_1_5_to_3_0 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Murum_3_0_to_4_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Hard_Murum_0_to_1_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Murum_1_5_to_3_0 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Murum_3_0_to_4_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Soft_Rock_0_to_1_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Rock_1_5_to_3_0 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Rock_3_0_to_4_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Hard_Rock_0_to_1_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_1_5_to_3_0 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_3_0_to_4_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_4_5_to_6_0 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_6_0_to_7_5 = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    # Totals
    Marshi_Muddy_Slushy_0_to_1_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Marshi_Muddy_Slushy_1_5_to_3_0_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Marshi_Muddy_Slushy_3_0_to_4_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Soft_Murum_0_to_1_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Murum_1_5_to_3_0_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Murum_3_0_to_4_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Hard_Murum_0_to_1_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Murum_1_5_to_3_0_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Murum_3_0_to_4_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Soft_Rock_0_to_1_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Rock_1_5_to_3_0_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Soft_Rock_3_0_to_4_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Hard_Rock_0_to_1_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_1_5_to_3_0_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_3_0_to_4_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_4_5_to_6_0_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Hard_Rock_6_0_to_7_5_total = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    Total = db.Column(db.Numeric(12, 2), default=Decimal("0.00"))
    Remarks = db.Column(db.String(500), default="Import File")
    created_at = db.Column(db.DateTime,default=datetime.now,nullable=False)

    def __repr__(self):
        return f"<TrenchExcavation {self.Location}>"
    
    def serialize(self):
            return {c.name: getattr(self, c.name) for c in self.__table__.columns}


# AUTO TOTAL USING REGEX
def calculate_trench_client_total(mapper, connection, target):
    total = Decimal("0.00")
    for column in target.__table__.columns:
        if RegularExpression.STR_TOTAL_PATTERN.match(column.name):
            value = getattr(target, column.name)
            if value is not None:
                total += Decimal(value)
    target.Total = total.quantize(Decimal("0.01"))

event.listen(TrenchExcavationClient, "before_insert", calculate_trench_client_total)
event.listen(TrenchExcavationClient, "before_update", calculate_trench_client_total)