yisroelbaum/BudgetingApp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

change return objects from rows to scalars and add method to display the date in a user friendly way

Browse source
This commit is contained in:
Yisroel Baum 2024-10-13 14:40:17 +03:00
parent 2eb6a03826
commit 095ee66072
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
app/models.py
View file

@ -1,5 +1,6 @@
from app import db from app import db
from sqlalchemy import Column, INTEGER, String, select from sqlalchemy import Column, INTEGER, String, select
import datetime
class Vendor(db.Model): class Vendor(db.Model):
__tablename__ = 'vendor' __tablename__ = 'vendor'
@ -43,10 +44,14 @@ class LineItem(db.Model):
note = Column('note', String(), nullable=True) note = Column('note', String(), nullable=True)
def get_parent_line(self): def get_parent_line(self):
return db.session.execute(select(LineItem).where(LineItem.id==self.parent_line_item_id)).first() return db.session.execute(select(LineItem).where(LineItem.id==self.parent_line_item_id)).scalar()
def get_children_lines(self): def get_children_lines(self):
return db.session.execute(select(LineItem).where(LineItem.parent_line_item_id==self.id)).all() return db.session.execute(select(LineItem).where(LineItem.parent_line_item_id==self.id)).all()
def get_vendor(self): def get_vendor(self):
return db.session.execute(select(Vendor).where(Vendor.id==self.vendor_id)).first() return db.session.execute(select(Vendor).where(Vendor.id==self.vendor_id)).scalar()
def display_date(self):
dt_object = datetime.datetime.fromtimestamp(self.date)
return dt_object.strftime('%Y-%m-%d')