CHIEFSOFT\ameye
81 lines
3.2 KiB
Python
81 lines
3.2 KiB
Python
from datetime import datetime, timezone
|
|
from app.api.enums.loan_status import LoanStatus
|
|
from app.extensions import db
|
|
from app.models.customer import Customer
|
|
from app.models.loan import Loan
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.sql import func
|
|
|
|
|
|
class Repayment(db.Model):
|
|
__tablename__ = 'repayments'
|
|
|
|
id = db.Column(
|
|
db.Integer,
|
|
primary_key=True,
|
|
autoincrement=True,
|
|
)
|
|
loan_id = db.Column(db.String(50), nullable=False)
|
|
customer_id = db.Column(db.String(50), nullable=False)
|
|
product_id = db.Column(db.String(20), 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())
|
|
transaction_id = db.Column(db.String(50), nullable=True)
|
|
# repay_date = db.Column(db.DateTime, default=datetime.now(timezone.utc))
|
|
repay_date = db.Column(db.DateTime, nullable=True)
|
|
repay_result = db.Column(db.String(10), nullable=True)
|
|
repay_description = db.Column(db.String(100), nullable=True)
|
|
# verify_date = db.Column(db.DateTime, default=datetime.now(timezone.utc))
|
|
verify_date = db.Column(db.DateTime, nullable=True)
|
|
verify_result = db.Column(db.String(10), nullable=True)
|
|
verify_description = db.Column(db.String(100), nullable=True)
|
|
initiated_by = db.Column(db.String(50), nullable=True)
|
|
salary_amount = db.Column(db.Float, nullable=True, default=0.0)
|
|
|
|
@classmethod
|
|
def create_repayment(cls, customer_id, loan, transaction_id):
|
|
|
|
# Check that the loan is active
|
|
if loan.status not in [LoanStatus.ACTIVE, LoanStatus.START_REPAY]:
|
|
raise ValueError(f"Repayment cannot be processed. Loan status: ({loan.status})")
|
|
|
|
|
|
repayment = cls(
|
|
customer_id=customer_id,
|
|
loan_id=loan.id,
|
|
product_id=loan.product_id,
|
|
transaction_id = transaction_id,
|
|
created_at=datetime.now(timezone.utc),
|
|
updated_at=datetime.now(timezone.utc),
|
|
initiated_by='USER_INITIATED'
|
|
)
|
|
|
|
try:
|
|
db.session.add(repayment)
|
|
except IntegrityError as err:
|
|
raise ValueError(f"Database integrity error: {err}")
|
|
|
|
return repayment
|
|
|
|
def to_dict(self):
|
|
return {
|
|
"id": self.id,
|
|
"loan_id": self.loan_id,
|
|
"customer_id": self.customer_id,
|
|
"product_id": self.product_id,
|
|
"created_at": self.created_at.isoformat() if self.created_at else None,
|
|
"updated_at": self.updated_at.isoformat() if self.updated_at else None,
|
|
"transaction_id": self.transaction_id,
|
|
"repay_date": self.repay_date.isoformat() if self.repay_date else None,
|
|
"repay_result": self.repay_result,
|
|
"repay_description": self.repay_description,
|
|
"verify_date": self.verify_date.isoformat() if self.verify_date else None,
|
|
"verify_result": self.verify_result,
|
|
"verify_description": self.verify_description,
|
|
"initiated_by": self.initiated_by,
|
|
"salary_amount": self.salary_amount
|
|
}
|
|
|
|
def __repr__(self):
|
|
return f'<Repayment {self.id}>'
|