Init commit

cogs/role_panel.py

@@ -0,0 +1,62 @@
+import discord
+from discord.ext import commands
+from typing import Optional
+import re
+
+from bot import db
+
+
+class Panel(commands.Cog, name="Панель ролей"):
+    """Создание и изменение панели для выбора ролей"""
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.has_permissions(administrator=True)
+    @commands.command()
+    async def add_role_panel(self, ctx):
+        emoji_patter = re.compile(r"(<:\w+:(\d+)>)|([^\s\d\w])", flags=re.IGNORECASE)
+
+
+        def check(mes):
+            return mes.author == ctx.author  # len(mes.role_mentions) == 1
+
+        menu = await ctx.send("Для добавления ролей в меню напишите `эмодзи роль`\n"
+                       "Для удаления роли из меню напишите порядковый номер в списке\n"
+                       "Для подтверждения напишите `.`\n\n"
+                       "*Добавлено:*")
+        while True:
+            mes = await self.bot.wait_for('message', check=check)
+            if mes.content == ".":
+                break
+            elif len(mes.role_mentions) == 1:
+                content = mes.content.split()
+                result = ""
+                if emoji_patter.match(content[0]):
+                    result = f"{content[0]} - {content[1]}"
+                if content[0] is int:
+                    pass
+
+
+        await ctx.send("Done")
+
+    @commands.has_permissions(administrator=True)
+    @commands.command()
+    async def default_role(self, ctx, role: Optional[discord.Role] = None):
+        if role is None:
+            data = db.guild.find_one({"id": ctx.guild.id})
+            if data is not None:
+                if "default_role" in data.keys():
+                    await ctx.send(ctx.guild.get_role(data["default_role"]).mention)
+                else:
+                    await ctx.send("Роли по умолчанию не установлено")
+            else:
+                await ctx.send("Роли по умолчанию не установлено")
+        else:
+            if db.guild.find_one({"id": ctx.guild.id}) is not None:
+                db.guild.update_one({"id": ctx.guild.id}, {"$set": {"default_role": role.id}})
+            else:
+                db.guild.insert_one({"id": ctx.guild.id, "default_role": role.id})
+
+
+async def setup(bot):
+    await bot.add_cog(Panel(bot))