Fixed grocery list logic

Author: rjLelis

grocery-shopping/grocery_list_repository.py

@@ -1,31 +1,24 @@
 import pickle
-import grocery_list
+from pathlib import Path
 
-file_name = "grocery-list.data"
+file = Path("grocery-list.data")
 
 
 def persist(items):
-    with open(file_name, "wb") as f:
+    with open(file, "wb") as f:
         pickle.dump(items, f)
 
 
 def read():
     items = []
-    with open(file_name, "rb") as f:
-        items = pickle.load(f)
+    if file.is_file():
+        with open(file, "rb") as f:
+            items = pickle.load(f)
     return items
 
 
-def update(item, new_item):
-    items = read()
-    items[items.index(item)] = new_item
-    with open(file_name, "r+b") as f:
-        pickle.dump(items, f)
-
-
 def delete(item):
     items = read()
     items.remove(item)
-    with open(file_name, "r+b") as f:
-        items = grocery_list.get_list()
+    with open(file, "r+b") as f:
         pickle.dump(items, f)

grocery-shopping/main.py

@@ -1,36 +1,47 @@
-import grocery_list
 import grocery_list_repository
 
+items = grocery_list_repository.read()
+
+
+def options(option):
+    if option == 1:
+        item = input("Enter the item: ")
+        global items
+        items.append(item)
+        grocery_list_repository.persist(items)
+        print(f"Item {item} saved in the list")
+        return 0
+    elif option == 2:
+        item = input("Enter the item you want to remove: ")
+        items.remove(item)
+        grocery_list_repository.persist(items)
+        print(f"Item {item} removed from the list")
+        return 0
+    elif option == 3:
+        print("Items - ")
+        if grocery_list_repository.read():
+            items = grocery_list_repository.read()
+            for item in items:
+                print(item, end=" ")
+            print()
+        else:
+            print("Empty list")
+        return 0
+    elif option == 0:
+        return 1
+
 
 def main():
     option = 0
-    items = grocery_list_repository.read()
     while True:
         print("1 - Add item to the list")
-        print("2 - Update item from the list")
-        print("3 - Delete item from the list")
-        print("4 - Get all items from the list")
-        print("5 - Save list")
+        print("2 - Delete item from the list")
+        print("3 - Get all items from the list")
         print("0 - Exit")
-        option = int(input("> "))
-
+        option = options(int(input("> ")))
         if option == 1:
-            item = input("Enter the item: ")
-            grocery_list.add(item.title())
-        elif option == 2:
-            item = input("Enter the item you want to replace: ")
-            new_item = input("Enter the new item to store on the list: ")
-            grocery_list_repository.update(item, new_item)
-        elif option == 3:
-            item = input("Enter the item you want to remove: ")
-            grocery_list_repository.delete(item)
-        elif option == 4:
-            print("Items - ")
-            print(grocery_list_repository.read())
-        elif option == 5:
-            grocery_list_repository.persist(grocery_list.get_list())
-        elif option == 0:
             break
 
+
 if __name__ == '__main__':
     main()