149 lines
5.3 KiB
Python
149 lines
5.3 KiB
Python
import logging
|
|
import os
|
|
import re
|
|
from typing import Union
|
|
|
|
import discord
|
|
from discord import RawReactionActionEvent, RawReactionClearEmojiEvent
|
|
from dotenv import load_dotenv
|
|
|
|
import data
|
|
import jokes
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
LIL_STINKY_ID = 704043422276780072
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class RoboPage(discord.Client):
|
|
db_path: str = 'messages.db'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(RoboPage, self).__init__(*args, **kwargs)
|
|
attrs = filter(lambda n: n.endswith('Joke') and n not in ['Joke', 'GifJoke'], dir(jokes))
|
|
attrs = map(lambda n: getattr(jokes, n)(), attrs)
|
|
self.jokes = list(attrs)
|
|
self.most_regex = re.compile(
|
|
"^who is the most (?P<emoji>\w+)(?: in the past (?P<days>\d+) days)?\??$",
|
|
re.IGNORECASE & re.UNICODE,
|
|
)
|
|
self.leaderboard_regex = re.compile(
|
|
'^most (?P<emoji>\S+) (leaderboard|((?:.+?(?P<days>\d+) days)))',
|
|
re.IGNORECASE & re.UNICODE
|
|
)
|
|
|
|
def run(self):
|
|
return super().run(os.getenv('DISCORD_TOKEN'))
|
|
|
|
async def handle_ready(self):
|
|
async def alive():
|
|
channel: discord.TextChannel = discord.utils.get(self.get_all_channels(), name='robotics-facility')
|
|
await channel.send(f"I'm aliiiiiive {discord.utils.get(self.emojis, name='kaylon')}")
|
|
|
|
self.data: data.MsgData = await data.MsgData.create(
|
|
client=self,
|
|
limit=5000,
|
|
days=30,
|
|
# limit=500,
|
|
# days=3,
|
|
)
|
|
self.data.to_sql('messages.db')
|
|
LOGGER.info(f'{self.data.msgs.shape[0]} messages total')
|
|
# await alive()
|
|
|
|
async def handle_message(self, message):
|
|
if message.author != self.user:
|
|
if hasattr(self, 'data'):
|
|
await self.data.add_msg(message)
|
|
|
|
if (m := self.leaderboard_regex.match(message.content)) is not None:
|
|
try:
|
|
await message.reply(await self.leaderboard(match=m))
|
|
except KeyError as e:
|
|
LOGGER.exception(e)
|
|
await message.reply(f"I couldn't find any {m.group('emoji')} reactions. Leave me alone!")
|
|
return
|
|
|
|
elif (m := self.most_regex.match(message.content)) is not None:
|
|
days = m.group('days') or 14
|
|
try:
|
|
await message.reply(
|
|
await self.data.biggest_single(client=self,
|
|
emoji=get_emoji_name(m.group('emoji')),
|
|
days=int(days)))
|
|
except IndexError as e:
|
|
await message.reply('NObody')
|
|
return
|
|
else:
|
|
LOGGER.warning(f'No self.data attribute')
|
|
|
|
for joke in self.jokes:
|
|
if (scan_res := joke.scan(message)):
|
|
LOGGER.info(f'{joke.__class__.__name__} detected: {message.content}, {scan_res.group()}')
|
|
await joke.respond(message, self, scan_res)
|
|
|
|
async def handle_raw_reaction(self, payload: Union[RawReactionActionEvent, RawReactionClearEmojiEvent]):
|
|
LOGGER.info(payload)
|
|
guild = await client.fetch_guild(payload.guild_id)
|
|
channel = await guild.fetch_channel(payload.channel_id)
|
|
message = await channel.fetch_message(payload.message_id)
|
|
|
|
if payload.event_type == 'REACTION_REMOVE':
|
|
LOGGER.info(f'{payload.emoji} removed from\n{message.author}: {message.content}')
|
|
elif payload.event_type == 'REACTION_ADD':
|
|
LOGGER.info(
|
|
f'{payload.member.display_name} added {payload.emoji} to\n' + \
|
|
f'{message.author.display_name}: {message.content}')
|
|
|
|
if hasattr(self, 'data'):
|
|
await self.data.update_reaction(msg=message)
|
|
|
|
async def leaderboard(self, match: re.Match) -> str:
|
|
emoji_name = get_emoji_name(match.group('emoji'))
|
|
days = match.group('days') or 14
|
|
counts = await self.data.emoji_user_counts(client=self,
|
|
emoji_name=emoji_name,
|
|
days=int(days))
|
|
width = max([len(str(s)) for s in counts.index.values])
|
|
res = f'{match.group("emoji")} totals, past {days} days\n'
|
|
res += '\n'.join(f"`{str(name).ljust(width + 1)}with {cnt:<2.0f} total`"
|
|
for name, cnt in counts.iteritems())
|
|
return res
|
|
|
|
|
|
def get_emoji_name(string: str) -> str:
|
|
if (m := re.search('<:(?P<name>\w+):(?P<id>\d+)>', string)):
|
|
string = m.group('name')
|
|
return string.lower().strip()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
load_dotenv()
|
|
|
|
client = RoboPage()
|
|
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
await client.handle_ready()
|
|
|
|
|
|
@client.event
|
|
async def on_message(message: discord.Message):
|
|
await client.handle_message(message)
|
|
|
|
|
|
@client.event
|
|
async def on_raw_reaction_add(payload: RawReactionActionEvent):
|
|
await client.handle_raw_reaction(payload)
|
|
|
|
|
|
@client.event
|
|
async def on_raw_reaction_remove(payload: RawReactionActionEvent):
|
|
await client.handle_raw_reaction(payload)
|
|
|
|
|
|
client.run()
|