From 9831395e2a8a6cee9a7d00c2d53cf875cf74d473 Mon Sep 17 00:00:00 2001 From: VivianDee <115420678+VivianDee@users.noreply.github.com> Date: Wed, 25 Jun 2025 08:23:11 +0100 Subject: [PATCH] [add]: balance when calculating loan total --- app/models/loan.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/app/models/loan.py b/app/models/loan.py index 0d598e6..fe7d812 100644 --- a/app/models/loan.py +++ b/app/models/loan.py @@ -30,6 +30,7 @@ class Loan(db.Model): 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) + balance = 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') tenor = db.Column(db.Integer, nullable=True) @@ -230,10 +231,10 @@ class Loan(db.Model): raise ValueError(f"Customer with Id {customer_id} does not have any loan.") total_amount = ( - db.session.query(func.coalesce(func.sum(cls.current_loan_amount), 0.0)) + cls.query.with_entities(func.coalesce(func.sum(cls.balance), 0.0)) .filter_by(customer_id=customer_id) .scalar() - ) + ) logger.info(f"Found {len(customer_loans)} loans for customer ID: {customer_id} with total amount: {total_amount}") -- 2.34.1