Added base cog files

2022-07-02 09:25:20 +03:00
parent 3456faed4e
commit 216ab23f5b
2 changed files with 73 additions and 0 deletions
--- a/cogs/general.py
+++ b/cogs/general.py
@@ -0,0 +1,57 @@
+import logging
+
+import discord
+from discord.ext import commands
+from dislash import Option, OptionType
+
+from test import DB
+
+
+class General(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot  # defining bot as global var in class
+
+    @commands.Cog.listener()  # this is a decorator for events/listeners
+    async def on_ready(self):
+        logging.info(f'Cog {__name__.split(".")[1]} is ready!.')
+
+    @cog_ext.cog_slash(
+        name="info",
+        description="Read list of tracks for user",
+        options=[
+            Option("user", "Specify any user", OptionType.USER),
+        ]
+    )
+    async def info(ctx, user=None):
+        user = user or ctx.author
+        audio = DB.read_db(ctx.guild.id, user.id)
+        rolelist = [r.mention for r in user.roles if r != ctx.guild.default_role]
+        if rolelist:
+            roles = "\n".join(rolelist)
+        else:
+            roles = "Not added any role"
+
+        if audio is None:
+            audios = "Not selected audio"
+        else:
+            audios = "• " + "\n• ".join(sorted(audio.split(", ")))
+
+        emb = discord.Embed(
+            title=f"General information",
+            description=f"General information on server about {user}",
+            icon=user.avatar_url
+        )
+        emb.set_thumbnail(url=user.avatar_url)
+        emb.add_field(name="General info",
+                      value=f"Username: {user}\n"
+                            f"Nickname: {user.nick}\n"
+                            f"Joined at: {user.joined_at.strftime('%A, %B %d %Y @ %H:%M:%S')}", inline=False)
+        emb.add_field(name="Audio list", value=f"{audios}", inline=True)
+        emb.add_field(name="Roles list", value=f"{roles}", inline=True)
+        emb.set_footer(text="Information requested by: {}".format(ctx.author.display_name))
+
+        await ctx.reply(embed=emb, ephemeral=True)
+
+
+def setup(bot):  # a extension must have a setup function
+    bot.add_cog(General(bot))  # adding a cog