Fix KeyError in inventory.py

When we have no items of type, there are no "count" key in the dict.
PokemonGoF · Aug 11, 2016 · ed2769c · ed2769c
1 parent 42e9d9c
commit ed2769c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pokemongo_bot/inventory.py b/pokemongo_bot/inventory.py
@@ -108,7 +108,10 @@ class Items(_BaseInventoryComponent):
     STATIC_DATA_FILE = os.path.join(_base_dir, 'data', 'items.json')
 
     def count_for(self, item_id):
-        return self._data[item_id]['count']
+        try:
+            return self._data[item_id]['count']
+        except KeyError:
+            return 0
 
     def decrement_count(self, item_id):
         self._data[item_id]["count"] -= 1