[add]: Loan table extention

app/models/loan.py

@@ -25,6 +25,9 @@ class Loan(db.Model):
     initial_loan_amount = db.Column(db.Float, nullable=False) 
     default_penalty_fee = db.Column(db.Float, default=0)  
     continuous_fee = db.Column(db.Float, default=0)
+    upfront_fee = db.Column(db.Float, nullable=True, default=0.0)
+    repayment_amount = db.Column(db.Float, nullable=True, default=0.0)
+    installment_amount = db.Column(db.Float, nullable=True, default=0.0)
     status = db.Column(db.String(20), default='pending')
     due_date = db.Column(db.DateTime, nullable=True) 
     created_at = db.Column(db.DateTime, default=datetime.now(timezone.utc))
@@ -45,8 +48,20 @@ class Loan(db.Model):
     )
 
     @classmethod
-    def create_loan(cls, customer_id, account_id, offer_id, product_id, initial_loan_amount, collection_type, transaction_id, status='pending'):
-
+    def create_loan(
+        cls,
+        customer_id,
+        account_id,
+        offer_id,
+        product_id,
+        initial_loan_amount,
+        collection_type,
+        transaction_id,
+        upfront_fee,
+        repayment_amount,
+        installment_amount,
+        status="pending",
+    ):
         # Check if customer exists
         customer = Customer.is_valid_customer(customer_id)
         if not customer:
@@ -64,6 +79,9 @@ class Loan(db.Model):
             transaction_id = transaction_id,
             initial_loan_amount = initial_loan_amount,
             current_loan_amount = initial_loan_amount,
+            upfront_fee = upfront_fee,
+            repayment_amount = repayment_amount,
+            installment_amount = installment_amount,
             due_date=now, 
             status = status
         )

app/models/rac_checks.py

@@ -1,12 +1,53 @@
 from datetime import datetime, timezone
 from app.extensions import db
+from sqlalchemy.orm import relationship
+from sqlalchemy.dialects.postgresql import UUID
+from uuid import uuid4
+from sqlalchemy.types import JSON
 
-class RACCheck(Base):
-    __tablename__ = "rac_checks"
+class RACCheck(db.Model):
+    __tablename__ = 'rac_checks'
 
-    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid4)
-    transaction_id = Column(UUID, ForeignKey('transactions.id'), nullable=False)
-    customer_id = Column(String, nullable=False)
-    account_id = Column(String, nullable=False)
-    rac_response = Column(JSON, nullable=False)
-    created_at = Column(DateTime, default=datetime.utcnow)
+    id = db.Column(db.String, primary_key=False)
+    transaction_id = db.Column(db.String(50), nullable=False)
+    customer_id = db.Column(db.String, nullable=False)
+    account_id = db.Column(db.String, nullable=False)
+    rac_response = db.Column(db.JSON, nullable=False)
+    created_at = db.Column(db.DateTime, default=datetime.now(timezone.utc))
+    updated_at = db.Column(db.DateTime, default=datetime.now(timezone.utc), onupdate=datetime.now(timezone.utc))
+
+    @classmethod
+    def get_all_rac_checks(cls):
+        """
+        Return all RAC checks in dictionary format.
+        """
+        rac_checks = cls.query.all()
+
+        if not rac_checks:
+            raise ValueError("No available RAC checks")
+        return rac_checks
+    
+    @classmethod
+    def get_rac_check_by_id(cls, check_id):
+        """
+        Return a RAC check by its ID.
+        """
+        rac_check = cls.query.filter_by(id=check_id).first()
+
+        if not rac_check:
+            raise ValueError(f"RAC Check with ID {check_id} not found")
+        return rac_check
+
+    def to_dict(self):
+        return {
+            "id": str(self.id),
+            "transactionId": str(self.transaction_id),
+            "customerId": self.customer_id,
+            "accountId": self.account_id,
+            "racResponse": self.rac_response,
+            "createdAt": self.created_at.isoformat(),
+            "updatedAt": self.updated_at.isoformat() if self.updated_at else None
+        }
+
+    def __repr__(self):
+        return f'<RACCheck {self.id}>'