[update]: Loan Status

app/models/loan.py

@@ -16,10 +16,16 @@ class Loan(db.Model):
         autoincrement=True,
     )
     customer_id = db.Column(db.String(50), nullable=False)
+    transaction_id = db.Column(db.String(50), nullable=True)
     account_id = db.Column(db.String(50), nullable=False)
     offer_id = db.Column(db.String(20), nullable=False)
-    principal_amount = db.Column(db.Float, nullable=False)
+    collection_type = db.Column(db.String(20), nullable=True) 
+    current_loan_amount = db.Column(db.Float, nullable=True)  
+    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)
     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))
     updated_at = db.Column(db.DateTime, default=datetime.now(timezone.utc), onupdate=datetime.now(timezone.utc))
 
@@ -31,26 +37,26 @@ class Loan(db.Model):
     )
 
     @classmethod
-    def create_loan(cls, customer_id, account_id, offer_id, principal_amount, status='pending'):
+    def create_loan(cls, customer_id, account_id, offer_id, initial_loan_amount, collection_type, transaction_id, status='pending'):
 
         # Check if customer exists
         is_valid = Customer.is_valid_customer(customer_id)
         if not is_valid:
             raise ValueError("Customer does not exist")
-        
-        # # Check for active loans
-        # has_active_loans = cls.has_active_loans(customer_id)
-        # if has_active_loans:
-        #     raise ValueError("Customer has active loans")
 
+        now = datetime.now(timezone.utc)
 
         # Create and save the loan
         loan = cls(
-            customer_id=customer_id,
-            account_id=account_id,
-            offer_id=offer_id,
-            principal_amount=principal_amount,
-            status=status
+            customer_id = customer_id,
+            account_id = account_id,
+            offer_id = offer_id,
+            collection_type = collection_type,
+            transaction_id = transaction_id,
+            initial_loan_amount = initial_loan_amount,
+            current_loan_amount = initial_loan_amount,
+            due_date=now, 
+            status = status
         )
 
         try:
@@ -104,14 +110,15 @@ class Loan(db.Model):
         Convert the Loan object to a dictionary format for JSON serialization.
         """
         return {
-            'id': self.id,
-            'customer_id': self.customer_id,
-            'account_id': self.account_id,
-            'offer_id': self.offer_id,
-            'principal_amount': self.principal_amount,
+            'debtId': self.id,
+            'initialLoanAmount': self.initial_loan_amount,
+            'currentLoanAmount': self.current_loan_amount,
+            'defaultPenaltyFee': self.default_penalty_fee,
+            'continuousFee': self.continuous_fee,
+            'collectionType': self.collection_type,
             'status': self.status,
-            'created_at': self.created_at.isoformat() if self.created_at else None,
-            'updated_at': self.updated_at.isoformat() if self.updated_at else None
+            'dueDate': self.due_date.isoformat() if self.due_date else None,
+            'loanDate': self.created_at.isoformat() if self.created_at else None,
         }
     
     def __repr__(self):