Add inventory DB cog and fix music cog

cogs/inventory/inventoryCog.py

@@ -0,0 +1,61 @@
+import os
+import discord
+import DiscordUtils
+import asyncio
+from google.cloud import firestore
+from discord.ext import commands
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class InventoryCog(commands.Cog):
+
+    def __init__(self, client):
+        self.client = client
+    # Commands
+
+    # Get a single item from inventory
+    @commands.command(aliases=['inv'])
+    @commands.is_owner()
+    async def inventory(self, ctx, arg1):
+        """Get a single item from the inventory database"""
+
+        # Make single input embed
+        def single_embed(item, id):
+            embed = discord.Embed(title=item["item_name"], description=id)
+            if item["item_type"] != "": embed.add_field(name="Item Type", value=item["item_type"], inline=True)
+            if item["color"] != "": embed.add_field(name="Color", value=item["color"], inline=True)
+            if item["brand"] != "": embed.add_field(name="Brand", value=item["brand"], inline=True)
+            if item["model"] != "": embed.add_field(name="Model", value=item["model"], inline=True)
+            if item["size"] != "": embed.add_field(name="Size", value=item["size"], inline=True)
+            if item["quantity"] != "": embed.add_field(name="Quantity", value=item["quantity"], inline=True)
+            if item["box"] != "": embed.add_field(name="Box", value=item["box"], inline=True)
+            return embed
+
+        # Connect to Firestore DB inventory collection and get the item arg1 as doc
+        if "INVENTORY_PROJECT_ID" in os.environ and "GOOGLE_APPLICATION_CREDENTIALS" in os.environ:
+            try:
+                db = firestore.AsyncClient(project=os.environ.get("INVENTORY_PROJECT_ID"))
+                inventory_ref = db.collection("inventory").document(arg1)
+                doc = await inventory_ref.get()
+
+                if doc.exists:
+                    await ctx.send(embed=single_embed(doc.to_dict(), doc.id))
+                else:
+                    await ctx.send("That ID doesn't exist!")
+            except:
+                await ctx.send("The DB connection is not working.")
+        elif "INVENTORY_PROJECT_ID" not in os.environ and "GOOGLE_APPLICATION_CREDENTIALS" in os.environ:
+            await ctx.send("Inventory Project ID not found!")
+        elif "GOOGLE_APPLICATION_CREDENTIALS" not in os.environ:
+            await ctx.send("GAPP Credentials not found!")
+
+    @inventory.error
+    async def inventory_error(self, ctx, error):
+        if isinstance(error, commands.MissingRequiredArgument):
+            await ctx.send("Missing required argument! (e.g. d!inventory __10000__)")
+        if isinstance(error, commands.NotOwner):
+            await ctx.send("You do not have permissions to run this command.")
+
+def setup(client):
+    client.add_cog(InventoryCog(client))