CHIEFSOFT\ameye
92 lines
3.3 KiB
Python
92 lines
3.3 KiB
Python
from datetime import datetime, timezone, timedelta
|
|
from app.extensions import db
|
|
from sqlalchemy.orm import relationship
|
|
from app.utils.logger import logger
|
|
from sqlalchemy.sql import func
|
|
|
|
|
|
class LoanCharge(db.Model):
|
|
__tablename__ = 'loan_charges'
|
|
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
loan_id = db.Column(db.Integer, nullable=False)
|
|
transaction_id = db.Column(db.String(50), nullable=True)
|
|
code = db.Column(db.String(50), nullable=False)
|
|
amount = db.Column(db.Float, default=0.0)
|
|
percent = db.Column(db.Float, default=0.0)
|
|
description = db.Column(db.Text, nullable=True)
|
|
due = db.Column(db.Integer, nullable=False)
|
|
due_date = db.Column(db.DateTime, nullable=True)
|
|
created_at = db.Column(db.DateTime(timezone=True), server_default=func.now())
|
|
updated_at = db.Column(db.DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
|
|
loan = relationship(
|
|
"Loan",
|
|
primaryjoin="LoanCharge.loan_id == Loan.id",
|
|
foreign_keys=[loan_id],
|
|
back_populates="loan_charges",
|
|
)
|
|
|
|
@classmethod
|
|
def create_charges_for_loan(cls, loan_id, transaction_id, charges, referenced_amount = 0.0):
|
|
"""
|
|
Create loan charges for a given loan.
|
|
|
|
Args:
|
|
loan_id (int): ID of the loan to associate charges with.
|
|
charges (list): A list of dictionaries with keys:
|
|
code (str), amount (float), percent (float), description (str), due (int)
|
|
"""
|
|
# if not charges or not isinstance(charges, list):
|
|
# raise ValueError("Charges must be a non-empty list of dictionaries")
|
|
|
|
if loan_id is None:
|
|
raise ValueError("loan_id cannot be None")
|
|
|
|
loan_charges = []
|
|
now = datetime.now(timezone.utc)
|
|
|
|
|
|
subset_keys = ['interest', 'management', 'insurance', 'vat']
|
|
for item in subset_keys:
|
|
charge = charges[item]
|
|
due_days = charge['due_days'] # getattr(charge, "due_days", 0)
|
|
amount = charge['fee'] # getattr(charge, "fee", 0.0)
|
|
percent = charge['rate'] # getattr(charge, "rate", 0.0)
|
|
code = charge['code'] # getattr(charge, "code","")
|
|
description = charge['description'] # getattr(charge, "description", "")
|
|
|
|
charge_obj = cls(
|
|
loan_id = loan_id,
|
|
transaction_id = transaction_id,
|
|
code = code,
|
|
amount = round(amount, 2),
|
|
percent = percent,
|
|
description = description,
|
|
due = due_days,
|
|
due_date = now + timedelta(days=due_days),
|
|
created_at=datetime.now(timezone.utc),
|
|
updated_at=datetime.now(timezone.utc)
|
|
)
|
|
|
|
db.session.add(charge_obj)
|
|
loan_charges.append(charge_obj)
|
|
|
|
return loan_charges
|
|
|
|
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'loanId': self.loan_id,
|
|
'transactionId': self.transaction_id,
|
|
'code': self.code,
|
|
'amount': self.amount,
|
|
'percent': self.percent,
|
|
'description': self.description,
|
|
'due': self.due,
|
|
}
|
|
|
|
def __repr__(self):
|
|
return f"<LoanCharge {self.id} - Loan {self.loan_id} - {self.code}>"
|