from interactions import Extension, slash_command, SlashContext, EmbedAttachment, EmbedFooter, EmbedField, Embed import requests from random import choice from urllib.parse import quote from datetime import datetime, timedelta tag_blacklist = ["loli", "shota", "gore", "guro", "scat", "3d", "futanari", "animated", "video"] timeout = datetime.now() class YuriExtension(Extension): @slash_command(name="yuri", description="Pošle náhodný yuri obrázek z Gelbooru (NSFW pouze v NSFW kanálech)") async def test(self, ctx: SlashContext): global timeout await ctx.defer() if datetime.now() < timeout: await ctx.respond(ephemeral="Počkej chvíli než pošleš tenhle příkaz znovu") return timeout = datetime.now() + timedelta(seconds=5) p = "yuri sort:random" for tag in tag_blacklist: p += f" -{tag}" if not ctx.channel.nsfw: p += " rating:general score:>100" else: p += " rating:explicit score:>100" r = requests.get("https://gelbooru.com/index.php?page=dapi&s=post&q=index&api_key=anonymous&user_id=9455&json" f"=1&tags={quote(p)}", headers={"User-Agent": "Maplebot/1.0.0"}) if r.status_code > 399: print("Error while requesting from Gelbooru!") print(r.status_code) print(r.json()) await ctx.respond(ephemeral=True, content="Chyba při zpracování požadavku") return posts = r.json()["post"] pick = choice(posts) while pick["has_children"] is True and pick["rating"] != "general": pick = choice(posts) tags = pick["tags"] if len(tags) >= 4096: tags = tags[0:4093] + "..." await ctx.respond(embed=Embed( title="Čáry máry!", url=f"https://gelbooru.com/index.php?page=post&s=view&id={pick['id']}", images=[EmbedAttachment(url=pick["file_url"])], description=tags, footer=EmbedFooter(text="Skrz gelbooru.com", icon_url="https://gelbooru.com/favicon.png") ) )