[add]: balance when calculating loan total

This commit was merged in pull request #32.
This commit is contained in:
VivianDee
2025-06-25 08:23:11 +01:00
parent aa408cc720
commit 9831395e2a
+2 -1
View File
@@ -30,6 +30,7 @@ class Loan(db.Model):
continuous_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) upfront_fee = db.Column(db.Float, nullable=True, default=0.0)
repayment_amount = 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) installment_amount = db.Column(db.Float, nullable=True, default=0.0)
status = db.Column(db.String(20), default='pending') status = db.Column(db.String(20), default='pending')
tenor = db.Column(db.Integer, nullable=True) tenor = db.Column(db.Integer, nullable=True)
@@ -230,7 +231,7 @@ class Loan(db.Model):
raise ValueError(f"Customer with Id {customer_id} does not have any loan.") raise ValueError(f"Customer with Id {customer_id} does not have any loan.")
total_amount = ( 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) .filter_by(customer_id=customer_id)
.scalar() .scalar()
) )