From ca12aa47337a700fe7647700079ad0bb96ae349e Mon Sep 17 00:00:00 2001 From: b4-sharp <Bradley2.Sharp@live.uwe.ac.uk> Date: Mon, 27 Mar 2023 11:20:28 +0100 Subject: [PATCH] Add calculate_price() --- store/models.py | 7 +++++-- store/site.db | 0 2 files changed, 5 insertions(+), 2 deletions(-) delete mode 100644 store/site.db diff --git a/store/models.py b/store/models.py index b5dec8b..6cb85f7 100644 --- a/store/models.py +++ b/store/models.py @@ -60,7 +60,7 @@ class ItemSet(db.Model): description = db.Column(db.String(256), nullable=False) price = db.Column( db.Integer, nullable=False - ) # In pounds, as we're not dividing or multiplying this will not matter in calculations. Do we neccesarily need this? It could be calculated dynamically from the items held. + ) # This is actually the quantity not price... TODO: Rename items = db.relationship( "Item", secondary=itemSets, @@ -68,8 +68,11 @@ class ItemSet(db.Model): backref=db.backref("ItemSets", lazy=True), ) + def calculate_price(self): + return sum(int(item.price) for item in self.items) + def __repr__(self): - return f"id: {self.id}, description: {self.description}, items: {self.items}" + return f"id: {self.id}, description: {self.description}, items: {self.items}, price: {self.calculate_price()}" class User(db.Model, UserMixin): diff --git a/store/site.db b/store/site.db deleted file mode 100644 index e69de29..0000000 -- GitLab