aboutsummaryrefslogtreecommitdiffstats
path: root/bot/exts/evergreen/emoji_count.py
diff options
context:
space:
mode:
Diffstat (limited to 'bot/exts/evergreen/emoji_count.py')
-rw-r--r--bot/exts/evergreen/emoji_count.py97
1 files changed, 0 insertions, 97 deletions
diff --git a/bot/exts/evergreen/emoji_count.py b/bot/exts/evergreen/emoji_count.py
deleted file mode 100644
index cc43e9ab..00000000
--- a/bot/exts/evergreen/emoji_count.py
+++ /dev/null
@@ -1,97 +0,0 @@
-import datetime
-import logging
-import random
-from collections import defaultdict
-from typing import List, Tuple
-
-import discord
-from discord.ext import commands
-
-from bot.constants import Colours, ERROR_REPLIES
-from bot.utils.pagination import LinePaginator
-
-log = logging.getLogger(__name__)
-
-
-class EmojiCount(commands.Cog):
- """Command that give random emoji based on category."""
-
- def __init__(self, bot: commands.Bot):
- self.bot = bot
-
- @staticmethod
- def embed_builder(emoji: dict) -> Tuple[discord.Embed, List[str]]:
- """Generates an embed with the emoji names and count."""
- embed = discord.Embed(
- color=Colours.orange,
- title="Emoji Count",
- timestamp=datetime.datetime.utcnow()
- )
- msg = []
-
- if len(emoji) == 1:
- for category_name, category_emojis in emoji.items():
- if len(category_emojis) == 1:
- msg.append(f"There is **{len(category_emojis)}** emoji in **{category_name}** category")
- else:
- msg.append(f"There are **{len(category_emojis)}** emojis in **{category_name}** category")
- embed.set_thumbnail(url=random.choice(category_emojis).url)
-
- else:
- for category_name, category_emojis in emoji.items():
- emoji_choice = random.choice(category_emojis)
- if len(category_emojis) > 1:
- emoji_info = f"There are **{len(category_emojis)}** emojis in **{category_name}** category"
- else:
- emoji_info = f"There is **{len(category_emojis)}** emoji in **{category_name}** category"
- if emoji_choice.animated:
- msg.append(f'<a:{emoji_choice.name}:{emoji_choice.id}> {emoji_info}')
- else:
- msg.append(f'<:{emoji_choice.name}:{emoji_choice.id}> {emoji_info}')
- return embed, msg
-
- @staticmethod
- def generate_invalid_embed(emojis: list) -> Tuple[discord.Embed, List[str]]:
- """Generates error embed."""
- embed = discord.Embed(
- color=Colours.soft_red,
- title=random.choice(ERROR_REPLIES)
- )
- msg = []
-
- emoji_dict = defaultdict(list)
- for emoji in emojis:
- emoji_dict[emoji.name.split("_")[0]].append(emoji)
-
- error_comp = ', '.join(emoji_dict)
- msg.append(f"These are the valid categories\n```{error_comp}```")
- return embed, msg
-
- @commands.command(name="emojicount", aliases=["ec", "emojis"])
- 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)
-
- if not ctx.guild.emojis:
- await ctx.send("No emojis found.")
- return
- log.trace(f"Emoji Category {'' if category_query else 'not '}provided by the user")
- for emoji in ctx.guild.emojis:
- 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 not emoji_dict:
- log.trace("Invalid name provided by the user")
- embed, msg = self.generate_invalid_embed(ctx.guild.emojis)
- else:
- embed, msg = self.embed_builder(emoji_dict)
- await LinePaginator.paginate(lines=msg, ctx=ctx, embed=embed)
-
-
-def setup(bot: commands.Bot) -> None:
- """Emoji Count Cog load."""
- bot.add_cog(EmojiCount(bot))