38 lines
1.8 KiB
Python
38 lines
1.8 KiB
Python
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}>'
|