Fixed tree.command request timeout

discord_bot.py

@@ -26,36 +26,36 @@ async def greet(interaction: discord.Interaction, name: str):
 
 @tree.command(name="get-kudos", description="The amount of Kudos this user has.")
 async def getKudos(interaction: discord.Interaction):
-    details = horde.getUserDetails()
-    if "kudos" not in details:
-        await interaction.response.send_message(f'Error: {details["code"]} {details["reason"]} {details["rc"]}')
-        return
-    await interaction.response.send_message(f'The amount of Kudos this user has is {details["kudos"]}')
+    async with horde.getUserDetails() as details:
+        if "kudos" not in details:
+            await interaction.response.send_message(f'Error: {details["code"]} {details["reason"]} {details["rc"]}')
+            return
+        await interaction.response.send_message(f'The amount of Kudos this user has is {details["kudos"]}')
 
 @tree.command(name="generate-status", description="Retrieve the status of an Asynchronous generation request.")
 async def generateStatus(interaction: discord.Interaction, id: str):
-    details = horde.generateCheck(id)
-    if "kudos" not in details:
-        await interaction.response.send_message(f'Check Error: {details["code"]} {details["reason"]} {details["rc"]}')
-        return
-    if bool(details["is_possible"]) == False:
-        await interaction.response.send_message("This generation is impossible.")
-        return
-    if bool(details["faulted"]) == True:
-        await interaction.response.send_message("This generation is faulted.")
-        return
-    if bool(details["done"]) == True:
-        generationDetail = horde.generateStatus(id)
-        if "generations" not in generationDetail:
-            await interaction.response.send_message(f'Status Error: {generationDetail["code"]} {generationDetail["reason"]} {generationDetail["rc"]}')
-        for i in range(len(generationDetail["generations"])):
-            await interaction.response.send_message(generationDetail["generations"][i]["img"])
-        return
-    if int(details["processing"]) > 0:
-        total = int(details["finished"]) + int(details["processing"]) + int(details["queue_position"]) + int(details["restarted"]) + int(details["waiting"])
-        await interaction.response.send_message(f'Processing image: {details["processing"]}/{total}')
-        return
-    await interaction.response.send_message(f'Position in queue: {details["queue_position"]}, wait time: {details["wait_time"]}s')
+    async with horde.generateCheck(id) as details:
+        if "kudos" not in details:
+            await interaction.response.send_message(f'Check Error: {details["code"]} {details["reason"]} {details["rc"]}')
+            return
+        if bool(details["is_possible"]) == False:
+            await interaction.response.send_message("This generation is impossible.")
+            return
+        if bool(details["faulted"]) == True:
+            await interaction.response.send_message("This generation is faulted.")
+            return
+        if bool(details["done"]) == True:
+            async with horde.generateStatus(id) as generationDetail:
+                if "generations" not in generationDetail:
+                    await interaction.response.send_message(f'Status Error: {generationDetail["code"]} {generationDetail["reason"]} {generationDetail["rc"]}')
+                for i in range(len(generationDetail["generations"])):
+                    await interaction.response.send_message(generationDetail["generations"][i]["img"])
+            return
+        if int(details["processing"]) > 0:
+            total = int(details["finished"]) + int(details["processing"]) + int(details["queue_position"]) + int(details["restarted"]) + int(details["waiting"])
+            await interaction.response.send_message(f'Processing image: {details["processing"]}/{total}')
+            return
+        await interaction.response.send_message(f'Position in queue: {details["queue_position"]}, wait time: {details["wait_time"]}s')
 
 @bot.command()
 async def ping(ctx):

horde.py

@@ -34,15 +34,15 @@ def getResponseByPost(url):
             "rc": response.rc
         }
 
-def getUserDetails():
+async def getUserDetails():
     url = f"{baseUrl}find_user"
     return getResponseByGet(url)
 
-def generateCheck(id):
+async def generateCheck(id):
     url = f"{baseUrl}generate/check/{id}"
     return getResponseByGet(url)
 
-def generateStatus(id):
+async def generateStatus(id):
     url = f"{baseUrl}generate/status/{id}"
     return getResponseByGet(url)