From 095ee66072e90e4a042b889aa6b483f297d0fbaf Mon Sep 17 00:00:00 2001 From: ydb5755 Date: Sun, 13 Oct 2024 14:40:17 +0300 Subject: [PATCH] change return objects from rows to scalars and add method to display the date in a user friendly way --- app/models.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/app/models.py b/app/models.py index 969c3a3..13c9652 100644 --- a/app/models.py +++ b/app/models.py @@ -1,5 +1,6 @@ from app import db from sqlalchemy import Column, INTEGER, String, select +import datetime class Vendor(db.Model): __tablename__ = 'vendor' @@ -43,10 +44,14 @@ class LineItem(db.Model): note = Column('note', String(), nullable=True) 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): return db.session.execute(select(LineItem).where(LineItem.parent_line_item_id==self.id)).all() def get_vendor(self): - return db.session.execute(select(Vendor).where(Vendor.id==self.vendor_id)).first() \ No newline at end of file + 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') \ No newline at end of file