change of comparison report of 4 model update

app/models/mh_dc_client_model.py

@@ -1,5 +1,10 @@
 from app import db
 from datetime import datetime
+from sqlalchemy import event
+import re
+
+# REGEX PATTERN
+D_RANGE_PATTERN = re.compile(r"^d_\d+(?:_\d+)?_to_\d+(?:_\d+)?$")
 
 class ManholeDomesticChamberClient(db.Model):
     __tablename__ = "mh_dc_client"
@@ -24,18 +29,17 @@ class ManholeDomesticChamberClient(db.Model):
     d_1_5_to_2_0 = db.Column(db.Float)
     d_2_0_to_2_5 = db.Column(db.Float)
     d_2_5_to_3_0 = db.Column(db.Float)
-
     d_3_0_to_3_5 = db.Column(db.Float)
     d_3_5_to_4_0 = db.Column(db.Float)
     d_4_0_to_4_5= db.Column(db.Float)
     d_4_5_to_5_0 = db.Column(db.Float)
-
     d_5_0_to_5_5 = db.Column(db.Float)
     d_5_5_to_6_0 = db.Column(db.Float)
     d_6_0_to_6_5 = db.Column(db.Float)
 
     Domestic_Chambers = db.Column(db.Float)
 
+    Total = db.Column(db.Float)
     created_at = db.Column(db.DateTime, default=datetime.today)
 
     def __repr__(self):
@@ -54,4 +58,20 @@ class ManholeDomesticChamberClient(db.Model):
             "d_4_66_to_5_15", "d_5_16_to_5_65", "d_5_66_to_6_15",
             "d_6_16_to_6_65", "d_6_66_to_7_15", "d_7_16_to_7_65",
             "d_7_66_to_8_15", "d_8_16_to_8_65", "d_8_66_to_9_15",
-            "d_9_16_to_9_65" ]
+            "d_9_16_to_9_65" ]
+    
+# ===============================
+# AUTO TOTAL USING REGEX
+# ===============================
+def calculate_mh_dc_total(mapper, connection, target):
+    total = 0
+
+    for column in target.__table__.columns:
+        if 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)