class Budget(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
budget = db.Column(db.Float)
closed = db.Column(db.Boolean, default=False)
term_id = db.Column(db.Integer, db.ForeignKey('term.id'))
expenditures = db.relationship('Expenditure', backref='budget', lazy=True)
class Transaction(db.Model):
id = db.Column(db.Integer, primary_key=True)
tdate = db.Column(db.Date)
category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
file_id = db.Column(db.Integer, db.ForeignKey('trans_file.id'))
name = db.Column(db.String)
amount = db.Column(db.Float)
bdate = db.Column(db.Date)
yearly = db.Column(db.Boolean, default=False)
category = db.relationship("Category")
trans_file = db.relationship("TransFile")
def __init__(self,
tdate,
name,
category,
amount,
trans_file,
bdate=None,
yearly=False):
self.tdate = tdate
self.name = name
self.amount = amount
self.bdate = bdate or self.tdate
self.yearly = yearly
if isinstance(category, int):
self.category_id = category
else:
self.category = category
if isinstance(trans_file, int):
self.file_id = trans_file
else:
self.trans_file = trans_file
def __repr__(self):
return "<Tran('%d, %s, %s, %s, %s, %s, %s')>" % (
(self.id or 0), self.tdate, self.name, self.category, self.amount,
self.trans_file, self.bdate)
class Transactions(db.Model):
trans_id = db.Column(db.Integer, primary_key=True)
trans_user = db.Column(db.Integer,
db.ForeignKey('account.account_id'),
nullable=False)
trans_amount = db.Column(db.Float, nullable=False)
trans_date = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
trans_mode = db.Column(db.String(10), nullable=False)
trans_type = db.Column(db.String(10), nullable=False)
trans_descr = db.Column(db.Text)
def __repr__(self):
return "Transcation('{}', '{}', '{}')".format(self.trans_id,
self.trans_user,
self.trans_mode)
class CategoryRE(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String)
category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
pattern = db.Column(db.String)
yearly = db.Column(db.Boolean, default=False)
category = db.relationship("Category")
def __init__(self, name, category, pattern, yearly=False):
self.name = name
self.category = category
self.pattern = pattern
self.yearly = yearly
def __repr__(self):
return "<CategoryRE('%s, %s, %s')>" % (self.name, self.category.name,
self.pattern)
class Category(db.Model):
id = db.Column(db.Integer, primary_key=True)
parent_id = db.Column(db.Integer, db.ForeignKey('category.id'))
name = db.Column(db.String)
depth = db.Column(db.Integer, default=0)
children = db.relationship("Category",
backref=db.backref('parent', remote_side=[id]))
def __init__(self, name, parent=None, depth=0):
self.name = name
if parent:
self.parent = parent
self.depth = parent.depth + 1
else:
self.depth = depth
def __repr__(self):
return "<Category('%s')>" % (self.name)
class Fixedcost(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
price = db.Column(db.Float)
payed = db.Column(db.Boolean)
fixedcosttemplate_id = db.Column(db.Integer, db.ForeignKey('fixedcosttemplate.id'))
class Expenditure(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
price = db.Column(db.Float)
payed = db.Column(db.Boolean)
budget_id = db.Column(db.Integer, db.ForeignKey('budget.id'))