142 lines
4.3 KiB
Python
142 lines
4.3 KiB
Python
import re
|
|
|
|
import discord
|
|
import nltk
|
|
import stockquotes
|
|
|
|
|
|
class Joke:
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
raise NotImplementedError
|
|
|
|
def scan(self, message: discord.Message):
|
|
if (match := self.regex.search(message.content)):
|
|
return match
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, scan_res):
|
|
raise NotImplementedError
|
|
|
|
|
|
class CumJoke(Joke):
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
words = [
|
|
'come',
|
|
'coming',
|
|
'came',
|
|
'cum',
|
|
'cumming',
|
|
'cummed'
|
|
]
|
|
return re.compile(f"(?<!\w)({'|'.join(words)})(?!\w)", re.IGNORECASE)
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, match: re.Match):
|
|
if not match.group(0).startswith('be'):
|
|
await message.add_reaction(discord.utils.get(client.emojis, name='kaylon'))
|
|
|
|
|
|
class BlackJoke(Joke):
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
return re.compile('black (\w+)', re.IGNORECASE)
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, match: re.Match):
|
|
res = unblack(message.content)
|
|
if res is not None:
|
|
msg = await message.reply(res)
|
|
await msg.add_reaction(discord.utils.get(client.emojis, name='kaylon'))
|
|
|
|
|
|
class AssJoke(Joke):
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
return re.compile('[ \-]ass[ \-](?P<target>\w+)', re.IGNORECASE)
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, match: re.Match):
|
|
res = assify(message.content)
|
|
if res is not None:
|
|
await message.reply(f'{res} {discord.utils.get(client.emojis, name="kaylon")}')
|
|
|
|
|
|
class DominosJoke(Joke):
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
return re.compile('domino\'?s', re.IGNORECASE)
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, match: re.Match):
|
|
cp = stockquotes.Stock('DPZ').current_price
|
|
msg = f'You know, my friend Ben has made about ${cp - 16:.0f} on Domino\'s stock. He basically owns it now'
|
|
if (e := discord.utils.get(client.emojis, name="pizza")):
|
|
await message.add_reaction(e)
|
|
await message.reply(msg)
|
|
|
|
|
|
class GifJoke(Joke):
|
|
url: str
|
|
|
|
def __init__(self, url: str):
|
|
self.url = url
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, scan_res):
|
|
await message.channel.send(self.url)
|
|
|
|
|
|
class BeansJoke(GifJoke):
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
return re.compile('beans', re.IGNORECASE)
|
|
|
|
def __init__(self):
|
|
super().__init__(url='https://c.tenor.com/TjX1yORoln0AAAAM/this-is-beans-beans.gif')
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, scan_res):
|
|
await message.reply('Somebody help! I\'ve got beans in my motherboard!\n')
|
|
await super().respond(message, client, scan_res)
|
|
|
|
|
|
class NotLikeThisJoke(GifJoke):
|
|
@property
|
|
def regex(self) -> re.Pattern:
|
|
return re.compile('not like this', re.IGNORECASE)
|
|
|
|
def __init__(self):
|
|
super().__init__(url='https://tenor.com/view/not-like-this-the-matrix-panic-neo-angry-gif-5216157')
|
|
|
|
async def respond(self, message: discord.Message, client: discord.Client, scan_res):
|
|
await message.reply(self.url)
|
|
|
|
|
|
pattern = 'NP: {<DT>?<JJ>*<NN>}'
|
|
cp = nltk.RegexpParser(pattern)
|
|
|
|
|
|
def token_list(s):
|
|
return nltk.chunk.tree2conlltags(
|
|
cp.parse(
|
|
nltk.pos_tag(
|
|
nltk.word_tokenize(s)
|
|
)))
|
|
|
|
|
|
def assify(s):
|
|
tag_list = token_list(s)
|
|
for i, (text, tag, iob) in enumerate(tag_list):
|
|
if text[-3:].lower() == 'ass':
|
|
try:
|
|
next_tag = tag_list[i + 1][1]
|
|
if next_tag == 'NN' or next_tag == 'NNS':
|
|
return f'ass-{tag_list[i + 1][0]}'
|
|
except IndexError as e:
|
|
return
|
|
|
|
|
|
def unblack(s):
|
|
tag_list = token_list(s)
|
|
for i, (text, tag, iob) in enumerate(tag_list):
|
|
if text.lower() == 'black':
|
|
if tag.startswith('JJ') or tag.startswith('NN'):
|
|
for text, tag, iob in tag_list[i + 1:]:
|
|
if tag.startswith('NN'):
|
|
return f'Or as I would say, {text.lower()}'
|