Merge branch 'info_commands' into 'master'

Info commands

See merge request python-discord/projects/bot!32

5 files changed, 146 insertions, 11 deletions

diff --git a/bot/__main__.py b/bot/__main__.py
index 9a2ef9319..6790da79e 100644
--- a/bot/__main__.py
+++ b/bot/__main__.py
@@ -5,7 +5,7 @@ from aiohttp import AsyncResolver, ClientSession, TCPConnector
 from discord import Game
 from discord.ext.commands import Bot, when_mentioned_or
 
-from bot.constants import Bot as BotConfig  # , ClickUp
+from bot.constants import Bot as BotConfig, DEBUG_MODE
 from bot.utils.service_discovery import wait_for_rmq
 
 
@@ -38,7 +38,6 @@ else:
 
 # Internal/debug
 bot.load_extension("bot.cogs.logging")
-bot.load_extension("bot.cogs.modlog")
 bot.load_extension("bot.cogs.security")
 bot.load_extension("bot.cogs.events")
 
@@ -49,19 +48,19 @@ bot.load_extension("bot.cogs.bot")
 bot.load_extension("bot.cogs.clean")
 bot.load_extension("bot.cogs.cogs")
 
-# Local setups usually don't have the clickup key set,
-# and loading the cog would simply spam errors in the console.
-# if ClickUp.key is not None:
-#     bot.load_extension("bot.cogs.clickup")
-# else:
-#     log.info("`CLICKUP_KEY` not set in the environment, not loading the ClickUp cog.")
+# Only load this in production
+if not DEBUG_MODE:
+    bot.load_extension("bot.cogs.modlog")
+    bot.load_extension("bot.cogs.verification")
 
+# Feature cogs
 bot.load_extension("bot.cogs.deployment")
 bot.load_extension("bot.cogs.defcon")
 bot.load_extension("bot.cogs.doc")
 bot.load_extension("bot.cogs.eval")
 bot.load_extension("bot.cogs.fun")
 bot.load_extension("bot.cogs.hiphopify")
+bot.load_extension("bot.cogs.information")
 bot.load_extension("bot.cogs.moderation")
 bot.load_extension("bot.cogs.off_topic_names")
 bot.load_extension("bot.cogs.snakes")
@@ -69,7 +68,6 @@ bot.load_extension("bot.cogs.snekbox")
 bot.load_extension("bot.cogs.tags")
 bot.load_extension("bot.cogs.token_remover")
 bot.load_extension("bot.cogs.utils")
-bot.load_extension("bot.cogs.verification")
 
 if has_rmq:
     bot.load_extension("bot.cogs.rmq")
diff --git a/bot/cogs/information.py b/bot/cogs/information.py
new file mode 100644
index 000000000..f61b130a6
--- /dev/null
+++ b/bot/cogs/information.py
@@ -0,0 +1,126 @@
+import logging
+import textwrap
+from datetime import datetime
+
+from dateutil.relativedelta import relativedelta
+from discord import CategoryChannel, Colour, Embed, TextChannel, VoiceChannel
+from discord.ext.commands import Bot, Context, command
+
+from bot.constants import Emojis
+from bot.utils.time import humanize
+
+log = logging.getLogger(__name__)
+
+
+class Information:
+    """
+    A cog with commands for generating embeds with
+    server information, such as server statistics
+    and user information.
+    """
+
+    def __init__(self, bot: Bot):
+        self.bot = bot
+
+    @command(name="roles")
+    async def roles_info(self, ctx: Context):
+        """
+        Returns a list of all roles and their
+        corresponding IDs.
+        """
+
+        # Sort the roles alphabetically and remove the @everyone role
+        roles = sorted(ctx.guild.roles, key=lambda role: role.name)
+        roles = [role for role in roles if role.name != "@everyone"]
+
+        # Build a string
+        role_string = ""
+        for role in roles:
+            role_string += f"`{role.id}` - {role.mention}\n"
+
+        # Build an embed
+        embed = Embed(
+            title="Role information",
+            colour=Colour.blurple(),
+            description=role_string
+        )
+
+        embed.set_footer(text=f"Total roles: {len(roles)}")
+
+        await ctx.send(embed=embed)
+
+    @command(name="server", aliases=["server_info", "guild", "guild_info"])
+    async def server_info(self, ctx: Context):
+        """
+        Returns an embed full of
+        server information.
+        """
+
+        now = datetime.now()
+        created_delta = relativedelta(now, ctx.guild.created_at)
+        created = humanize(created_delta, accuracy="days")
+        features = ", ".join(ctx.guild.features)
+        region = ctx.guild.region
+
+        # How many of each type of channel?
+        roles = len(ctx.guild.roles)
+        channels = ctx.guild.channels
+        text_channels = 0
+        category_channels = 0
+        voice_channels = 0
+        for channel in channels:
+            if type(channel) == TextChannel:
+                text_channels += 1
+            elif type(channel) == CategoryChannel:
+                category_channels += 1
+            elif type(channel) == VoiceChannel:
+                voice_channels += 1
+
+        # How many of each user status?
+        member_count = ctx.guild.member_count
+        members = ctx.guild.members
+        online = 0
+        dnd = 0
+        idle = 0
+        offline = 0
+        for member in members:
+            if str(member.status) == "online":
+                online += 1
+            elif str(member.status) == "offline":
+                offline += 1
+            elif str(member.status) == "idle":
+                idle += 1
+            elif str(member.status) == "dnd":
+                dnd += 1
+
+        embed = Embed(
+            colour=Colour.blurple(),
+            description=textwrap.dedent(f"""
+                **Server information**
+                Created: {created} ago
+                Voice region: {region}
+                Features: {features}
+
+                **Counts**
+                Members: {member_count}
+                Roles: {roles}
+                Text: {text_channels}
+                Voice: {voice_channels}
+                Channel categories: {category_channels}
+
+                **Members**
+                {Emojis.status_online} {online}
+                {Emojis.status_idle} {idle}
+                {Emojis.status_dnd} {dnd}
+                {Emojis.status_offline} {offline}
+            """)
+        )
+
+        embed.set_thumbnail(url=ctx.guild.icon_url)
+
+        await ctx.send(embed=embed)
+
+
+def setup(bot):
+    bot.add_cog(Information(bot))
+    log.info("Cog loaded: Information")
diff --git a/bot/cogs/moderation.py b/bot/cogs/moderation.py
index 585bba6a6..0a0bbae53 100644
--- a/bot/cogs/moderation.py
+++ b/bot/cogs/moderation.py
@@ -663,8 +663,8 @@ def parse_rfc1123(time_str):
 def _silent_exception(future):
     try:
         future.exception()
-    except Exception as e:
-        log.debug(f"_silent_exception silenced the following exception: {e}")
+    except Exception:  # noqa: S110
+        pass
 
 
 def setup(bot):
diff --git a/bot/constants.py b/bot/constants.py
index 68155e800..a980de15d 100644
--- a/bot/constants.py
+++ b/bot/constants.py
@@ -217,6 +217,12 @@ class Emojis(metaclass=YAMLGetter):
     green_chevron: str
     red_chevron: str
     white_chevron: str
+    lemoneye2: str
+
+    status_online: str
+    status_offline: str
+    status_idle: str
+    status_dnd: str
 
     bullet: str
     new: str
diff --git a/config-default.yml b/config-default.yml
index 8f90b71ab..dfc9fe306 100644
--- a/config-default.yml
+++ b/config-default.yml
@@ -26,6 +26,11 @@ style:
         white_chevron: "<:whitechevron:418110396973711363>"
         lemoneye2:     "<:lemoneye2:435193765582340098>"
 
+        status_online:  "<:status_online:470326272351010816>"
+        status_idle:    "<:status_idle:470326266625785866>"
+        status_dnd:     "<:status_dnd:470326272082313216>"
+        status_offline: "<:status_offline:470326266537705472>"
+
         bullet:   "\u2022"
         pencil:   "\u270F"
         new:      "\U0001F195"