X-Git-Url: https://us-delaporte.pagekite.me/gitweb/matrix-shop-bot.git/blobdiff_plain/068a02cde78b07286384ee7de34cc442eb5b7233..2fe28d1dc4ee451541c9f4e921b5c6eb71e91aed:/matrix_shop_bot/db.py

diff --git a/matrix_shop_bot/db.py b/matrix_shop_bot/db.py
index 58bb4ff..186350a 100644
--- a/matrix_shop_bot/db.py
+++ b/matrix_shop_bot/db.py
@@ -44,10 +44,6 @@ class Store(SQLModel, table=True):
     def __str__(self):
         return f'{self.name} (#{self.id})'
 
-
-
-
-
 # May get used later...
 #class PurchaseRecord(SQLModel, table=True):
 #    id: int | None = Field(default=None, primary_key=True)
@@ -131,6 +127,18 @@ class Data():
                 pass # Already assigned
         return ['Success']
 
+
+    def purchase_items(self, item_ids):
+        results = []
+        with Session(self.engine) as session:
+            for item_id in item_ids.split(' '):
+                item = session.exec(_get_with_id(Item, item_id)).first()
+                if item:
+                    item.purchase_date = dt.today()
+                results.append(str(item))
+            session.commit()
+        return results
+
     def purchase_item(self, item_id):
         with Session(self.engine) as session:
             item = session.exec(_get_with_id(Item, item_id)).first()