Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

add command for loading logs in to markov (via http)

Browse source
This commit is contained in:
jesopo 2019-09-23 10:03:32 +01:00
parent 72ff24227d
commit e448a4abb8
1 changed files with 43 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

61
modules/markov.py
View file

@ -1,6 +1,8 @@
import random
from src import ModuleManager, utils
NO_MARKOV = "Markov chains not enabled in this channel"
class Module(ModuleManager.BaseModule):
def on_load(self):
if not self.bot.database.has_table("markov"):
@ -12,29 +14,52 @@ class Module(ModuleManager.BaseModule):
@utils.hook("received.message.channel")
def channel_message(self, event):
words = [word.lower() for word in event["message_split"]]
if event["channel"].get_setting("markov", False):
self._create(event["channel"].id, event["message_split"])
@utils.hook("received.command.markovlog")
@utils.kwarg("min_args", 1)
@utils.kwarg("permission", "markovlog")
@utils.kwarg("help", "Load a message-only newline-delimited log in to this "
"channel's markov chain")
def load_log(self, event):
if not event["target"].get_setting("markov", False):
raise utils.EventError(NO_MARKOV)
page = utils.http.request(event["args_split"][0])
if page.code == 200:
for line in page.data.split("\n"):
self._create(event["target"].id, line.strip("\r").split(" "))
else:
event["stdout"].write("Failed to load log (%d)" % page.code)
def _create(self, channel_id, words):
words = list(filter(None, words))
words = [word.lower() for word in words]
words_n = len(words)
if words_n > 2 and event["channel"].get_setting("markov", False):
inserts = []
inserts.append([None, None, words[0]])
inserts.append([None, words[0], words[1]])
if not words_n > 2:
return
for i in range(words_n-2):
inserts.append(words[i:i+3])
inserts = []
inserts.append([None, None, words[0]])
inserts.append([None, words[0], words[1]])
inserts.append([words[-2], words[-1], None])
for i in range(words_n-2):
inserts.append(words[i:i+3])
for insert in inserts:
frequency = self.bot.database.execute_fetchone("""SELECT
frequency FROM markov WHERE channel_id=? AND first_word=?
AND second_word=? AND third_word=?""",
[event["channel"].id]+insert)
frequency = (frequency or [0])[0]+1
inserts.append([words[-2], words[-1], None])
self.bot.database.execute(
"INSERT OR REPLACE INTO markov VALUES (?, ?, ?, ?, ?)",
[event["channel"].id]+insert+[frequency])
for insert in inserts:
frequency = self.bot.database.execute_fetchone("""SELECT
frequency FROM markov WHERE channel_id=? AND first_word=?
AND second_word=? AND third_word=?""",
[channel_id]+insert)
frequency = (frequency or [0])[0]+1
self.bot.database.execute(
"INSERT OR REPLACE INTO markov VALUES (?, ?, ?, ?, ?)",
[channel_id]+insert+[frequency])
def _choose(self, words):
words, frequencies = list(zip(*words))
@ -60,7 +85,7 @@ class Module(ModuleManager.BaseModule):
def _markov_for(self, channel, stdout, stderr):
if not channel.get_setting("markov", False):
stderr.write("Markov chains not enabled in this channel")
stderr.write(NO_MARKOV)
else:
out = self._generate(channel.id)
if not out == None: