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

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)
    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)
    fbn_transaction_id = db.Column(db.String(255),nullable=True)
    account_id = db.Column(db.String(50), nullable=True)
    customer_id = db.Column(db.String(50), nullable=True)
    repayment_amount = db.Column(db.Float, nullable=True)
    amount_collected = db.Column(db.Float, 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,
            "added_date": self.added_date.isoformat() if self.added_date else None,
            "response_code": self.response_code,
            "response_descr": self.response_descr,
            "customerId": self.customer_id,
            "accountId": self.account_id,
            "fbnTransactionId": self.fbn_transaction_id,
            "repaymentAmount": self.repayment_amount,
            "amountCollected": self.amount_collected,
            "balance": self.balance
        }


    def __repr__(self):
        return f"<RepaymentsData id={self.id}, transaction_id={self.transaction_id}>"
    
    @classmethod
    def add_repayment_data(cls, data):
        """
        Add a new repayment data entry.
        """
        try:
            repayment_amount = float(data.get('repaymentAmount', 0.0))
            amount_collected = float(data.get('amountCollected', 0.0))

            if amount_collected < 0 or repayment_amount < 0:
                raise ValueError("Amounts cannot be negative.")

            account_balance = round(repayment_amount - amount_collected, 2)

            new_data = cls(
                transaction_id=data.get('transactionId'),
                response_code=data.get('responseCode'),
                response_descr=data.get('responseDescr'),
                fbn_transaction_id=data.get('fbnTransactionId'),
                account_id=data.get('accountId'),
                customer_id=data.get('customerId'),
                amount_collected=amount_collected,
                repayment_amount=repayment_amount,
                balance=account_balance,
            )

            db.session.add(new_data)
            db.session.commit()

            logger.info("Repayment data committed successfully")
            return new_data

        except Exception as e:
            db.session.rollback()
            logger.error(f"Error adding repayment data: {e}")
            raise Exception(f"Error adding repayment data: {str(e)}")