import datetime import re import time import dateutil.parser import discord import fuzzywuzzy.process import twitter from commands import CommandError # @everyone Lilia is Live! Playing $game ! Come and say hi at $url ! ANNOUNCEMENT_PATTERN = re.compile(r'''^ @everyone\s+ Lilia\s+is\s+Live!\s+ Playing\s+(?P.+)\s+!\s+ Come\s+and\s+say\s+hi\s+at\s+(?P.+)\s+! $''', re.VERBOSE) EMOJI_PATTERN = re.compile(r'lilial1\w+') def cooldown(retries, timeout, failure): def do_cooldown(function): def wrapper(self, server, user, *args, **kwargs): cooldowns = getattr(function, 'cooldowns', {}) hash = '{0}:{1}'.format(server, user) if hash not in cooldowns: cooldowns[hash] = dict(tries=0, last_run=0) cd = cooldowns[hash] if cd['tries'] < retries: cd['tries'] += 1 cd['last_run'] = time.time() setattr(function, 'cooldowns', cooldowns) return function(self, server, user, *args, **kwargs) if time.time() - cd['last_run'] > timeout: cd['tries'] = 1 cd['last_run'] = time.time() setattr(function, 'cooldowns', cooldowns) return function(self, server, user, *args, **kwargs) return failure(self, server, user, *args, **kwargs) return wrapper return do_cooldown class DiscordClient(discord.Client): def __init__(self, config, logger, commands, extra_commands): self.config = config self.logger = logger self.commands = commands self.extra_commands = extra_commands self.twitter_api = twitter.Api( consumer_key=self.config['Twitter'].get('consumer_key'), consumer_secret=self.config['Twitter'].get('consumer_secret'), access_token_key=self.config['Twitter'].get('access_token_key'), access_token_secret=self.config['Twitter'].get('access_token_secret')) self.supported_commands = [ (re.compile(r'^!lastquote$'), self._do_lastquote), (re.compile(r'^!findquote\s+(?P")?(?P.+)(?(q)")$'), self._do_findquote), (re.compile(r'^!(bella(gram|pics)|insta(gram|bella))$'), self._do_bellagram), (re.compile(r'^!yt\s+(?P")?(?P.+)(?(q)")$'), self._do_yt), (re.compile(r'^!clip\s+(?P")?(?P.+)(?(q)")$'), self._do_clip), (re.compile(r'^!cheese\s+(?P")?(?P.+)(?(q)")$'), self._do_cheese), ] super(DiscordClient, self).__init__() async def start_(self): token = self.config['Discord'].get('token') await self.start(token) async def on_ready(self): self.logger.info('Logged in as {0}'.format(self.user.name)) async def on_message(self, message): await self._process_new_member(message) await self._process_announcement(message) server = message.server.id if message.server else None for pattern, action in self.supported_commands: m = pattern.match(message.content) if m: await action(server, message.author.id, message, **m.groupdict()) for cmd, resp in self.extra_commands.items(): if cmd == message.content: def repl(m): emoji = discord.utils.get(message.server.emojis, name=m.group(0)) if not emoji: emoji = discord.utils.get(self.get_all_emojis(), name=m.group(0)) return str(emoji) if emoji else m.string resp = EMOJI_PATTERN.sub(repl, resp) await self.send_message(message.channel, resp.format(user=message.author.mention)) def _guess_gifted_subs(self, nick, time, threshold): subs = self.commands.plausible_gifted_subs(time) choices = {i: s['receiver'] for i, s in enumerate(subs)} guesses = fuzzywuzzy.process.extractWithoutOrder(nick, choices, score_cutoff=threshold) result = [] for _, score, i in guesses: days = (time - dateutil.parser.parse(subs[i]['time'])).days result.append((subs[i]['receiver'], score, days)) return sorted(result, key=lambda x: (x[1], 32 - x[2], x[0]), reverse=True) async def _process_gifted_subs(self, user, gifted_subs): header = '{0: <20} | {1: <5} | {2: <11}'.format('Twitch username', 'Match', 'When') table = [header, '-' * len(header)] for nick, score, days in gifted_subs: table.append('{0: <20} | {1: >4}% | {2: >2} days ago'.format(nick, score, days)) message = ('It seems that {0} could have been gifted a sub. ' 'Here are the most probable candidates:\n```{1}```').format(user.mention, '\n'.join(table)) channel = self.config['Discord'].get('gifted_sub_notice_channel') try: await self.send_message(discord.Object(channel), message) except discord.errors.Forbidden: pass async def _process_new_member(self, message): new_members_channel = self.config['Discord'].get('new_members_channel') info_channel = self.config['Discord'].get('info_channel') welcome_channel = self.config['Discord'].get('welcome_channel') welcome_pattern = self.config['Discord'].get('welcome_pattern') if not message.channel or not message.server: return if message.channel.id != new_members_channel: return info_channel = [c for c in message.server.channels if c.id == info_channel] if info_channel: info_channel = info_channel[0] else: return welcome_channel = discord.Object(welcome_channel) try: await self.send_message(welcome_channel, welcome_pattern.format( user=message.author.mention, info_channel=info_channel.mention)) except discord.errors.Forbidden: pass threshold = self.config['Discord'].getint('gifted_sub_threshold') self.logger.info('Looking for gifted subs matching {0}'.format(message.author.display_name)) gifted_subs = self._guess_gifted_subs(message.author.display_name, datetime.datetime.utcnow(), threshold) if gifted_subs: await self._process_gifted_subs(message.author, gifted_subs) async def _process_announcement(self, message): announcer = self.config['Discord'].get('announcer') channels = self.config['Discord'].get('announcement_channels').split(',') if not message.author or not message.channel: return if message.author.id != announcer or message.channel.id not in channels: return after = datetime.datetime.utcnow() - datetime.timedelta(minutes=5) for channel in [discord.Object(c) for c in channels if c != message.channel.id]: async for msg in self.logs_from(channel, after=after): if msg.content == message.content: return try: await self.send_message(channel, message.content) except discord.errors.Forbidden: pass announcement = self.config['Twitter'].get('announcement_pattern') m = ANNOUNCEMENT_PATTERN.match(message.content) if m: self.twitter_api.PostUpdate(announcement.format(**m.groupdict())) async def _cooldown_failure(self, server, user, message, **kwargs): await self.send_message(message.channel, 'Sorry {0}, you have to wait a while before running ' 'the same command again'.format(message.author.mention)) def _format_quote(self, quote): if '"' not in quote['text']: quote['text'] = '"{0}"'.format(quote['text']) return 'One time, Lilia said this... `#{id}: {text} [{game}] [{date}]`'.format(**quote) async def _do_lastquote(self, server, user, message, **kwargs): try: result = self.commands.last_quote() except CommandError as e: await self.send_message(message.channel, 'Sorry {0}, {1}'.format(message.author.mention, e)) else: await self.send_message(message.channel, self._format_quote(result)) async def _do_findquote(self, server, user, message, filter, **kwargs): try: result = self.commands.find_quote(filter) except CommandError as e: await self.send_message(message.channel, 'Sorry {0}, {1}'.format(message.author.mention, e)) else: await self.send_message(message.channel, self._format_quote(result)) @cooldown(retries=2, timeout=5*60, failure=_cooldown_failure) async def _do_bellagram(self, server, user, message, **kwargs): try: bellagram = self.commands.bellagram() except CommandError as e: await self.send_message(message.channel, 'Sorry {0}, {1}'.format(message.author.mention, e)) else: embed = discord.Embed(title=bellagram['caption'], url=bellagram['url'], color=0xd2a517) embed.set_image(url=bellagram['display_url']) embed.set_author(name=bellagram['owner_username'], url=bellagram['owner_url'], icon_url=bellagram['owner_profile_pic_url']) await self.send_message(message.channel, embed=embed) @cooldown(retries=3, timeout=5*60, failure=_cooldown_failure) async def _do_yt(self, server, user, message, query, **kwargs): try: result = self.commands.query_youtube(query) except CommandError as e: await self.send_message(message.channel, 'Sorry {0}, {1}'.format(message.author.mention, e)) else: if result['kind'] == 'playlist': embed = discord.Embed(title=result['title'], url=result['url'], description=result['description'], color=0xff0000) embed.set_thumbnail(url=result['thumbnail_url']) embed.set_author(name=result['channel_title'], url=result['channel_url'], icon_url=result['channel_thumbnail_url']) await self.send_message(message.channel, embed=embed) else: await self.send_message(message.channel, '{title}\n{url}'.format(**result)) @cooldown(retries=3, timeout=5*60, failure=_cooldown_failure) async def _do_clip(self, server, user, message, filter, **kwargs): try: result = self.commands.find_clip(filter) except CommandError as e: await self.send_message(message.channel, 'Sorry {0}, {1}'.format(message.author.mention, e)) else: await self.send_message(message.channel, '{title}\n{url}'.format(**result)) @cooldown(retries=3, timeout=5*60, failure=_cooldown_failure) async def _do_cheese(self, server, user, message, query, **kwargs): try: result = self.commands.query_cheese_com(query) except CommandError as e: await self.send_message(message.channel, 'Sorry {0}, {1}'.format(message.author.mention, e)) else: embed = discord.Embed(title=result['name'], url=result['url'], description=result['description'], color=0xffd700) embed.set_thumbnail(url=result['image']) await self.send_message(message.channel, embed=embed)