aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Anubhav1603 <[email protected]>2020-10-05 17:07:42 +0530
committerGravatar Anubhav1603 <[email protected]>2020-10-05 17:07:42 +0530
commit0397361cc16ebc646d83d38b9dc8411df73e871d (patch)
tree510b956531be7b537e2564e96e797990fe497aad
parentcorrected annotation (diff)
added 1 logging and modified for loop in emoji_count
-rw-r--r--bot/exts/evergreen/emoji_count.py16
1 files changed, 9 insertions, 7 deletions
diff --git a/bot/exts/evergreen/emoji_count.py b/bot/exts/evergreen/emoji_count.py
index c030fa75..4999fc15 100644
--- a/bot/exts/evergreen/emoji_count.py
+++ b/bot/exts/evergreen/emoji_count.py
@@ -60,19 +60,21 @@ class EmojiCount(commands.Cog):
return embed, msg
@commands.command(name="emoji_count", aliases=["ec"])
- async def emoji_count(self, ctx: commands.Context, *, emoji_category: str = None) -> None:
+ async def emoji_count(self, ctx: commands.Context, *, category_query: str = None) -> None:
"""Returns embed with emoji category and info given by the user."""
emoji_dict = defaultdict(list)
+ log.trace(f"Emoji Category {'' if category_query else 'not '}provided by the user")
for emoji in ctx.guild.emojis:
- if emoji_category is None:
- log.trace("Emoji Category not provided by the user")
- emoji_dict[emoji.name.split("_")[0]].append(emoji)
- elif emoji.name.split("_")[0] in emoji_category:
- log.trace("Emoji Category provided by the user")
- emoji_dict[emoji.name.split("_")[0]].append(emoji)
+ emoji_category = emoji.name.split("_")[0]
+
+ if category_query is not None and emoji_category not in category_query:
+ continue
+
+ emoji_dict[emoji_category].append(emoji)
if len(emoji_dict) == 0:
+ log.trace("Invalid name provided by the user")
embed, msg = self.generate_invalid_embed(ctx)
else:
embed, msg = self.embed_builder(emoji_dict)