From 3e1d6b47848a1564d940d4ef1947a020bbdd2a5c Mon Sep 17 00:00:00 2001 From: sgoudham Date: Sat, 8 Aug 2020 00:15:00 +0100 Subject: [PATCH] Checking member/channel in a single line --- cogs/info.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/cogs/info.py b/cogs/info.py index e1a1ac13..86e3e3be 100644 --- a/cogs/info.py +++ b/cogs/info.py @@ -8,7 +8,7 @@ from platform import python_version from time import time from typing import Optional -from discord import Colour, Member +from discord import Colour, Member, TextChannel from discord import Embed from discord import __version__ as discord_version from discord.ext.commands import BucketType, cooldown, bot_has_permissions, guild_only, Cog @@ -137,12 +137,9 @@ class Info(Cog): async def user_info(self, ctx, member: Optional[Member] = None): """User Information! (Created At/Joined/Roles etc)""" - # If a member has been specified, set them as the user - if member: - member = member - # If no member has been specified, choose the author - else: - member = ctx.author + # Use member when mentioned + # Use author if no member is mentioned + member = ctx.author if not member else member # Get the member avatar userAvatar = member.avatar_url @@ -302,11 +299,11 @@ class Info(Cog): @command(name="channelinfo", aliases=["chinfo"]) @guild_only() @bot_has_permissions(embed_links=True) - async def channel_info(self, ctx): + async def channel_info(self, ctx, channel: Optional[TextChannel] = None): """Channel Statistics! (Category/Created At etc)""" # Get the channel that the user is in - channel = ctx.channel + channel = ctx.channel if not channel else channel # Set up Embed embed = Embed(title=f"Statistics For #{channel.name}",