67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
from .base import BaseModule
|
|
import random
|
|
import time
|
|
import json
|
|
|
|
RHYMES_FILE = "rhymes.txt"
|
|
|
|
|
|
class RhymesModule(BaseModule):
|
|
rhymes: dict = {}
|
|
cooldown: dict = {}
|
|
|
|
def load(self):
|
|
with open("rhymes.json", "r") as f:
|
|
self.rhymes = json.load(f)
|
|
|
|
def save(self):
|
|
dump = {}
|
|
for key, value in self.cooldown.items():
|
|
dump[key.id] = value - time.time()
|
|
return dump
|
|
|
|
def get_last_word(self, ch: str) -> str:
|
|
truncated = ch
|
|
while True:
|
|
if len(truncated) < 2 or truncated[-1].isnumeric():
|
|
return ""
|
|
if truncated[-1].isalpha() and truncated[-2].isalpha():
|
|
break
|
|
truncated = truncated[:-1]
|
|
return truncated
|
|
|
|
def poil_auquel(self, ch: str) -> str:
|
|
for key in self.rhymes:
|
|
if ch.endswith(tuple(self.rhymes[key]["keys"])):
|
|
return random.choice(self.rhymes[key]["rhymes"])
|
|
return ""
|
|
|
|
async def handle_message(self, message):
|
|
message_content = message.content.lower()
|
|
|
|
if message_content == "debug":
|
|
dump = {}
|
|
for key, value in self.cooldown.items():
|
|
channel_name = await self.get_channel_name(key)
|
|
dump[channel_name] = value - time.time()
|
|
await message.author.send(
|
|
"```json\n{0}```".format(json.dumps(dump, ensure_ascii=False, indent=4))
|
|
)
|
|
return
|
|
|
|
last_word = self.get_last_word(message_content)
|
|
if message.author != self._client.user and last_word:
|
|
poil = self.poil_auquel(last_word)
|
|
# cooldown = 0
|
|
# if message.channel in self.cooldown:
|
|
# cooldown = self.cooldown[message.channel]
|
|
cooldown = self.cooldown.get(message.channel, 0)
|
|
|
|
if poil and time.time() - cooldown > 0:
|
|
wait_time = random.randint(0, 900)
|
|
if bool(random.getrandbits(1)):
|
|
wait_time = random.randint(900, 10800)
|
|
self.cooldown[message.channel] = time.time() + wait_time
|
|
await message.channel.send(poil)
|
|
return
|