63 lines
2.5 KiB
Python
Executable File
63 lines
2.5 KiB
Python
Executable File
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))
|