Inventory Management.py

"""Functions to keep track and alter inventory."""
def create_inventory(items):
    """Create a dict that tracks the amount (count) of each element on the `items` list.
 
    :param items: list - list of items to create an inventory from.
    :return: dict - the inventory dictionary.
    """
    result_dict = {}
    for item in items:
        result_dict[item] = items.count(item)
    return result_dict
def add_items(inventory, items):
    """Add or increment items in inventory using elements from the items `list`.
 
    :param inventory: dict - dictionary of existing inventory.
    :param items: list - list of items to update the inventory with.
    :return: dict - the inventory updated with the new items.
    """
    for item in items:
        if item not in inventory.keys():
            inventory[item] = 1
            continue
        inventory[item] += 1
    return inventory
def decrement_items(inventory, items):
    """Decrement items in inventory using elements from the `items` list.
 
    :param inventory: dict - inventory dictionary.
    :param items: list - list of items to decrement from the inventory.
    :return: dict - updated inventory with items decremented.
    """
    for item in items:
        if inventory[item] != 0:
            inventory[item] -= 1
    return inventory
def remove_item(inventory, item):
    """Remove item from inventory if it matches `item` string.
 
    :param inventory: dict - inventory dictionary.
    :param item: str - item to remove from the inventory.
    :return: dict - updated inventory with item removed. Current inventory if item does not match.
    """
    if item in inventory:
        inventory.pop(item)
    return inventory
def list_inventory(inventory):
    """Create a list containing all (item_name, item_count) pairs in inventory.
 
    :param inventory: dict - an inventory dictionary.
    :return: list of tuples - list of key, value pairs from the inventory dictionary.
    """
    for key in list(inventory.keys()):
        if inventory[key] == 0:
            inventory.pop(key)
    return list(inventory.items())