49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
"""Commands for populating and retrieving markov chains, for silly chatter."""
|
|
import logging
|
|
from urllib.parse import quote
|
|
|
|
from hitomi import bot
|
|
from hitomi.backends import dr_botzo, DrBotzoError
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@bot.listen(name='on_message')
|
|
async def on_message_learn(message):
|
|
"""Learn lines, potentially also respond to them."""
|
|
if hasattr(message.channel, 'guild'):
|
|
target = f'{message.channel.guild.name} ({message.channel.guild.id})'
|
|
else:
|
|
target = f'{message.author.name} ({message.author.id})'
|
|
|
|
logger.info("learning to %s: '%s'", target, message.clean_content)
|
|
try:
|
|
response = dr_botzo.post('/markov/rpc/context/{context}/learn/'.format(context=quote(target)),
|
|
json={'line': message.clean_content})
|
|
logger.debug("succeeded to learn '%s'", response.json()['line'])
|
|
except DrBotzoError as drex:
|
|
logger.exception("received an error from dr.botzo attempting to learn to %s: %s", target, drex)
|
|
|
|
|
|
@bot.listen(name='on_message')
|
|
async def on_message_reply(message):
|
|
"""Learn lines, potentially also respond to them."""
|
|
if hasattr(message.channel, 'guild'):
|
|
target = f'{message.channel.guild.name} ({message.channel.guild.id})'
|
|
else:
|
|
target = f'{message.author.name} ({message.author.id})'
|
|
|
|
if bot.user in message.mentions:
|
|
logger.info("I was addressed, going to reply with chatter")
|
|
await message.channel.trigger_typing()
|
|
try:
|
|
response = dr_botzo.post('/markov/rpc/context/{context}/generate/'
|
|
''.format(context=quote(target)),
|
|
json={'topics': message.clean_content.split(' ')})
|
|
reply = response.json()['generated_line']
|
|
logger.info("replying to %s with '%s'", message.channel, reply)
|
|
await message.channel.send(reply)
|
|
except DrBotzoError as drex:
|
|
logger.exception("received an error from dr.botzo attempting to reply to %s: %s",
|
|
message.clean_content, drex)
|