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: Enhance Teto image handling with interactive buttons and improved image fetching

Browse Source
This commit is contained in:
pancakes-proxy 2025-05-19 22:17:17 -07:00
parent 9ebbb17eb9
commit 2f0431ff22
1 changed files with 77 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
cogs/randomteto.py
View File

@ -4,45 +4,101 @@ from discord import app_commands
import aiohttp
import random
class TetoImageView(discord.ui.View):
def __init__(self, image_urls, current_index, history, message_id, bot):
super().__init__(timeout=180)
self.image_urls = image_urls
self.current_index = current_index
self.history = history
self.message_id = message_id
self.bot = bot
async def update_message(self, interaction):
embed = discord.Embed(
title="Random Kasane Teto Image",
color=discord.Color.pink()
)
embed.set_image(url=self.image_urls[self.current_index])
embed.set_footer(text=f"Image {self.current_index + 1} of {len(self.image_urls)}")
await interaction.response.edit_message(embed=embed, view=self)
@discord.ui.button(label="Previous", style=discord.ButtonStyle.secondary, custom_id="teto_prev")
async def previous(self, interaction: discord.Interaction, button: discord.ui.Button):
if self.current_index > 0:
self.current_index -= 1
await self.update_message(interaction)
else:
await interaction.response.defer()
@discord.ui.button(label="Next", style=discord.ButtonStyle.secondary, custom_id="teto_next")
async def next(self, interaction: discord.Interaction, button: discord.ui.Button):
if self.current_index < len(self.image_urls) - 1:
self.current_index += 1
await self.update_message(interaction)
else:
await interaction.response.defer()
@discord.ui.button(label="New Random", style=discord.ButtonStyle.primary, custom_id="teto_new")
async def new_random(self, interaction: discord.Interaction, button: discord.ui.Button):
# Fetch new images and reset history
cog = self.bot.get_cog("TetoImage")
if cog:
new_urls = await cog.fetch_teto_images()
if new_urls:
self.image_urls = new_urls
self.current_index = 0
self.history[self.message_id] = (self.image_urls, self.current_index)
await self.update_message(interaction)
else:
await interaction.response.send_message("Failed to fetch new images.", ephemeral=True)
else:
await interaction.response.send_message("Cog not found.", ephemeral=True)
class TetoImage(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.base_url = "https://safebooru.org/index.php"
self.history = {} # message_id: (image_urls, current_index)
@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
async def fetch_teto_images(self):
params = {
"page": "dapi",
"s": "post",
"q": "index",
"tags": "kasane_teto",
"limit": 100, # Request up to 100 images
"limit": 100,
"json": 1
}
async with aiohttp.ClientSession() as session:
try:
async with session.get(self.base_url, params=params) as response:
if response.status == 200:
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 images from Safebooru. API returned status: {response.status}")
except aiohttp.ClientError as 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}")
urls = [post.get('file_url') for post in data if post.get('file_url')]
return urls
except Exception:
pass
return []
@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()
image_urls = await self.fetch_teto_images()
if not image_urls:
await interaction.followup.send("No images found for 'kasane_teto'.")
return
current_index = random.randint(0, len(image_urls) - 1)
embed = discord.Embed(
title="Random Kasane Teto Image",
color=discord.Color.pink()
)
embed.set_image(url=image_urls[current_index])
embed.set_footer(text=f"Image {current_index + 1} of {len(image_urls)}")
view = TetoImageView(image_urls, current_index, self.history, None, self.bot)
msg = await interaction.followup.send(embed=embed, view=view)
view.message_id = msg.id
self.history[msg.id] = (image_urls, current_index)
async def setup(bot):
await bot.add_cog(TetoImage(bot))