diff --git a/fantasy-game-inventory/game-inventory.py b/fantasy-game-inventory/game-inventory.py index c7ce94a..de3b804 100644 --- a/fantasy-game-inventory/game-inventory.py +++ b/fantasy-game-inventory/game-inventory.py @@ -29,13 +29,21 @@ def addToInventory(inventory, addedItems): Returns: updatedInventory (dict): Inventory containing updated items and their counts """ - updatedInventory = dict(inventory) - # your code goes here + # YOUR CODE GOES HERE + newInventory = dict(inventory) for item in addedItems: - updatedInventory.setdefault(item, 0) - updatedInventory[item] += 1 - - return updatedInventory + newInventory.setdefault(item, 0) + + if item not in inventory.keys(): + inventory[item] = 0 + newInventory = inventory + for i in newInventory.keys(): + if i in addedItems: + newInventory[i] += 1 + + elif item in inventory.keys(): + newInventory[item] += 1 + return newInventory @@ -47,4 +55,4 @@ def addToInventory(inventory, addedItems): inv = {'gold coin': 42, 'rope': 1} dragonLoot = ['gold coin', 'dagger', 'gold coin', 'gold coin', 'ruby'] inv = addToInventory(inv, dragonLoot) - displayInventory(inv) \ No newline at end of file + displayInventory(inv)