129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
|
|
import discord
|
|
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 not n.startswith('Joke'), 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
|
|
)
|
|
self.leaderboard_regex = re.compile(
|
|
'^most (?P<emoji>\w+)(?=(?:.+)?leaderboard)(?:.+(?P<days>\d+) days)?',
|
|
re.IGNORECASE
|
|
)
|
|
|
|
with Path('gifs.json').open('r') as file:
|
|
self.gifs = json.load(file)
|
|
|
|
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,
|
|
# limit=20,
|
|
days=30,
|
|
)
|
|
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 hasattr(self, 'data'):
|
|
await self.data.add_msg(message)
|
|
|
|
async with self.data.lock:
|
|
if message.author != self.user:
|
|
if (m := self.leaderboard_regex.match(message.content)) is not None:
|
|
try:
|
|
await message.reply(await self.data.emoji_leaderboard(
|
|
client=self,
|
|
emoji_name=m.group('emoji').lower(),
|
|
days=14
|
|
))
|
|
except KeyError as e:
|
|
LOGGER.exception(e)
|
|
await message.reply(f"I couldn't find any {m.group('emoji')} reactions. Leave me alone!")
|
|
|
|
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=m.group('emoji').lower(), days=int(days))
|
|
)
|
|
except IndexError as e:
|
|
await message.reply('NObody')
|
|
|
|
elif 'not like this' in message.content.lower():
|
|
await message.reply(self.gifs['not like this'])
|
|
|
|
elif 'beans' in message.content.lower():
|
|
await message.reply('Somebody help! I\'ve got beans in my motherboard!\n')
|
|
await message.channel.send(self.gifs['beans'])
|
|
|
|
else:
|
|
for joke in self.jokes:
|
|
if (scan_res := joke.scan(message)):
|
|
print(f'{joke.__class__.__name__} detected:\n{message.content}\n{scan_res}')
|
|
await joke.respond(message, self, scan_res)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
load_dotenv()
|
|
|
|
client = RoboPage()
|
|
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
# print(len(list(client.get_all_members())))
|
|
await client.handle_ready()
|
|
# print('\n'.join(client.data.reactions.index.get_level_values(1).drop_duplicates().sort_values()))
|
|
|
|
|
|
@client.event
|
|
async def on_message(message: discord.Message):
|
|
await client.handle_message(message)
|
|
|
|
|
|
@client.event
|
|
async def on_raw_reaction_add(payload):
|
|
LOGGER.info(payload)
|
|
await client.data.update_reaction(payload=payload, client=client)
|
|
|
|
|
|
@client.event
|
|
async def on_raw_reaction_remove(payload):
|
|
LOGGER.info(payload)
|
|
await client.data.update_reaction(payload=payload, client=client)
|
|
|
|
|
|
client.run()
|