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