From 1392c1b8cf4a868d904f17c9634da8dfd6e5d2ac Mon Sep 17 00:00:00 2001 From: sgoudham Date: Tue, 7 Jul 2020 10:07:06 +0100 Subject: [PATCH] Reusing code by making functions Increased modularity --- cogs/fun/enso.py | 30 ++++++++++++++++++------------ 1 file changed, 18 insertions(+), 12 deletions(-) diff --git a/cogs/fun/enso.py b/cogs/fun/enso.py index 27887a20..e2d6ac46 100644 --- a/cogs/fun/enso.py +++ b/cogs/fun/enso.py @@ -1,5 +1,4 @@ import asyncio -import datetime import random import string @@ -9,20 +8,29 @@ from discord.ext.commands import cooldown, BucketType, command import settings from cogs.anime.interactive import error_function +from settings import time, colour_list -# Function to display all the images requested of the people -def displayServerImage(array, ctx, name): +# Gets the member and user avatar +def getMember(ctx): # Set member as the author member = ctx.message.author - # Get the member's avatar + # Get the member avatar userAvatar = member.avatar_url + return member, userAvatar + + +# Function to display all the images requested of the people +def displayServerImage(array, ctx, name): + # Get the member and the userAvatar + member, userAvatar = getMember(ctx) + # Set embed up for the person requested by the user embed = Embed( title=f"**Look At What A Cutie {name.capitalize()} is!! **", - colour=Colour(random.choice(settings.colour_list)), - timestamp=datetime.datetime.utcnow()) + colour=Colour(random.choice(colour_list)), + timestamp=time) embed.set_image(url=random.choice(array)) embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar)) @@ -92,16 +100,14 @@ class Enso(commands.Cog): with open(f'images/ServerMembers/{random.choice(array)}.txt') as file: array = file.readlines() - # Set member as the author - member = ctx.message.author - # Get the member's avatar - userAvatar = member.avatar_url + # Get the member and the userAvatar + member, userAvatar = getMember(ctx) # Embed the image in a message and send it to the channel embed = Embed( title=f"Oh Look! A Cute Person ", - colour=Colour(random.choice(settings.colour_list)), - timestamp=datetime.datetime.utcnow()) + colour=Colour(random.choice(colour_list)), + timestamp=time) embed.set_image(url=random.choice(array)) embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))