from datetime import datetime, timezone from app.extensions import db from app.models.customer import Customer from app.models.account import Account from sqlalchemy.exc import IntegrityError from sqlalchemy.orm import relationship from sqlalchemy import and_, or_, not_ from app.models import Customer from app.utils.logger import logger class Loan(db.Model): __tablename__ = 'loans' id = db.Column( db.Integer, primary_key=True, autoincrement=True, ) customer_id = db.Column(db.String(50), nullable=False) transaction_id = db.Column(db.String(50), nullable=True) original_transaction = 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) product_id = db.Column(db.String(20), nullable=True) 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) upfront_fee = db.Column(db.Float, nullable=True, default=0.0) repayment_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') tenor = db.Column(db.Integer, nullable=True) 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)) eligible_amount = db.Column(db.Float, nullable=True, default=0.0) disburse_date = db.Column(db.DateTime, nullable=True) disburse_verify = db.Column(db.DateTime, nullable=True) disburse_result = db.Column(db.String(10), nullable=True) disburse_description = db.Column(db.String(100), nullable=True) verify_result = db.Column(db.String(10), nullable=True) verify_description = db.Column(db.String(100), nullable=True) reference = db.Column(db.String(50), nullable=True) balance = db.Column(db.Float, nullable=True, default=0.0) customer = relationship( "Customer", primaryjoin="Customer.id == Loan.customer_id", foreign_keys=[customer_id], back_populates="loans", ) @classmethod def get_all_loans(cls, id=None, customer_id=None, account_id=None, status=None, tenor=None, offer_id=None, product_id=None, start_date=None, end_date=None, due_before=None, due_after=None, transaction_id=None, original_transaction=None, page=1, limit=20): """ Get all loans with optional filtering Args: id (int, optional): Filter by loan ID customer_id (str, optional): Filter by customer ID account_id (str, optional): Filter by account ID status (str, optional): Filter by loan status tenor (str, optional): Filter by loan tenor offer_id (str, optional): Filter by offer ID product_id (str, optional): Filter by product ID start_date (datetime, optional): Filter by start date (created_at) end_date (datetime, optional): Filter by end date (created_at) due_before (datetime, optional): Filter loans due before this date due_after (datetime, optional): Filter loans due after this date transaction_id (str, optional): Filter by transaction ID original_transaction (str, optional): Filter by original transaction Returns: list: List of Loan objects """ query = cls.query logger.info(f"Get all loan models from loans model cme back") # Apply filters if provided if id: query = query.filter(cls.id == id) if customer_id: query = query.filter(cls.customer_id == customer_id) if account_id: query = query.filter(cls.account_id == account_id) if status: query = query.filter(cls.status == status) if tenor: query = query.filter(cls.tenor == tenor) if offer_id: query = query.filter(cls.offer_id == offer_id) if product_id: query = query.filter(cls.product_id == product_id) if transaction_id: query = query.filter(cls.transaction_id == transaction_id) if original_transaction: query = query.filter(cls.original_transaction == original_transaction) if start_date: query = query.filter(cls.created_at >= start_date) if end_date: query = query.filter(cls.created_at <= end_date) if due_before: query = query.filter(cls.due_date <= due_before) if due_after: query = query.filter(cls.due_date >= due_after) # Order by created_at descending (newest first) query = query.order_by(cls.created_at.desc()) # Get total count before pagination total_count = query.count() # Apply pagination offset = (page - 1) * limit query = query.limit(limit).offset(offset) return query.all(), total_count def to_dict(self): """ 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, 'transaction_id': self.transaction_id, 'eligible_amount': self.eligible_amount, 'original_transaction': self.original_transaction, 'offer_id': self.offer_id, 'initial_loan_amount': self.initial_loan_amount, 'current_loan_amount': self.current_loan_amount, 'status': self.status, 'tenor': self.tenor, 'product_id': self.product_id, 'default_penalty_fee': self.default_penalty_fee, 'continuous_fee': self.continuous_fee, 'upfront_fee': self.upfront_fee, 'repayment_amount': self.repayment_amount, 'installment_amount': self.installment_amount, 'due_date': self.due_date.isoformat() if self.due_date else None, 'created_at': self.created_at.isoformat() if self.created_at else None, 'updated_at': self.updated_at.isoformat() if self.updated_at else None, 'disburseResult': self.disburse_result, 'disburseDescription': self.disburse_description, 'verifyResult': self.verify_result, 'verifyDescription': self.verify_description, 'disburseDate': self.disburse_date.isoformat() if self.disburse_date else None, 'disburseVerify': self.disburse_verify.isoformat() if self.disburse_verify else None, 'reference': self.reference, 'balance': self.balance, } def __repr__(self): return f''