intercal/discord-markov
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add random auto replies, add better logging

Browse Source 
Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
Alek Ratzloff
2022-10-22 18:47:09 -07:00
parent 7f9714263d
commit ce357354a7
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
discord_markov/client.py
View File

@@ -115,9 +115,11 @@ class Client(discord.Client):
# was recorded # was recorded
last_message = self.get_last_tracked_message(channel.id) last_message = self.get_last_tracked_message(channel.id)
log.debug( log.debug(
"Guild %s: channel %s: fetching messages and training model", "Guild %s (%s): channel %s (%s): fetching messages and training model",
guild.id, guild.id,
guild.name,
channel.id, channel.id,
channel.name,
) )
count = 0 count = 0
skipped = 0 skipped = 0
@@ -131,9 +133,11 @@ class Client(discord.Client):
self.chain.add(guild.id, message.author.id, message.content, commit=False) self.chain.add(guild.id, message.author.id, message.content, commit=False)
self.db.commit() self.db.commit()
log.info( log.info(
"Guild %s: channel %s: synchronized %s messages (skipped %s messages that looked like commands)", "Guild %s (%s): channel %s (%s): synchronized %s messages (skipped %s messages that looked like commands)",
guild.id, guild.id,
guild.name,
channel.id, channel.id,
channel.name,
count, count,
skipped, skipped,
) )
@@ -248,3 +252,10 @@ class Client(discord.Client):
# Update the channel's last_message time # Update the channel's last_message time
self.set_last_tracked_message(message.channel.id, message.created_at) self.set_last_tracked_message(message.channel.id, message.created_at)
# Speak your mind, markov
chance = self.chain.get_reply_chance(guild.id, message.author.id)
if random.random() < chance:
line = self.chain.generate(guild.id, message.author.id)
if line:
await message.reply(line)