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

class ManholeDomesticChamberClient(db.Model):
    __tablename__ = "mh_dc_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))
    MH_TOP_LEVEL = db.Column(db.Float, default=0)
    MH_IL_LEVEL = db.Column(db.Float, default=0)
    Depth_of_MH = db.Column(db.Float, default=0)
   
    # Excavation categories 
    d_0_to_1_5 = db.Column(db.Float, default=0)
    d_1_5_to_2_0 = db.Column(db.Float, default=0)
    d_2_0_to_2_5 = db.Column(db.Float, default=0)
    d_2_5_to_3_0 = db.Column(db.Float, default=0)
    d_3_0_to_3_5 = db.Column(db.Float, default=0)
    d_3_5_to_4_0 = db.Column(db.Float, default=0)
    d_4_0_to_4_5= db.Column(db.Float, default=0)
    d_4_5_to_5_0 = db.Column(db.Float, default=0)
    d_5_0_to_5_5 = db.Column(db.Float, default=0)
    d_5_5_to_6_0 = db.Column(db.Float, default=0)
    d_6_0_to_6_5 = db.Column(db.Float, default=0)

    Domestic_Chambers = db.Column(db.Float, default=0)

    Total = db.Column(db.Float, default=0)
    created_at = db.Column(db.DateTime, default=datetime.today)

    def __repr__(self):
        return f"<HanholeDomesticChamberConstruction {self.Location}>"


    def serialize(self):
        return {c.name: getattr(self, c.name) for c in self.__table__.columns}


# AUTO TOTAL USING REGEX
def calculate_mh_dc_total(mapper, connection, target):
    total = 0
    for column in target.__table__.columns:
        if RegularExpression.D_RANGE_PATTERN.match(column.name):
            total += getattr(target, column.name) or 0
    target.Total = total


event.listen(ManholeDomesticChamberClient, "before_insert", calculate_mh_dc_total)
event.listen(ManholeDomesticChamberClient, "before_update", calculate_mh_dc_total)