diff --git a/discord_markov/client.py b/discord_markov/client.py index 6d278d4..42fe656 100644 --- a/discord_markov/client.py +++ b/discord_markov/client.py @@ -115,9 +115,11 @@ class Client(discord.Client): # was recorded last_message = self.get_last_tracked_message(channel.id) 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.name, channel.id, + channel.name, ) count = 0 skipped = 0 @@ -131,9 +133,11 @@ class Client(discord.Client): self.chain.add(guild.id, message.author.id, message.content, commit=False) self.db.commit() 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.name, channel.id, + channel.name, count, skipped, ) @@ -248,3 +252,10 @@ class Client(discord.Client): # Update the channel's last_message time 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)