from app.extensions import db
from datetime import datetime


class RecurringRule(db.Model):
    __tablename__ = 'recurring_rules'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)
    account_id = db.Column(db.Integer, db.ForeignKey('accounts.id'), nullable=False)
    category_id = db.Column(db.Integer, db.ForeignKey('categories.id'), nullable=True)
    transaction_type = db.Column(
        db.Enum('income', 'expense'),
        nullable=False
    )
    amount = db.Column(db.Numeric(15, 2), nullable=False)
    description = db.Column(db.String(255), nullable=False)
    frequency = db.Column(
        db.Enum('daily', 'weekly', 'biweekly', 'monthly', 'quarterly', 'yearly'),
        nullable=False,
        default='monthly'
    )
    start_date = db.Column(db.Date, nullable=False)
    end_date = db.Column(db.Date, nullable=True)
    last_run = db.Column(db.Date, nullable=True)
    next_run = db.Column(db.Date, nullable=True, index=True)
    is_active = db.Column(db.Boolean, default=True)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

    transactions = db.relationship('Transaction', back_populates='recurring_rule', lazy='dynamic')
    account = db.relationship('Account')
    category = db.relationship('Category')

    def __repr__(self):
        return f'<RecurringRule {self.name} {self.frequency}>'