Files
Comparison_Project/app/models/manhole_domestic_chamber_model.py

65 lines
2.6 KiB
Python

from app import db
from datetime import datetime
from sqlalchemy import event
from app.utils.regex_utils import RegularExpression
class ManholeDomesticChamber(db.Model):
__tablename__ = "manhole_domestic_chamber"
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="manhole_domestic_chamber_records")
# Basic Fields
Location = db.Column(db.String(500))
MH_NO = db.Column(db.String(100))
Depth_of_MH = db.Column(db.Float, default=0)
# Excavation categories
d_0_to_0_75 = db.Column(db.Float, default=0)
d_0_76_to_1_05 = db.Column(db.Float, default=0)
d_1_06_to_1_65 = db.Column(db.Float, default=0)
d_1_66_to_2_15 = db.Column(db.Float, default=0)
d_2_16_to_2_65 = db.Column(db.Float, default=0)
d_2_66_to_3_15 = db.Column(db.Float, default=0)
d_3_16_to_3_65= db.Column(db.Float, default=0)
d_3_66_to_4_15 = db.Column(db.Float, default=0)
d_4_16_to_4_65 = db.Column(db.Float, default=0)
d_4_66_to_5_15 = db.Column(db.Float, default=0)
d_5_16_to_5_65 = db.Column(db.Float, default=0)
d_5_66_to_6_15 = db.Column(db.Float, default=0)
d_6_16_to_6_65 = db.Column(db.Float, default=0)
d_6_66_to_7_15 = db.Column(db.Float, default=0)
d_7_16_to_7_65 = db.Column(db.Float, default=0)
d_7_66_to_8_15 = db.Column(db.Float, default=0)
d_8_16_to_8_65 = db.Column(db.Float, default=0)
d_8_66_to_9_15 = db.Column(db.Float, default=0)
d_9_16_to_9_65 = db.Column(db.Float, default=0)
Domestic_Chambers = db.Column(db.Float, default=0)
DWC_Pipe_Length = db.Column(db.Float, default=0)
UPVC_Pipe_Length = db.Column(db.Float, default=0)
RA_Bill_No=db.Column(db.String(500))
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(ManholeDomesticChamber, "before_insert", calculate_mh_dc_total)
event.listen(ManholeDomesticChamber, "before_update", calculate_mh_dc_total)