mirror of
https://github.com/zoriya/Nemo.git
synced 2025-12-06 02:46:10 +00:00
270 lines
9.3 KiB
Python
270 lines
9.3 KiB
Python
#!/usr/bin/env python3
|
|
import asyncio
|
|
|
|
import nemo
|
|
import helper
|
|
import config
|
|
import discord
|
|
from discord.ext.commands import has_permissions
|
|
|
|
number_emojis = ["1⃣", "2⃣", "3⃣", "4⃣", "5⃣", "6⃣", "7⃣", "8⃣", "9⃣"]
|
|
|
|
nemo = nemo.Nemo()
|
|
|
|
|
|
@nemo.command("!")
|
|
@has_permissions(administrator=True)
|
|
async def setup(*, message: discord.Message, guild: discord.Guild, **_):
|
|
if any(x.name == config.CATEGORY_NAME and x.type == discord.ChannelType.category for x in guild.channels):
|
|
return
|
|
category = await guild.create_category(config.CATEGORY_NAME)
|
|
organization = await guild.create_text_channel(config.ORGANIZATION_NAME, overwrites={
|
|
guild.default_role: discord.PermissionOverwrite(send_messages=False),
|
|
guild.me: discord.PermissionOverwrite(send_messages=True)
|
|
}, category=category)
|
|
await help_msg(organization)
|
|
await message.delete()
|
|
|
|
|
|
@nemo.command("!help")
|
|
@has_permissions(administrator=True)
|
|
@helper.auto_delete
|
|
async def help_command(*, message: discord.Message, **_):
|
|
await help_msg(message.channel)
|
|
|
|
|
|
async def help_msg(channel: discord.TextChannel):
|
|
await channel.send(config.HELP_MSG.replace("@Me", f"<@{nemo.user.id}>"))
|
|
await channel.send(config.EVENT_LIST)
|
|
msg = await channel.send(config.CREATE_MSG)
|
|
await msg.add_reaction("✅")
|
|
|
|
|
|
@nemo.reaction(lambda reaction, member: reaction.message.author.bot and config.LIST_KEY in reaction.message.content)
|
|
async def join_event(*,
|
|
reaction: discord.Reaction,
|
|
member: discord.Member,
|
|
guild: discord.Guild,
|
|
channel: discord.TextChannel,
|
|
**_):
|
|
try:
|
|
event: int = number_emojis.index(reaction.emoji) + 1
|
|
except ValueError:
|
|
return
|
|
role: str = discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}")
|
|
if role is None or role in member.roles:
|
|
return
|
|
|
|
list_msg: discord.Message = [x async for x in channel.history() if config.LIST_KEY in x.content][0]
|
|
try:
|
|
status = list_msg.content.split('\n')[event + 1]
|
|
except IndexError:
|
|
return
|
|
if config.EMPTY_SLOT in status or config.EVENT_CONFIGURING_KEY in status or config.PRIVATE_EVENT in status:
|
|
return
|
|
|
|
await member.add_roles(role)
|
|
event_channel: discord.TextChannel = [x for x in guild.channels if x.name[:1] == str(event)][0]
|
|
await event_channel.send(config.EVENT_JOIN.replace("@User", f"<@{member.id}>"))
|
|
|
|
|
|
@nemo.reaction(lambda reaction, member: reaction.message.author.bot and config.CREATE_KEY in reaction.message.content)
|
|
async def create_event(reaction: discord.Reaction,
|
|
message: discord.Message,
|
|
member: discord.Member,
|
|
guild: discord.Guild, **_):
|
|
list_msg: discord.Message = [x async for x in message.channel.history() if config.LIST_KEY in x.content][0]
|
|
index = get_new_event_index(list_msg)
|
|
if index == -1:
|
|
await reaction.remove(member)
|
|
return
|
|
|
|
await edit_event_status(index, config.CONFIGURING_EVENT.replace("@User", f"<@{member.id}>"), list_msg)
|
|
await reaction.remove(member)
|
|
|
|
org_role = await guild.create_role(name=f"{config.ORGANIZER_PREFIX}{index}", color=discord.Color.purple())
|
|
await member.add_roles(org_role)
|
|
user_role = await guild.create_role(name=f"{config.PARTICIPANT_PREFIX}{index}")
|
|
await member.add_roles(user_role)
|
|
channel = await guild.create_text_channel(f"{index}{config.EVENT_SUFFIX}", overwrites={
|
|
guild.default_role: discord.PermissionOverwrite(read_messages=False),
|
|
guild.me: discord.PermissionOverwrite(read_messages=True),
|
|
org_role: discord.PermissionOverwrite(read_messages=True, manage_messages=True),
|
|
user_role: discord.PermissionOverwrite(read_messages=True)
|
|
}, category=message.channel.category)
|
|
await list_msg.add_reaction(number_emojis[index - 1])
|
|
await channel.send(config.SETUP_MSG)
|
|
|
|
|
|
def get_new_event_index(msg: discord.Message):
|
|
if config.EMPTY_KEY in msg.content:
|
|
return 1
|
|
for x in msg.content.split('\n'):
|
|
if config.EMPTY_SLOT in x:
|
|
return int(x[2:x.index(':')])
|
|
last_slot = msg.content.split('\n')[-1]
|
|
slot = int(last_slot[2:last_slot.index(':')]) + 1
|
|
return slot if slot < 10 else -1
|
|
|
|
|
|
async def edit_event_status(index: int, text: str, list_msg: discord.Message):
|
|
content = list_msg.content.split('\n')
|
|
try:
|
|
content[index + 1] = f"**{index}:** {text}"
|
|
except IndexError:
|
|
content.append(f"**{index}:** {text}")
|
|
await list_msg.edit(content="\n".join(content))
|
|
|
|
|
|
@nemo.command("!close")
|
|
@nemo.command("!stop")
|
|
@helper.event_command
|
|
@helper.organizer_only
|
|
async def stop(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
message: discord.Message,
|
|
guild: discord.Guild,
|
|
event: int,
|
|
**_):
|
|
await message.delete()
|
|
msg = await channel.send(config.STOP_MSG.replace("@User", f"<@{member.id}>"))
|
|
await msg.add_reaction("✅")
|
|
try:
|
|
await nemo.wait_for("reaction_add", timeout=60, check=lambda reaction, user: user == message.author)
|
|
except asyncio.TimeoutError:
|
|
await msg.delete()
|
|
return
|
|
await discord.utils.get(guild.roles, name=f"{config.ORGANIZER_PREFIX}{event}").delete()
|
|
await discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}").delete()
|
|
await channel.delete()
|
|
org_channel: discord.TextChannel = discord.utils.get(guild.channels, name=config.ORGANIZATION_NAME)
|
|
list_msg: discord.Message = [x async for x in org_channel.history() if config.LIST_KEY in x.content][0]
|
|
if f"**{event + 1}:**" in list_msg.content:
|
|
await edit_event_status(event, config.EMPTY_SLOT, list_msg)
|
|
elif event == 1:
|
|
await list_msg.edit(content=config.EVENT_LIST)
|
|
else:
|
|
await list_msg.edit(content='\n'.join(list_msg.content.split('\n')[:-1]))
|
|
await list_msg.clear_reaction(number_emojis[event - 1])
|
|
|
|
|
|
@nemo.command("!open")
|
|
@helper.event_command
|
|
@helper.organizer_only
|
|
@helper.auto_delete
|
|
async def open_cmd(*,
|
|
message: discord.Message,
|
|
guild: discord.Guild,
|
|
channel: discord.TextChannel,
|
|
event: int,
|
|
**_):
|
|
org_channel: discord.TextChannel = discord.utils.get(guild.channels, name=config.ORGANIZATION_NAME)
|
|
list_msg: discord.Message = [x async for x in org_channel.history() if config.LIST_KEY in x.content][0]
|
|
status = message.content[len("!open "):]
|
|
skip_broadcast = config.EVENT_CONFIGURING_KEY not in list_msg.content
|
|
await edit_event_status(event, status, list_msg)
|
|
await channel.send(config.SET_EVENT_MSG.replace("@Status", status).replace("@User", f"<@{message.author.id}>"))
|
|
if skip_broadcast:
|
|
return
|
|
async for x in org_channel.history():
|
|
if config.NEW_EVENT_KEY in x.content:
|
|
await x.delete()
|
|
await org_channel.send(config.NEW_EVENT.replace("@everyone", f"{guild.default_role}").replace("@id", str(event)))
|
|
|
|
|
|
@nemo.command("!leave")
|
|
@helper.event_command
|
|
@helper.auto_delete
|
|
async def leave(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
guild: discord.Guild,
|
|
event: int,
|
|
**_):
|
|
event_role = discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}")
|
|
await member.remove_roles(event_role)
|
|
await channel.send(config.LEAVE_MSG.replace("@User", f"<@{member.id}>"))
|
|
|
|
|
|
@nemo.command("!private")
|
|
@helper.event_command
|
|
@helper.auto_delete
|
|
async def leave(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
guild: discord.Guild,
|
|
event: int,
|
|
**_):
|
|
org_channel: discord.TextChannel = discord.utils.get(guild.channels, name=config.ORGANIZATION_NAME)
|
|
list_msg: discord.Message = [x async for x in org_channel.history() if config.LIST_KEY in x.content][0]
|
|
await edit_event_status(event, config.PRIVATE_EVENT, list_msg)
|
|
await channel.send(config.SET_PRIVATE_MSG.replace("@User", f"<@{member.id}>"))
|
|
|
|
|
|
@nemo.command("!name")
|
|
@helper.event_command
|
|
@helper.auto_delete
|
|
async def rename(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
message: discord.Message,
|
|
event: int,
|
|
**_):
|
|
await channel.edit(name=f"{event}-{message.content[len('!name '):]}")
|
|
|
|
|
|
@nemo.command("!invite")
|
|
@helper.event_command
|
|
@helper.auto_delete
|
|
async def invite(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
message: discord.Message,
|
|
guild: discord.Guild,
|
|
event: int,
|
|
**_):
|
|
role = discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}")
|
|
for user in message.mentions:
|
|
await user.add_roles(role)
|
|
users = ", ".join([f"<@{user.id}>" for user in message.mentions])
|
|
await channel.send(config.INVITE_MSG.replace("@User", f"<@{member.id}>").replace("@Invited", users))
|
|
|
|
|
|
@nemo.command("!kick")
|
|
@helper.event_command
|
|
@helper.auto_delete
|
|
async def kick(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
message: discord.Message,
|
|
guild: discord.Guild,
|
|
event: int,
|
|
**_):
|
|
role = discord.utils.get(guild.roles, name=f"{config.PARTICIPANT_PREFIX}{event}")
|
|
for user in message.mentions:
|
|
await user.remove_roles(role)
|
|
users = ", ".join([f"<@{user.id}>" for user in message.mentions])
|
|
await channel.send(config.KICK_MSG.replace("@User", f"<@{member.id}>").replace("@Invited", users))
|
|
|
|
|
|
@nemo.command("!colab")
|
|
@helper.event_command
|
|
@helper.auto_delete
|
|
async def colab(*,
|
|
channel: discord.TextChannel,
|
|
member: discord.Member,
|
|
message: discord.Message,
|
|
guild: discord.Guild,
|
|
event: int,
|
|
**_):
|
|
role = discord.utils.get(guild.roles, name=f"{config.ORGANIZER_PREFIX}{event}")
|
|
for user in message.mentions:
|
|
await user.add_roles(role)
|
|
users = ", ".join([f"<@{user.id}>" for user in message.mentions])
|
|
await channel.send(config.COLAB_MSG.replace("@User", f"<@{member.id}>").replace("@Invited", users))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
nemo.run(config.TOKEN)
|