Separate loading slash commands from normal commands, split slash commands to separate cogs, make init_extensions support directories

cogs/google.py

@@ -1,30 +1,8 @@
 import discord
 from discord.ext import commands
-from dislash import slash_command, SlashInteraction, Option, OptionType
 from decorators import help
 from enums.help_categories import Category
-from functions.scrapers.google import google_search, SearchResult
-
-
-def _create_google_embed(result: SearchResult) -> discord.Embed:
-    embed = discord.Embed(colour=discord.Colour.blue())
-    embed.set_author(name="Google Search")
-
-    # Empty list of results
-    if len(result.results) == 0:
-        embed.colour = discord.Colour.red()
-        embed.description = "Geen resultaten gevonden."
-        return embed
-
-    # Add results into a field
-    links = []
-
-    for index, link in enumerate(result.results):
-        links.append(f"{index + 1}: {link}")
-
-    embed.description = "\n".join(links)
-
-    return embed
+from functions.scrapers.google import google_search, create_google_embed
 
 
 class Google(commands.Cog):
@@ -35,22 +13,6 @@ class Google(commands.Cog):
     def cog_check(self, ctx):
         return not self.client.locked
 
-    @slash_command(name="google",
-                   description="Google search",
-                   options=[
-                     Option("query", "Search query", OptionType.STRING, required=True)
-                   ],
-                   guild_ids=[880175869841277008]
-                   )
-    async def _google_slash(self, interaction: SlashInteraction, query: str):
-        result = google_search(query)
-
-        if not result.results:
-            return await interaction.reply("Er ging iets fout (Response {})".format(result.status_code))
-
-        embed = _create_google_embed(result)
-        await interaction.reply(embed=embed)
-
     @commands.command(name="Google", aliases=["Gtfm", "Search"], usage="[Query]", case_insensitive=True)
     @help.Category(Category.Other)
     async def google(self, ctx, *query):
@@ -62,7 +24,7 @@ class Google(commands.Cog):
         if not result.results:
             return await ctx.send("Er ging iets fout (Response {})".format(result.status_code))
 
-        embed = _create_google_embed(result)
+        embed = create_google_embed(result)
         await ctx.reply(embed=embed, mention_author=False)
 
 

cogs/slash/google_slash.py

@@ -0,0 +1,30 @@
+from discord.ext import commands
+from dislash import slash_command, SlashInteraction, Option, OptionType
+from functions.scrapers.google import google_search, create_google_embed
+from startup.didier import Didier
+
+
+class GoogleSlash(commands.Cog):
+    def __init__(self, client: Didier):
+        self.client: Didier = client
+
+    @slash_command(name="google",
+                   description="Google search",
+                   options=[
+                     Option("query", "Search query", OptionType.STRING, required=True)
+                   ],
+                   guild_ids=[728361030404538488, 880175869841277008]
+                   )
+    async def _google_slash(self, interaction: SlashInteraction, query: str):
+        result = google_search(query)
+
+        if not result.results:
+            return await interaction.reply("Er ging iets fout (Response {})".format(result.status_code))
+
+        embed = create_google_embed(result)
+        print("got here")
+        await interaction.reply(embed=embed)
+
+
+def setup(client: Didier):
+    client.add_cog(GoogleSlash(client))