Merge pull request #17 from placeTW/shiba

shiba->doge; return either shiba or akita
placeTW · Aug 9, 2023 · d841ec7 · d841ec7
2 parents 6228a60 + f5d47b1
commit d841ec7
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/commands/shiba/random_shiba.py b/commands/shiba/random_shiba.py
@@ -1,19 +1,24 @@
 import discord
 from ..modules.async_utils import _async_get_json
+from random import choice
+
+POSSIBLE_BREEDS = (
+    "https://dog.ceo/api/breed/shiba/images/random",
+    "https://dog.ceo/api/breed/akita/images/random",
+)
 
 
 def register_commands(tree, this_guild: discord.Object):
     @tree.command(
-        name="shiba",
-        description="Shiba",
+        name="doge",
+        description="Random Shiba or Akita image",
         guild=this_guild,
     )
     async def random_shiba(
         interaction: discord.Interaction,
     ):
-        shiba_json = await _async_get_json(
-            "https://dog.ceo/api/breed/shiba/images/random"
-        )
+        link = choice(POSSIBLE_BREEDS)
+        shiba_json = await _async_get_json(link)
         if shiba_json is None or shiba_json["status"] != "success":  # ):
             return await interaction.response.send_message(
                 "Sorry, we couldn't find any dogs ):"