[add]: account_id to loan_status and repayment

app/api/services/repayment.py

@@ -2,6 +2,7 @@ from flask import request, jsonify
 from marshmallow import ValidationError
 from app.api.enums.loan_status import LoanStatus
 from app.models import Repayment
+from app.models.customer import Customer
 from app.models.loan import Loan
 from app.utils.logger import logger
 from app.api.schemas.repayment import RepaymentSchema
@@ -27,42 +28,52 @@ class RepaymentService(BaseService):
         try:
             with db.session.begin():
                 validated_data = RepaymentService.validate_data(data, RepaymentSchema())
+                account_id = validated_data.get('accountId')
                 customer_id =  validated_data.get('customerId')
                 request_id =  validated_data.get('requestId')
                 loan_id = validated_data.get('debtId')
                 product_id = validated_data.get('productId')
+                customer = Customer.get_customer(customer_id)
+
+                if not customer:
+                    return jsonify({
+                        "message": "Customer not found."
+                    }), 404
 
     
-        
-            # Save the repayment details  
-                repayment = Repayment.create_repayment(
-                    customer_id = customer_id,
-                    loan_id = loan_id,
-                    product_id = product_id
+                if (RepaymentService.validate_account_ownership(account_id = account_id, customer_id = customer_id)):
 
-                )
+                    # Save the repayment details  
+                    repayment = Repayment.create_repayment(
+                        customer_id = customer_id,
+                        loan_id = loan_id,
+                        product_id = product_id
 
-                if not repayment:
-                    logger.error(f"Failed to save repayment details")
+                    )
+
+                    if not repayment:
+                        logger.error(f"Failed to save repayment details")
+                        return jsonify({
+                            "message": "Failed to save repayment details."
+                        }), 400
+            
+                
+                    #Update Loan status
+                    Loan.update_status(loan_id = loan_id, status = LoanStatus.REPAID)
+                    
+                    transaction = RepaymentService.log_transaction(validated_data = validated_data)
+
+                    if not transaction:
+                        logger.error(f"Failed to log transaction")
+                        return jsonify({
+                            "message": "Failed to log transaction."
+                        }), 400
+                    
+                else: 
                     return jsonify({
-                        "message": "Failed to save repayment details."
-                    }), 400
+                            "message": "Invalid Customer or Account"
+                        }), 400
                 
-                db.session.flush()
-                
-                validated_data['refId'] = repayment.id
-                validated_data['refModel'] = "repayment"
-                
-                #Update Loan status
-                Loan.update_status(loan_id = loan_id, status = LoanStatus.REPAID)
-                
-                transaction = RepaymentService.log_transaction(validated_data = validated_data)
-
-                if not transaction:
-                    logger.error(f"Failed to log transaction")
-                    return jsonify({
-                        "message": "Failed to log transaction."
-                    }), 400
 
 
                 # Simulated processing logic