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

add !markovfor

Browse source
This commit is contained in:
jesopo 2019-09-20 18:08:22 +01:00
parent c2e3fa4e80
commit 3dfee55d42
1 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
modules/markov.py
View file

@ -43,14 +43,31 @@ class Module(ModuleManager.BaseModule):
@utils.hook("received.command.markov") @utils.hook("received.command.markov")
@utils.kwarg("channel_only", True) @utils.kwarg("channel_only", True)
def generate(self, event): @utils.kwarg("help", "Generate a markov chain for the current channel")
if not event["target"].get_setting("markov", False): def markov(self, event):
raise utils.EventError("Markov chains not enabled in this channel") self._markov_for(event["target"], event["stdout"], event["stderr"])
out = self._generate(event["target"].id)
if not out == None: @utils.hook("received.command.markov")
event["stdout"].write(out) @utils.kwarg("min_args", 1)
@utils.kwarg("permission", "markovfor")
@utils.kwarg("help", "Generate a markov chain for a given channel")
@utils.kwarg("usage", "<channel>")
def markov_for(self, event):
if event["args_split"][0] in event["server"].channels:
channel = event["server"].channels.get(event["args_split"][0])
self._markov_for(channel, event["stdout"], event["stderr"])
else: else:
event["stderr"].write("Failed to generate markov chain") event["stderr"].write("Unknown channel")
def _markov_for(self, channel, stdout, stderr):
if not channel.get_setting("markov", False):
stderr.write("Markov chains not enabled in this channel")
else:
out = self._generate(channel.id)
if not out == None:
stdout.write(out)
else:
stderr.write("Failed to generate markov chain")
def _generate(self, channel_id): def _generate(self, channel_id):
first_words = self.bot.database.execute_fetchall("""SELECT third_word, first_words = self.bot.database.execute_fetchall("""SELECT third_word,