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

support first-word argument to !markovfor

Browse source
This commit is contained in:
jesopo 2019-09-24 10:51:25 +01:00
parent 6669926746
commit 5ea714182d
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
modules/markov.py
View file

@ -84,6 +84,7 @@ class Module(ModuleManager.BaseModule):
@utils.hook("received.command.markov") @utils.hook("received.command.markov")
@utils.kwarg("channel_only", True) @utils.kwarg("channel_only", True)
@utils.kwarg("help", "Generate a markov chain for the current channel") @utils.kwarg("help", "Generate a markov chain for the current channel")
@utils.kwarg("usage", "[first-word]")
def markov(self, event): def markov(self, event):
self._markov_for(event["target"], event["stdout"], event["stderr"], self._markov_for(event["target"], event["stdout"], event["stderr"],
first_word=(event["args_split"] or [None])[0]) first_word=(event["args_split"] or [None])[0])
@ -92,11 +93,12 @@ class Module(ModuleManager.BaseModule):
@utils.kwarg("min_args", 1) @utils.kwarg("min_args", 1)
@utils.kwarg("permission", "markovfor") @utils.kwarg("permission", "markovfor")
@utils.kwarg("help", "Generate a markov chain for a given channel") @utils.kwarg("help", "Generate a markov chain for a given channel")
@utils.kwarg("usage", "<channel>") @utils.kwarg("usage", "<channel> [first-word]")
def markov_for(self, event): def markov_for(self, event):
if event["args_split"][0] in event["server"].channels: if event["args_split"][0] in event["server"].channels:
channel = event["server"].channels.get(event["args_split"][0]) channel = event["server"].channels.get(event["args_split"][0])
self._markov_for(channel, event["stdout"], event["stderr"]) self._markov_for(channel, event["stdout"], event["stderr"],
first_word=(event["args_split"][1:] or [None])[0])
else: else:
event["stderr"].write("Unknown channel") event["stderr"].write("Unknown channel")