mirror of
https://github.com/zoriya/Nemo.git
synced 2025-12-06 02:46:10 +00:00
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import traceback
|
|
from typing import Callable
|
|
|
|
import discord
|
|
|
|
|
|
class Nemo(discord.Client):
|
|
def __init__(self, **options):
|
|
super().__init__(**options)
|
|
self.commands = {}
|
|
self.reactions = []
|
|
|
|
async def on_ready(self):
|
|
print(f"Logged in as {self.user}.")
|
|
|
|
async def on_message(self, message: discord.Message):
|
|
cmd = message.content.split()
|
|
if not any(cmd):
|
|
return
|
|
f = self.commands.get(cmd[0])
|
|
if f is None:
|
|
if ":vote:" in cmd:
|
|
await message.add_reaction("✅")
|
|
await message.add_reaction("❎")
|
|
elif ":vote-4:" in cmd:
|
|
await message.add_reaction("1⃣")
|
|
await message.add_reaction("2⃣")
|
|
await message.add_reaction("3⃣")
|
|
await message.add_reaction("4⃣")
|
|
return
|
|
print(f"{message.author.display_name} used the command: '{cmd[0]}'")
|
|
try:
|
|
await f(args=cmd[1::], message=message, guild=message.guild, channel=message.channel, member=message.author)
|
|
except Exception:
|
|
await message.channel.send(f"Fatal error: ```{traceback.format_exc()}```")
|
|
raise
|
|
|
|
def command(self, cmd):
|
|
def wrapper(f):
|
|
self.commands[cmd] = f
|
|
return f
|
|
|
|
return wrapper
|
|
|
|
async def on_reaction_add(self, reaction: discord.Reaction, member: discord.Member):
|
|
if member.bot:
|
|
return
|
|
handler = next((x for x in self.reactions if x[0](reaction, member)), None)
|
|
if handler is None:
|
|
if ":vote:" in reaction.message.content or ":vote-4:" in reaction.message.content:
|
|
for vote in reaction.message.reactions:
|
|
if vote != reaction:
|
|
await vote.remove(member)
|
|
return
|
|
try:
|
|
await handler[1](reaction=reaction,
|
|
member=member,
|
|
message=reaction.message,
|
|
guild=reaction.message.guild,
|
|
channel=reaction.message.channel)
|
|
except Exception:
|
|
await reaction.message.channel.send(f"Fatal error: ```{traceback.format_exc()}```")
|
|
raise
|
|
|
|
async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent):
|
|
if payload.event_type != "REACTION_ADD":
|
|
return
|
|
|
|
if any([x for x in self.cached_messages if x.id == payload.message_id]):
|
|
return
|
|
|
|
channel: discord.TextChannel = self.get_channel(payload.channel_id)
|
|
message: discord.Message = await channel.fetch_message(payload.message_id)
|
|
key = f"{payload.emoji.name}:{payload.emoji.id}" if payload.emoji.id else payload.emoji.name
|
|
reaction = discord.utils.get(message.reactions, emoji=key)
|
|
await self.on_reaction_add(reaction, payload.member)
|
|
|
|
def reaction(self, predicate: Callable[[discord.Reaction, discord.Member], bool]):
|
|
def wrapper(f):
|
|
self.reactions.append((predicate, f))
|
|
return f
|
|
|
|
return wrapper
|