Added updated logic

qtsathish · qtsathish · commit ec8fe3de9fba · 2021-07-04T18:56:28.000+05:30
diff --git a/July21/EffectivePython/inventory/data/inventory.json b/July21/EffectivePython/inventory/data/inventory.json
@@ -11,6 +11,18 @@
             "name": "oneplus 9",
             "price": 90000.0,
             "quantity": 15
+        },
+        {
+            "id": 3,
+            "name": "ipad",
+            "price": 40000.0,
+            "quantity": 20
+        },
+        {
+            "id": 4,
+            "name": "samsung tab",
+            "price": 25000.0,
+            "quantity": 15
         }
     ]
 }
diff --git a/July21/EffectivePython/inventory/inventory.py b/July21/EffectivePython/inventory/inventory.py
@@ -50,6 +50,15 @@ def to_dict(self):
         inv['quantity'] = self.quantity
         return inv
 
+    def __str__(self) -> str:
+        return f"{self.id}, {self.name}, {self.price}, {self.quantity}"
+
+    @staticmethod
+    def header():
+        return f"id, name, price, quantity"
+
+
+
     @staticmethod
     def read_inventory_from_input():
         id = int(input('Enter the id: '))
diff --git a/July21/EffectivePython/inventory/main.py b/July21/EffectivePython/inventory/main.py
@@ -3,20 +3,58 @@
 """
 import inventory
 import json
+import menu
+
+def write_to_inventory(inventory_dict):
+    with open(inventory.inventory_file_path(), 'w') as inv_file:
+                json.dump(inventory_dict,inv_file,indent=4)
+
 
 if __name__ == '__main__':
     print("Enter the inventory details")
     inventory_dict = inventory.load_inventory_dict()
 
     while True:
-        inv = inventory.Inventory.read_inventory_from_input()
-        inventory_dict['items'].append(inv.to_dict())
-        choice = input('Enter n to stop and any other key to continue')
-        if choice == 'n':
+        choice = menu.main_menu()
+
+        if choice == 1:
+            # Add inventory
+            inv = inventory.Inventory.read_inventory_from_input()
+            inventory_dict['items'].append(inv.to_dict())
+            write_to_inventory(inventory_dict)
+        elif choice == 2:
+            # Update the inventory
+            id = int(input('Enter the Id: '))
+            count = 0
+            for item in inventory_dict['items']:
+                if id == item['id']:
+                    print('product found')
+                    print(item)
+                    price = item['price']
+                    choice = input(f'price: [{price}]')
+                    if choice.strip() != '':
+                        new_price = float(choice.strip())
+                        item['price'] = new_price
+                    quantity = item['quantity']
+                    choice = input(f'quantity: [{quantity}]')
+                    if choice.strip() != '':
+                        new_quantity = int(choice.strip())
+                        item['quantity'] = new_quantity
+                    inventory_dict['items'][count] = item
+                    write_to_inventory(inventory_dict)
+                    break
+                count = count + 1
+            
+        else:
+            # Show the inventory
+            for item in inventory_dict['items']:
+                print(str(item))
+
+        if not menu.does_user_wants_to_continue():
             break
+        
+    
     
-    with open(inventory.inventory_file_path(), 'w') as inv_file:
-        json.dump(inventory_dict,inv_file,indent=4)
 
 
         
diff --git a/July21/EffectivePython/inventory/menu.py b/July21/EffectivePython/inventory/menu.py
@@ -0,0 +1,33 @@
+
+def main_menu():
+    """
+    This implements the main menu functionality
+    Returns
+        1. To Add Inventory Items
+        2. To Update Inventory
+        3. To print all the inventory Items 
+    """
+    try:
+        print("Options")
+        print("1. Add Inventory Items")
+        print("2. Update Inventory Items")
+        print("3. Show Inventory Items")
+        choice = int(input("Enter the choice: "))
+        if not 1 <= choice <=3:
+            raise KeyError('Invalid Choice')
+        return choice
+    except ValueError as ve:
+        raise KeyError(ve)
+
+def does_user_wants_to_continue():
+    """
+    This method will ask for the user input to continue or to stop
+    Returns :
+        True if user wants to continue
+        False otherwise
+    """
+    choice = input('Enter n to stop and any other key to continue :')
+    if choice == 'n':
+        return False
+    return True
+    

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,18 @@`
`11`	`11`	`"name": "oneplus 9",`
`12`	`12`	`"price": 90000.0,`
`13`	`13`	`"quantity": 15`
	`14`	`+ },`
	`15`	`+ {`
	`16`	`+ "id": 3,`
	`17`	`+ "name": "ipad",`
	`18`	`+ "price": 40000.0,`
	`19`	`+ "quantity": 20`
	`20`	`+ },`
	`21`	`+ {`
	`22`	`+ "id": 4,`
	`23`	`+ "name": "samsung tab",`
	`24`	`+ "price": 25000.0,`
	`25`	`+ "quantity": 15`
`14`	`26`	`}`
`15`	`27`	`]`
`16`	`28`	`}`