broke out jokes into their own module

jokes.py

@@ -0,0 +1,106 @@
+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)
+
+
+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()}'

robopage.py

@@ -4,10 +4,9 @@ import re
 from threading import Lock
 
 import discord
-import nltk
-import stockquotes
 from dotenv import load_dotenv
 
+from jokes import CumJoke, BlackJoke, AssJoke, DominosJoke
 from msg import get_and_save, cancellations, cancelled_totals, report_string
 
 logging.basicConfig(level=logging.INFO)
@@ -31,19 +30,6 @@ class RoboPage(discord.Client):
     def run(self):
         return super().run(os.getenv('DISCORD_TOKEN'))
 
-    async def reaction_messages(self, target: str, **kwargs):
-        for c in self.get_all_channels():
-            if c.guild.name == 'Family Dinner' and \
-                    isinstance(c, discord.TextChannel):
-                if c.id != LIL_STINKY_ID:
-                    print(f'Scanning in {c.name} for cancellations')
-                    for m in await c.history(**kwargs).flatten():
-                        if len(m.reactions) > 0:
-                            for r in m.reactions:
-                                if isinstance(r.emoji, discord.Emoji) and r.emoji.name == target:
-                                    print(r.count, m.author.display_name)
-                                    yield m, r.count
-
     async def handle_ready(self):
         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')}")
@@ -88,107 +74,6 @@ class RoboPage(discord.Client):
         return res
 
 
-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)
-
-
-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()}'
-
-
 if __name__ == '__main__':
     load_dotenv()
 
@@ -199,6 +84,7 @@ if __name__ == '__main__':
     async def on_ready():
         print(f'{client.user} has connected to Discord!')
         # await client.handle_ready()
+        # await get_and_save(RoboPage.db_path, client=client, limit=5000, days=7)
         # msg_df, react_df = await get_and_save('messages.db', client=client, limit=5000, days=90)