[add]: ref_id and ref_model for transactions. And db session and rollback for transactions

app/models/account.py

@@ -31,9 +31,7 @@ class Account(db.Model):
 
         try:
             db.session.add(account)
-            db.session.commit()
         except IntegrityError as err:
-            db.session.rollback()
             raise ValueError(f"Database integrity error: {err}")
         return account
 

app/models/customer.py

@@ -44,9 +44,7 @@ class Customer(db.Model):
                 account_type=account_type
             )
             
-            db.session.commit()
         except IntegrityError as err:
-            db.session.rollback()
             raise ValueError(f"Database integrity error: {err}")
         return customer
 

app/models/loan.py

@@ -47,9 +47,7 @@ class Loan(db.Model):
 
         try:
             db.session.add(loan)
-            db.session.commit()
         except IntegrityError as err:
-            db.session.rollback()
             raise ValueError(f"Database integrity error: {err}")
         return loan
 
@@ -92,8 +90,7 @@ class Loan(db.Model):
         
         # Update loan status and the updated_at timestamp
         loan.status = status
-        
-        db.session.commit()
+
     
     def __repr__(self):
         return f'<Loan {self.id}>'

app/models/repayment.py

@@ -44,9 +44,7 @@ class Repayment(db.Model):
 
         try:
             db.session.add(repayment)
-            db.session.commit()
         except IntegrityError as err:
-            db.session.rollback()
             raise ValueError(f"Database integrity error: {err}")
 
         return repayment

app/models/transaction.py

@@ -10,9 +10,9 @@ class Transaction(db.Model):
         primary_key=True,
         autoincrement=True,
     )
-    #id = db.Column(db.Int, primary_key=True)
     transaction_id = db.Column(db.String(50), nullable=False)
-    account_id = db.Column(db.String(50), nullable=False)
+    ref_id = db.Column(db.String(50), nullable=False)
+    ref_model = db.Column(db.String(50), nullable=True, default='account')
     type = db.Column(db.String(50), nullable=False)
     channel = db.Column(db.String(50), nullable=False)
     created_at = db.Column(db.DateTime, default=datetime.now(timezone.utc))
@@ -22,7 +22,7 @@ class Transaction(db.Model):
         return f'<Transaction {self.id}>'
 
     @classmethod
-    def create_transaction(cls, transaction_id, account_id, type, channel):
+    def create_transaction(cls, transaction_id, ref_id, ref_model, type, channel):
 
         # if cls.query.filter_by(transaction_id=transaction_id).first():
         #     raise ValueError("Duplicate Transaction")
@@ -33,17 +33,16 @@ class Transaction(db.Model):
 
 
         transaction = cls(
-            transaction_id=transaction_id,
-            account_id=account_id,
-            type=type,
-            channel=channel
+            transaction_id = transaction_id,
+            ref_id = ref_id,
+            ref_model = ref_model,
+            type = type,
+            channel = channel
         )
 
         try:
             db.session.add(transaction)
-            db.session.commit()
         except IntegrityError as err:
-            db.session.rollback()
             raise ValueError(f"Database integrity error: {err}")
 
         return transaction