Update to allow comma separated item list add

diff --git a/README.md b/README.md
index fb3a4b5b9890331ce4e8ab50e978a34129c36a35..04caa1641847fb326aa6507c5280a86ae1932fcc 100644 (file)
--- a/README.md
+++ b/README.md
@@ -57,6 +57,7 @@ This bot can also be invoked from the commande line with `invoke`.
 
 ## TODO List
 
+- [x] allow comma separated list for shopping item add
 - [ ] add PurchaseRecord table to track how often an item is purchased at a given store
 - [ ] add a `quick.list` command that populates a store shopping list with the most commonly shopped items at that store from the PurchaseRecord table in the database
 

diff --git a/matrix_shop_bot/bot.py b/matrix_shop_bot/bot.py
index 4c0a916c148a9072cafa710e67f7ba53e8117dad..7c8d9a26878722360d846a271f0fca7228d446ad 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -27,27 +27,31 @@ async def do_shop(match):
 
     Commands:
       help - show this help
+      extra.help - show extended help
       add  - add an item to shop for
-      add-store - add a store
-      assign-item-to-store ITEM_ID STORE_ID - plan to shop for at store for item
+      add.store - add a store
       assign ITEM_ID - assign item to last used store id
       buy - mark item as purchased
       clear STORE_ID - clear all purchased items for STORE_ID from database
-      clear-all STORE_ID - clear **all** items for STORE_ID from database
       list - re-show list from last used store
-      list-for-store STORE_ID - list all items assigned to store
-      list-stores - list all stores
-      unsorted - list all items not assigned to any store
-      remove-item ITEM_ID - remove a shopping item from the datatbase
-      remove-store STORE_ID - remove a store from the datatbase (requires clearing the list first)
+      stores - list all stores
+      shop.at STORE_ID - list all items assigned to store and use this store for future commands
+    '''
+    extended_help = '''
+      assign.item.to.store ITEM_ID STORE_ID - plan to shop for at store for item
+      clear.all STORE_ID - clear **all** items for STORE_ID from database
+      remove.item ITEM_ID - remove a shopping item from the datatbase
+      remove.store STORE_ID - remove a store from the datatbase (requires clearing the list first)
       setup - run one time database setup
+      unsorted - list all items not assigned to any store
+
     '''
 
     if match.command('setup'):
         result = Data().setup()
         return [result]
 
-    if match.command('list-stores'):
+    if match.command('stores'):
         items = Data().get_stores()
         results = ['List of Stores']
         for item in items:
@@ -62,12 +66,12 @@ async def do_shop(match):
             results.append(str(item))
         return results
 
-    if match.command('add-store'):
+    if match.command('add.store'):
         new = ' '.join(match.args()[1:])
         result = Data().add_store(new)
         return [f'Store added with id {result}']
 
-    if match.command('assign-item-to-store'):
+    if match.command('assign.item.to.store'):
         item_id  = int(match.args()[1])
         store_id = int(match.args()[2])
         Data().assign_store(item_id, store_id)
@@ -95,13 +99,13 @@ async def do_shop(match):
         self.last_store_id = None
         return ['Success']
 
-    if match.command('clear-all'):
+    if match.command('clear.all'):
         store_id = ' '.join(match.args()[1:])
         Data().clear_list(store_id, clear_all=True)
         return ['Success']
 
 
-    if match.command('list-for-store'):
+    if match.command('shop.at'):
         store_id = ' '.join(match.args()[1])
         items = Data().get_shopping_list_for_store(store_id)
         results = ['Shopping List']
@@ -123,16 +127,20 @@ async def do_shop(match):
             results.append(str(item))
         return results
 
-    if match.command('remove-item'):
+    if match.command('remove.item'):
         item_id = ' '.join(match.args()[1:])
         Data().remove_item(item_id)
         return ['Success']
 
-    if match.command('remove-store'):
+    if match.command('remove.store'):
         store_id = ' '.join(match.args()[1:])
         Data().remove_store(store_id)
         return ['Success']
 
+
+    if match.command('extra.help'):
+        return [extended_help]
+
     return [do_shop.__doc__]
 
 async def shop_bot(room, message, bot):

diff --git a/matrix_shop_bot/db.py b/matrix_shop_bot/db.py
index cd075b923d05d97a0f2da5c36831355ef1fdebfd..58bb4ff29a8f0182c0a6381c10e76a681b1a5ca8 100644 (file)
--- a/matrix_shop_bot/db.py
+++ b/matrix_shop_bot/db.py
@@ -6,6 +6,8 @@ from typing import List
 
 from sqlmodel import Field, Session, SQLModel, create_engine, select, text, Relationship
 
+from sqlite3 import IntegrityError
+
 MODULE='shop'
 
 DATE_FORMAT = '%Y-%m-%d'
@@ -83,13 +85,20 @@ class Data():
             return pref.store_id
 
     def _add_with_name(self, cls, name):
-        id = -1
-        with Session(self.engine) as session:
-            new = cls(name=name)
-            session.add(new)
-            session.commit()
-            id = new.id
-        return id
+        item_id = -1
+        try:
+            with Session(self.engine) as session:
+                new = cls(name=name)
+                session.add(new)
+                session.commit()
+                item_id = new.id
+        except:
+            # duplicate add
+            # recover by returning the item_id
+            item = session.exec(select(Item).where(Item.name == name)).first()
+            item_id = item.id
+
+        return item_id
 
     def add_store(self, name):
         store_id = self._add_with_name(Store, name)
@@ -97,9 +106,11 @@ class Data():
         return store_id
             
     def add_item(self, name):
-        item_id = self._add_with_name(Item, name)
-        store_id = self._get_pref_store_id()
-        self.assign_store(item_id, store_id)
+        names = name.split(',')
+        for name in names:
+            item_id = self._add_with_name(Item, name.strip())
+            store_id = self._get_pref_store_id()
+            self.assign_store(item_id, store_id)
         return item_id
 
     def assign_last_store(self, item_id):
@@ -113,8 +124,11 @@ class Data():
         with Session(self.engine) as session:
             store = session.exec(_get_with_id(Store, store_id)).first()
             item = session.exec(_get_with_id(Item, item_id)).first()
-            item.stores.append(store)
-            session.commit()
+            try:
+                item.stores.append(store)
+                session.commit()
+            except:
+                pass # Already assigned
         return ['Success']
 
     def purchase_item(self, item_id):