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

class LayingClient(db.Model):
    __tablename__ = "laying_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"))
    # Bedding Qty.
    Outer_dia_of_MH_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Bedding_Length = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Width = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Depth = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Qty = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
 
    #  PIPE LAYING Qty. 
    Pipe_Dia_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    ID_of_MH_m = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    Laying_Length = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))

    pipe_150_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_200_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_250_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_300_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_350_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_400_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_450_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_500_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_600_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_700_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_900_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    pipe_1200_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    
    np4_pipe_200_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_250_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_300_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_350_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_400_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_450_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_500_mm = db.Column(db.Numeric(10, 2), default=Decimal("0.00"))
    np4_pipe_600_mm = 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"<LayingModel {self.Location}>"

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


# AUTO TOTAL USING REGEX
def calculate_laying_total(mapper, connection, target):
    total = Decimal("0.00")
    for column in target.__table__.columns:
        if RegularExpression.PIPE_MM_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(LayingClient, "before_insert", calculate_laying_total)
event.listen(LayingClient, "before_update", calculate_laying_total)