import datetime import random import string from typing import Optional from discord import Embed from discord.ext.commands import bot_has_permissions, Cog, group from settings import rndColour, enso_embedmod_colours # Defining the full names of the waifu's/husbando's anime = {"yumeko": "Jabami Yumeko", "toga": "Himiko Toga", "maki": "Maki Oze", "kakashi": "Hatake Kakashi", "tamaki": "Tamaki Suoh", } # Gets the member and user avatar def getMember(msg): # Set member as the author member = msg.author # Get the member avatar userAvatar = member.avatar_url return member, userAvatar # Function to turn the user inputted name into the full name def Abbrev(anime_msg): # Get the lowercase split_anime = anime_msg.split() new_msg = "" # For each word in split_anime for word in split_anime: # If the word exists in the anime array if word in anime: # Set a new string equal to the full name of the waifu/husbando new_msg = anime[word] return new_msg def randomWaifu(msg, waifu): """Return Embed for Specific Waifu""" # Retrieve a random image of a waifu within the bot with open(f'images/AnimeImages/Waifus/{random.choice(waifu)}.txt') as file: array = file.readlines() # Get the member and the userAvatar member, userAvatar = getMember(msg) # Set up the embed for a random waifu image waifu_embed = Embed( title=f"Oh Look! A Wonderful Waifu! ", colour=rndColour(), timestamp=datetime.datetime.utcnow()) waifu_embed.set_image(url=random.choice(array)) waifu_embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar)) return waifu_embed def randomHusbando(msg, husbando): """Display embed for Specific Husbando""" # Retrieve a random image of a husbando within the bot with open(f'images/AnimeImages/Husbandos/{random.choice(husbando)}.txt') as file: array = file.readlines() # Get the member and the userAvatar member, userAvatar = getMember(msg) # Set up the embed for a random husbando image husbando_embed = Embed( title=f"Oh Look! A Handsome Husbando! ", colour=rndColour(), timestamp=datetime.datetime.utcnow()) husbando_embed.set_image(url=random.choice(array)) husbando_embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar)) return husbando_embed def displayAnimeImage(array, msg, name): """Setup Embed for Specific Waifu/Husbando""" # Get the member and the userAvatar member, userAvatar = getMember(msg) # Set up embed for an image relating to a husbando or waifu anime_embed = Embed( title=f"**{name}**", colour=rndColour(), timestamp=datetime.datetime.utcnow()) anime_embed.set_image(url=random.choice(array)) anime_embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar)) return anime_embed def waifus(): """List of Waifu's""" return ["toga", "yumeko", "maki"] def husbandos(): """List of Husbando's""" return ["kakashi", "tamaki"] class Anime(Cog): """Waifus and Husbandos!""" def __init__(self, bot): self.bot = bot @Cog.listener() async def on_ready(self): """Printing out that Cog is ready on startup""" print(f"{self.__class__.__name__} Cog has been loaded!\n-----") @group(invoke_without_command=True) @bot_has_permissions(embed_links=True) async def waifu(self, ctx, waifu: Optional[str] = None): """Shows a Waifu""" if waifu: lcase_waifu = waifu.lower() try: # Retrieve image of the waifu specified with open(f'images/AnimeImages/Waifus/{lcase_waifu}.txt') as file: w_array = file.readlines() # Get the full name of the waifu full_name = Abbrev(lcase_waifu) # Embed the image into a message and send it to the channel embed = displayAnimeImage(w_array, ctx, full_name) await ctx.send(embed=embed) except FileNotFoundError as e: print(e) # Send error message saying that the Waifu isn't recognised embed = Embed(description="Sorry! That Waifu Doesn't Exist!" "\nPlease Do **{}waifu list** To View The List Of Waifu's Available" .format(ctx.prefix), colour=enso_embedmod_colours) await ctx.send(embed=embed) else: # Get embed from randomWaifu() and send it to the channel embed = randomWaifu(ctx, waifus()) await ctx.send(embed=embed) @group(invoke_without_command=True) @bot_has_permissions(embed_links=True) async def husbando(self, ctx, husbando: Optional[str] = None): """Shows a Husbando""" if husbando: # Get the lowercase lcase_husbando = husbando.lower() try: # Retrieve image of the husbando specified with open(f'images/AnimeImages/Husbandos/{lcase_husbando}.txt') as file: h_array = file.readlines() # Get the full name of the husbando full_name = Abbrev(lcase_husbando) # Embed the image into a message and send it to the channel embed = displayAnimeImage(h_array, ctx, full_name) await ctx.send(embed=embed) except FileNotFoundError as e: print(e) # Send error message saying that the Husbando isn't recognised embed = Embed(description="Sorry! That Husbando Doesn't Exist!" "\nPlease Do **{}husbando list** To View The List Of Husbando's Available" .format(ctx.prefix), colour=enso_embedmod_colours) await ctx.send(embed=embed) else: # Get embed from randomHusbando() and send it to the channel embed = randomHusbando(ctx, husbandos()) await ctx.send(embed=embed) @waifu.command(name="list") async def wlist(self, ctx): """Returns a list of Waifu's Available""" # Send the list of waifus in the bot to the channel waifu_list = string.capwords(', '.join(map(str, waifus()))) # Tell the user to try the waifus in the array embed = Embed(description="Try The Waifu's Listed Below!" "\n**{}**".format(waifu_list), colour=enso_embedmod_colours) await ctx.send(embed=embed) @husbando.command(name="list") async def hlist(self, ctx): """Returns a list of Husbando's Available""" # Send the list of waifus in the bot to the channel husbando_list = string.capwords(', '.join(map(str, husbandos()))) # Tell the user to try the husbando's in the array embed = Embed(description="Try The Husbando's Listed Below!" "\n**{}**".format(husbando_list), colour=enso_embedmod_colours) await ctx.send(embed=embed) def setup(bot): bot.add_cog(Anime(bot))