bitbot-3.11-fork/modules/factoids.py

38 lines
1.3 KiB
Python

#--depends-on commands
import re
from src import ModuleManager, utils
class Module(ModuleManager.BaseModule):
def _get_factoid(self, server, factoid):
name = factoid.lower().strip()
return name, server.get_setting("factoid-%s" % name, None)
@utils.hook("received.command.factoid", min_args=1)
def factoid(self, event):
"""
:help: Set/get a factoid
:usage: <key> [= value]
"""
if "=" in event["args"]:
key, _, value = event["args"].partition("=")
factoid = key.lower().strip()
event["server"].set_setting("factoid-%s" % factoid, value.strip())
event["stdout"].write("Set factoid '%s'" % factoid)
else:
name, value = self._get_factoid(event["server"], event["args"])
if value == None:
raise utils.EventError("Unknown factoid '%s'" % name)
event["stdout"].write("%s: %s" % (name, value))
@utils.hook("command.regex", ignore_action=False)
def channel_message(self, event):
"""
:command: factoid
:pattern: {!factoid ([^}]+)}
"""
name, value = self._get_factoid(event["server"],
event["match"].group(1))
if not value == None:
event["stdout"].write("%s: %s" % (name, value))