fix bug

alec-jensen · Aug 19, 2024 · 80e20eb · 80e20eb
1 parent ecadde1
commit 80e20eb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kidney-bot/cogs/economy.py b/kidney-bot/cogs/economy.py
@@ -205,11 +205,13 @@ async def bank(self) -> int:
 
     async def inventory(self) -> dict:
         doc = await self.database.currency.find_one({'userID': str(self.user.id)})
+        if type(doc.get('inventory')) is not dict:
+            doc['inventory'] = {}
         return doc['inventory']
 
     async def add_item(self, item: Item):
         doc = await self.database.currency.find_one({'userID': str(self.user.id)})
-        if type(doc['inventory']) is not dict:
+        if type(doc.get('inventory')) is not dict:
             doc['inventory'] = {}
         count = doc['inventory'].get(item.id, 0)
         count += 1