from app import db
from datetime import datetime

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)
    Invert_Level = db.Column(db.Float)
    MH_Top_Level = db.Column(db.Float)
    Ground_Level = db.Column(db.Float)
    ID_of_MH_m = db.Column(db.Float)
    Actual_Trench_Length = db.Column(db.Float)
    Pipe_Dia_mm = db.Column(db.Float)

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

    Upto_IL_Depth = db.Column(db.Float)
    Cutting_Depth = db.Column(db.Float)
    Avg_Depth = db.Column(db.Float)

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

    Hard_Murum_0_to_1_5 = db.Column(db.Float)
    Hard_Murum_1_5_to_3_0 = db.Column(db.Float)

    Soft_Rock_0_to_1_5 = db.Column(db.Float)
    Soft_Rock_1_5_to_3_0 = db.Column(db.Float)

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

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

    Hard_Murum_0_to_1_5_total = db.Column(db.Float)
    Hard_Murum_1_5_and_above_total = db.Column(db.Float)

    Soft_Rock_0_to_1_5_total = db.Column(db.Float)
    Soft_Rock_1_5_and_above_total = db.Column(db.Float)

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

    Total = db.Column(db.Float)
    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 {
    #         "id": self.id,
    #         "subcontractor_id": self.subcontractor_id,
    #         "subcontractor_name": self.subcontractor.subcontractor_name if self.subcontractor else None,

    #         "Location": self.Location,
    #         "MH_NO": self.MH_NO,
    #         "CC_length": self.CC_length,
    #         "Invert_Level": self.Invert_Level,
    #         "MH_Top_Level": self.MH_Top_Level,
    #         "Ground_Level": self.Ground_Level,
    #         "ID_of_MH_m": self.ID_of_MH_m,
    #         "Actual_Trench_Length": self.Actual_Trench_Length,
    #         "Pipe_Dia_mm": self.Pipe_Dia_mm,

    #         # Width
    #         "Width_0_to_2_5": self.Width_0_to_2_5,
    #         "Width_2_5_to_3_0": self.Width_2_5_to_3_0,
    #         "Width_3_0_to_4_5": self.Width_3_0_to_4_5,
    #         "Width_4_5_to_6_0": self.Width_4_5_to_6_0,

    #         # Depth
    #         "Upto_IL_Depth": self.Upto_IL_Depth,
    #         "Cutting_Depth": self.Cutting_Depth,
    #         "Avg_Depth": self.Avg_Depth,

    #         # Soft Murum
    #         "Soft_Murum_0_to_1_5": self.Soft_Murum_0_to_1_5,
    #         "Soft_Murum_1_5_to_3_0": self.Soft_Murum_1_5_to_3_0,
    #         "Soft_Murum_3_0_to_4_5": self.Soft_Murum_3_0_to_4_5,

    #         # Hard Murum
    #         "Hard_Murum_0_to_1_5": self.Hard_Murum_0_to_1_5,
    #         "Hard_Murum_1_5_to_3_0": self.Hard_Murum_1_5_to_3_0,

    #         # Soft Rock
    #         "Soft_Rock_0_to_1_5": self.Soft_Rock_0_to_1_5,
    #         "Soft_Rock_1_5_to_3_0": self.Soft_Rock_1_5_to_3_0,

    #         # Hard Rock
    #         "Hard_Rock_0_to_1_5": self.Hard_Rock_0_to_1_5,
    #         "Hard_Rock_1_5_to_3_0": self.Hard_Rock_1_5_to_3_0,
    #         "Hard_Rock_3_0_to_4_5": self.Hard_Rock_3_0_to_4_5,
    #         "Hard_Rock_4_5_to_6_0": self.Hard_Rock_4_5_to_6_0,
    #         "Hard_Rock_6_0_to_7_5": self.Hard_Rock_6_0_to_7_5,

    #         # Totals
    #         "Soft_Murum_0_to_1_5_total": self.Soft_Murum_0_to_1_5_total,
    #         "Soft_Murum_1_5_to_3_0_total": self.Soft_Murum_1_5_to_3_0_total,
    #         "Soft_Murum_3_0_to_4_5_total": self.Soft_Murum_3_0_to_4_5_total,

    #         "Hard_Murum_0_to_1_5_total": self.Hard_Murum_0_to_1_5_total,
    #         "Hard_Murum_1_5_and_above_total": self.Hard_Murum_1_5_and_above_total,

    #         "Soft_Rock_0_to_1_5_total": self.Soft_Rock_0_to_1_5_total,
    #         "Soft_Rock_1_5_and_above_total": self.Soft_Rock_1_5_and_above_total,

    #         "Hard_Rock_0_to_1_5_total": self.Hard_Rock_0_to_1_5_total,
    #         "Hard_Rock_1_5_to_3_0_total": self.Hard_Rock_1_5_to_3_0_total,
    #         "Hard_Rock_3_0_to_4_5_total": self.Hard_Rock_3_0_to_4_5_total,
    #         "Hard_Rock_4_5_to_6_0_total": self.Hard_Rock_4_5_to_6_0_total,
    #         "Hard_Rock_6_0_to_7_5_total": self.Hard_Rock_6_0_to_7_5_total,

    #         "Total": self.Total,
    #         "Remarks": self.Remarks,
    #         "RA_Bill_No": self.RA_Bill_No,

    #         "created_at": self.created_at.strftime("%d-%m-%Y") if self.created_at else None
    #     }

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