diff options
| -rw-r--r-- | app/model.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/app/model.py b/app/model.py index b68dc57..248647c 100644 --- a/app/model.py +++ b/app/model.py @@ -138,6 +138,12 @@ class SingleExpense (UserModel, CategoryModel): self.month = d.month self.day = d.day +class ConstExpenseGroup (UserModel, CategoryModel): + description = ReqColumn(db.Unicode(50)) + start = ReqColumn(db.Date, index = True) + end = ReqColumn(db.Date, index = True) + + expenses = db.relationship("ConstExpense", back_populates = "group") class ConstExpense (UserModel, CategoryModel): description = Column(db.Unicode(50)) @@ -146,9 +152,12 @@ class ConstExpense (UserModel, CategoryModel): start = ReqColumn(db.Date, index = True) end = ReqColumn(db.Date, index = True) prev_id = Column(db.Integer, db.ForeignKey('constexpense.id')) + prev = db.relationship('ConstExpense', + remote_side = 'ConstExpense.id', uselist = False, + backref=db.backref('next', uselist = False)) - prev = db.relationship('ConstExpense', remote_side = 'ConstExpense.id', uselist = False, - backref=db.backref('next', uselist = False)) + group_id = Column(db.Integer, db.ForeignKey(ConstExpenseGroup.id), index = True) + group = db.relationship(ConstExpenseGroup, back_populates = "expenses") @property def monthly(self): |