Untitled

mail@pastecode.io avatarunknown
plain_text
19 days ago
1.1 kB
1
Indexable
Never
class InventorySystem:
    def __init__(self):
        self.inventory = {}

    def add_item(self, item_name, quantity):
        if item_name in self.inventory:
            self.inventory[item_name] += quantity
        else:
            self.inventory[item_name] = quantity

    def view_inventory(self):
        print("Inventory:")
        for item, quantity in self.inventory.items():
            print(f"{item}: {quantity}")

def main():
    inventory_system = InventorySystem()

    while True:
        print("\nOptions:")
        print("1. Add item to inventory")
        print("2. View inventory")
        print("3. Exit")
        choice = input("Enter your choice: ")

        if choice == '1':
            item_name = input("Enter item name: ")
            quantity = int(input("Enter quantity: "))
            inventory_system.add_item(item_name, quantity)
        elif choice == '2':
            inventory_system.view_inventory()
        elif choice == '3':
            break
        else:
            print("Invalid choice. Please try again.")

if __name__ == "__main__":
    main()