Create a slash command for google search

cogs/google.py

@@ -1,10 +1,37 @@
 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
 
 
+def _create_google_embed(results) -> discord.Embed:
+    # Filter out all Nones
+    elements = list(filter(lambda x: x is not None, results))
+
+    embed = discord.Embed(colour=discord.Colour.blue())
+    embed.set_author(name="Google Search")
+
+    # Empty list of results
+    if len(elements) == 0:
+        embed.description = "Geen resultaten gevonden."
+        return embed
+
+    # Cut excess results out
+    if len(elements) > 10:
+        elements = elements[:10]
+
+    links = []
+
+    for index, (link, title) in enumerate(elements):
+        links.append("{}: [{}]({})".format(index + 1, title, link))
+
+    embed.description = "\n".join(links)
+
+    return embed
+
+
 class Google(commands.Cog):
     def __init__(self, client):
         self.client = client
@@ -13,6 +40,26 @@ 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):
+        results, status = google_search(query)
+
+        if results is None:
+            return await interaction.reply("Er ging iets fout (Response {})".format(status))
+
+        embed = _create_google_embed(results)
+        await interaction.reply(embed=embed)
+
+    @slash_command(name="test", description="Test")
+    async def test(self, interaction):
+        await interaction.reply(":eyes:")
+
     @commands.command(name="Google", aliases=["Gtfm", "Search"], usage="[Query]", case_insensitive=True)
     @help.Category(Category.Other)
     async def google(self, ctx, *query):
@@ -24,28 +71,7 @@ class Google(commands.Cog):
         if results is None:
             return await ctx.send("Er ging iets fout (Response {})".format(status))
 
-        # Filter out all Nones
-        elements = list(filter(lambda x: x is not None, results))
-
-        embed = discord.Embed(colour=discord.Colour.blue())
-        embed.set_author(name="Google Search")
-
-        # Empty list of results
-        if len(elements) == 0:
-            embed.description = "Geen resultaten gevonden."
-            return await ctx.reply(embed=embed, mention_author=False)
-
-        # Cut excess results out
-        if len(elements) > 10:
-            elements = elements[:10]
-
-        links = []
-
-        for index, (link, title) in enumerate(elements):
-            links.append("{}: [{}]({})".format(index + 1, title, link))
-
-        embed.description = "\n".join(links)
-
+        embed = _create_google_embed(results)
         await ctx.reply(embed=embed, mention_author=False)
 
 

cogs/school.py

@@ -62,7 +62,7 @@ class School(commands.Cog):
         return await ctx.send(embed=s.create_schedule().to_embed())
 
     @commands.command(name="Pin", usage="[Message]")
-    @help.Category(category=Category.School)
+    @help.Category(category=Category.Other)
     async def pin(self, ctx, message: discord.Message):
         # In case people abuse, check if they're blacklisted
         blacklist = []