phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed bug where command would not work in text channels

Browse source
This commit is contained in:
phixxy 2024-01-22 15:08:07 -08:00
parent df8c5c5ecd
commit f7f9ddef68
1 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
extensions/waifu.py
View file

@ -18,12 +18,12 @@ class Waifu(commands.Cog):
resp_data = await resp.json()
try:
image = resp_data['images'][0]
return image['url'], image['is_nsfw']
return image['url']
except:
if resp_data['detail'] == "No image found matching the criteria given.":
return "No image found matching the criteria given.", "False"
return "No image found matching the criteria given."
else:
return "Something went wrong", "False"
return "Something went wrong"
@commands.command()
async def waifu(self, ctx, nsfw=""):
@ -31,15 +31,14 @@ class Waifu(commands.Cog):
tag = random.choice(["ero", "ass", "hentai", "milf", "oral", "paizuri", "ecchi"])
else:
tag = random.choice(["waifu", "maid", "selfies", "uniform"])
image_url, nsfw = await self.get_waifu(tag)
image_url = await self.get_waifu(tag)
if ctx.channel.type == discord.ChannelType["private"]:
await ctx.send(image_url)
elif not ctx.channel.is_nsfw() and nsfw == "False":
await ctx.send(image_url)
elif ctx.channel.is_nsfw() and nsfw == "True":
await ctx.send(image_url)
else:
elif not ctx.channel.is_nsfw() and nsfw.lower() == "nsfw":
await ctx.send("Cannot post NSFW images in this channel.")
else:
await ctx.send(image_url)
async def setup(bot):
try: