model class update total cal fields all model

app/models/manhole_excavation_model.py

@@ -1,5 +1,6 @@
 from app import db
 from datetime import datetime
+from sqlalchemy import event
 
 class ManholeExcavation(db.Model):
     __tablename__ = "manhole_excavation"
@@ -65,3 +66,17 @@ class ManholeExcavation(db.Model):
 
     def serialize(self):
         return {c.name: getattr(self, c.name) for c in self.__table__.columns}
+
+# ==========================================
+# 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)

app/models/mh_ex_client_model.py

@@ -1,5 +1,6 @@
 from app import db
 from datetime import datetime
+from sqlalchemy import event
 
 class ManholeExcavationClient(db.Model):
     __tablename__ = "mh_ex_client"
@@ -81,3 +82,18 @@ class ManholeExcavationClient(db.Model):
 
     def serialize(self):
         return {c.name: getattr(self, c.name) for c in self.__table__.columns}
+
+
+# ==========================================
+# 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)

app/models/tr_ex_client_model.py

@@ -1,14 +1,11 @@
 from app import db
 from datetime import datetime
+from sqlalchemy import event
 
 class TrenchExcavationClient(db.Model):
     __tablename__ = "tr_ex_client"
 
     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="tr_ex_records")
 
     # Basic Fields
     RA_Bill_No=db.Column(db.String(500))
@@ -86,3 +83,19 @@ class TrenchExcavationClient(db.Model):
     
     def serialize(self):
             return {c.name: getattr(self, c.name) for c in self.__table__.columns}
+
+
+# ==========================================
+# AUTO CALCULATE GRAND TOTAL
+# ==========================================
+
+def calculate_trench_client_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(TrenchExcavationClient, "before_insert", calculate_trench_client_total)
+event.listen(TrenchExcavationClient, "before_update", calculate_trench_client_total)

app/models/trench_excavation_model.py

@@ -1,5 +1,6 @@
 from app import db
 from datetime import datetime
+from sqlalchemy import event
 
 class TrenchExcavation(db.Model):
     __tablename__ = "trench_excavation"
@@ -108,3 +109,17 @@ class TrenchExcavation(db.Model):
                 + 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)