import pickle

from inventory$\_$item import InventoryItem, CATEGORY$\_$LIST

def main$()$:

inventory $=$ load$\_$inventory$()$

display$\_$inventory$($inventory$)$

answer $=\text{'}\text{'}$

while answer.lower$()!=\text{'}$n$\text{'}$:

# TODO $-$ Create an InventoryItem object, ask the user for item input

# using the object's method, then append the object to the inventory

# list.

answer $=$ input$(\text{'}$Do you want to enter more items? $\text{'})$

for category in CATEGORY$\_$LIST:

display$\_$category$($category$,$ inventory$)$

display$\_$inventory$($inventory$)$

save$\_$inventory$($inventory$)$

def load$\_$inventory$()$:

inventory $=[]$

# TODO $-$ Attempt to load inventory data from a binary file named

# inventory.dat. If the file exists, load it into the inventory list.

# If the file does not exist, leave the inventory list empty.

return inventory

def save$\_$inventory$($inventory$)$:

# TODO $-$ Open a binary file named inventory.dat and dump the inventory

# list that has been passed in as a parameter to that file.

print$(\text{'}$Inventory was saved in inventory.dat.$\text{'})$

def display$\_$category$($category$\_$name, inventory$)$:

print$()$

header $=$ f'Items in $\{$category$\_$name$\}\text{'}$

divider $=$ len$($header$)*\text{'}-\text{'}$

print$($header$)$

print$($divider$)$

# TODO $-$ Display inventory items that are in the specific category

# Both the category name and inventory are passed in as parameters.

# If there are no items in the category, print $"$No items."

def display$\_$inventory$($inventory$)$:

print$()$

print$(\text{'}$Current Inventory'$)$

print$(\text{'}-----------------\text{'})$

# TODO $-$ Display the inventory items that are in the inventory list

# that was passed in as a parameter. If the list is empty, display

# "Inventory is empty."

print$()$

main$()$