Learnhelp/wdiscordbotserver
1
0
Fork 0
mirror of https://gitlab.com/pancakes1234/wdiscordbotserver.git synced 2025-06-16 07:14:21 -06:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: Update random Teto image command to fetch images from Safebooru and improve error handling

Browse Source
This commit is contained in:
pancakes-proxy 2025-05-19 20:54:40 +09:00
parent f599e26308
commit 8c50b1007a
1 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
cogs/randomteto.py
View File

@ -2,25 +2,45 @@ import discord
from discord.ext import commands
from discord import app_commands
import aiohttp
import random
class TetoImage(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.base_url = "https://safebooru.org/index.php"
@app_commands.command(name="randomtetoimage", description="Gets a random Teto image from slipstreamm.dev")
@app_commands.command(name="randomtetoimage", description="Gets a random Kasane Teto image from Safebooru")
async def random_teto_image(self, interaction: discord.Interaction):
await interaction.response.defer() # Acknowledge the interaction immediately
params = {
"page": "dapi",
"s": "post",
"q": "index",
"tags": "kasane_teto",
"limit": 100, # Request up to 100 images
"json": 1
}
async with aiohttp.ClientSession() as session:
try:
async with session.get("https://slipstreamm.dev/teto") as response:
async with session.get(self.base_url, params=params) as response:
if response.status == 200:
image_url = await response.text()
await interaction.followup.send(image_url)
data = await response.json()
if data:
# Safebooru returns a list of posts, pick a random one
random_post = random.choice(data)
image_url = random_post.get('file_url')
if image_url:
await interaction.followup.send(image_url)
else:
await interaction.followup.send("Could not find a valid image URL in the response.")
else:
await interaction.followup.send("No images found for 'kasane_teto'.")
else:
await interaction.followup.send(f"Failed to fetch image. API returned status: {response.status}")
await interaction.followup.send(f"Failed to fetch images from Safebooru. API returned status: {response.status}")
except aiohttp.ClientError as e:
await interaction.followup.send(f"An error occurred while connecting to the API: {e}")
await interaction.followup.send(f"An error occurred while connecting to Safebooru API: {e}")
except Exception as e:
await interaction.followup.send(f"An unexpected error occurred: {e}")