from app.extensions import db
from datetime import datetime


class Transaction(db.Model):
    __tablename__ = 'transactions'

    id = db.Column(db.Integer, primary_key=True)
    account_id = db.Column(db.Integer, db.ForeignKey('accounts.id'), nullable=False)
    to_account_id = db.Column(db.Integer, db.ForeignKey('accounts.id'), nullable=True)  # transfers
    category_id = db.Column(db.Integer, db.ForeignKey('categories.id'), nullable=True)
    recurring_rule_id = db.Column(db.Integer, db.ForeignKey('recurring_rules.id'), nullable=True)
    receipt_id = db.Column(db.Integer, db.ForeignKey('receipts.id'), nullable=True)

    transaction_type = db.Column(
        db.Enum('income', 'expense', 'transfer'),
        nullable=False
    )
    amount = db.Column(db.Numeric(15, 2), nullable=False)
    description = db.Column(db.String(255), nullable=False)
    notes = db.Column(db.Text, nullable=True)
    date = db.Column(db.Date, nullable=False, index=True)
    is_recurring = db.Column(db.Boolean, default=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    # Relationships
    account = db.relationship('Account', back_populates='transactions',
                               foreign_keys=[account_id])
    to_account = db.relationship('Account', back_populates='transfer_transactions',
                                  foreign_keys=[to_account_id])
    category = db.relationship('Category', back_populates='transactions')
    receipt = db.relationship('Receipt', back_populates='transaction')
    recurring_rule = db.relationship('RecurringRule', back_populates='transactions')

    def __repr__(self):
        return f'<Transaction {self.transaction_type} {self.amount} on {self.date}>'