from app import db
from datetime import datetime
from sqlalchemy import event

class TrenchExcavation(db.Model):
    __tablename__ = "trench_excavation"

    id = db.Column(db.Integer, primary_key=True)
    #  Foreign Key to Subcontractor table
    subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False)
    # Relationship for easy access (subcontractor.subcontractor_name)
    subcontractor = db.relationship("Subcontractor", backref="trench_records")

    # Basic Fields
    Location = db.Column(db.String(500))
    MH_NO = db.Column(db.String(100))
    CC_length = db.Column(db.Float, default=0)
    Invert_Level = db.Column(db.Float, default=0)
    MH_Top_Level = db.Column(db.Float, default=0)
    Ground_Level = db.Column(db.Float, default=0)
    ID_of_MH_m = db.Column(db.Float, default=0)
    Actual_Trench_Length = db.Column(db.Float, default=0)
    Pipe_Dia_mm = db.Column(db.Float, default=0)

    # width
    Width_0_to_2_5 = db.Column(db.Float, default=0)
    Width_2_5_to_3_0 = db.Column(db.Float, default=0)
    Width_3_0_to_4_5 = db.Column(db.Float, default=0)
    Width_4_5_to_6_0 = db.Column(db.Float, default=0)

    Upto_IL_Depth = db.Column(db.Float, default=0)
    Cutting_Depth = db.Column(db.Float, default=0)
    Avg_Depth = db.Column(db.Float, default=0)

    # Excavation categories
    Soft_Murum_0_to_1_5 = db.Column(db.Float, default=0)
    Soft_Murum_1_5_to_3_0 = db.Column(db.Float, default=0)
    Soft_Murum_3_0_to_4_5 = db.Column(db.Float, default=0)

    Hard_Murum_0_to_1_5 = db.Column(db.Float, default=0)
    Hard_Murum_1_5_to_3_0 = db.Column(db.Float, default=0)

    Soft_Rock_0_to_1_5 = db.Column(db.Float, default=0)
    Soft_Rock_1_5_to_3_0 = db.Column(db.Float, default=0)

    Hard_Rock_0_to_1_5 = db.Column(db.Float, default=0)
    Hard_Rock_1_5_to_3_0 = db.Column(db.Float, default=0)
    Hard_Rock_3_0_to_4_5 = db.Column(db.Float, default=0)
    Hard_Rock_4_5_to_6_0 = db.Column(db.Float, default=0)
    Hard_Rock_6_0_to_7_5 = db.Column(db.Float, default=0)

    # Totals
    Soft_Murum_0_to_1_5_total = db.Column(db.Float, default=0)
    Soft_Murum_1_5_to_3_0_total = db.Column(db.Float, default=0)
    Soft_Murum_3_0_to_4_5_total = db.Column(db.Float, default=0)

    Hard_Murum_0_to_1_5_total = db.Column(db.Float, default=0)
    Hard_Murum_1_5_and_above_total = db.Column(db.Float, default=0)

    Soft_Rock_0_to_1_5_total = db.Column(db.Float, default=0)
    Soft_Rock_1_5_and_above_total = db.Column(db.Float, default=0)

    Hard_Rock_0_to_1_5_total = db.Column(db.Float, default=0)
    Hard_Rock_1_5_to_3_0_total = db.Column(db.Float, default=0)
    Hard_Rock_3_0_to_4_5_total = db.Column(db.Float, default=0)
    Hard_Rock_4_5_to_6_0_total = db.Column(db.Float, default=0)
    Hard_Rock_6_0_to_7_5_total = db.Column(db.Float, default=0)

    Total = db.Column(db.Float, default=0)
    Remarks = db.Column(db.String(500))
    RA_Bill_No=db.Column(db.String(500))
    
    created_at = db.Column(db.DateTime, default=datetime.today)


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

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

        def safe(val):
            return val or 0

        return {
            "Soft_Murum_Total": (
                safe(self.Soft_Murum_0_to_1_5)
                + safe(self.Soft_Murum_1_5_to_3_0)
                + safe(self.Soft_Murum_3_0_to_4_5)
            ),

            "Hard_Murum_Total": (
                safe(self.Hard_Murum_0_to_1_5)
                + safe(self.Hard_Murum_1_5_to_3_0)
            ),

            "Soft_Rock_Total": (
                safe(self.Soft_Rock_0_to_1_5)
                + safe(self.Soft_Rock_1_5_to_3_0)
            ),

            "Hard_Rock_Total": (
                safe(self.Hard_Rock_0_to_1_5)
                + safe(self.Hard_Rock_1_5_to_3_0)
                + safe(self.Hard_Rock_3_0_to_4_5)
                + safe(self.Hard_Rock_4_5_to_6_0)
                + safe(self.Hard_Rock_6_0_to_7_5)
            ),
        }

# ==========================================
# AUTO CALCULATE GRAND TOTAL
# ==========================================
def calculate_trench_total(mapper, connection, target):
    total = 0
    for column in target.__table__.columns:
        if column.name.endswith("_total"):
            total += getattr(target, column.name) or 0
    target.Total = total


event.listen(TrenchExcavation, "before_insert", calculate_trench_total)
event.listen(TrenchExcavation, "before_update", calculate_trench_total)