Fix for buying multiple items in one command

[matrix-shop-bot.git] / matrix_shop_bot / db.py

diff --git a/matrix_shop_bot/db.py b/matrix_shop_bot/db.py
index 58bb4ff29a8f0182c0a6381c10e76a681b1a5ca8..186350acfd606f2fd1c5d03da8f25e931341b040 100644 (file)
--- 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})'
 
     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)
 # 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']
 
                 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()
     def purchase_item(self, item_id):
         with Session(self.engine) as session:
             item = session.exec(_get_with_id(Item, item_id)).first()