from datetime import datetime, timezone
from app.extensions import db


class RepaymentsData(db.Model):
    __tablename__ = "repayments_data"

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    transaction_id = db.Column(db.String(50), nullable=False)
    fbn_transaction_id = db.Column(db.String(50), nullable=True)
    customer_id = db.Column(db.String(50), nullable=True)
    account_id = db.Column(db.String(50), nullable=True)
    repayment_amount = db.Column(db.Float, nullable=True, default=0.0)
    amount_collected = db.Column(db.Float, nullable=True, default=0.0)
    added_date = db.Column(db.DateTime(timezone=True), default=datetime.now(timezone.utc), nullable=False)
    response_code = db.Column(db.String(10), nullable=True)
    response_descr = db.Column(db.String(255), nullable=True)
    balance = db.Column(db.Float, nullable=True, default=0.0)

    def to_dict(self):
        return {
            "id": self.id,
            "transaction_id": self.transaction_id,
            "fbn_transaction_id": self.fbn_transaction_id,
            "customer_id": self.customer_id,
            "account_id": self.account_id,
            "repayment_amount": self.repayment_amount,
            "amount_collected": self.amount_collected,
            "added_date": self.added_date.isoformat() if self.added_date else None,
            "response_code": self.response_code,
            "response_descr": self.response_descr,
            "balance": self.balance,
        }

    def __repr__(self):
        return f"<RepaymentsData id={self.id}, transaction_id={self.transaction_id}>"