Garmelon/yaboli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add help topic and multibot drafts

Browse source
This commit is contained in:
Joscha 2017-09-05 22:36:39 +00:00
parent 828bb978c2
commit a6d4a0779f
1 changed files with 41 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

49
yaboli/bot.py
View file

@ -25,9 +25,11 @@ class Bot(Controller):
self._callbacks = Callbacks() self._callbacks = Callbacks()
self.register_default_callbacks() self.register_default_callbacks()
self._help_topics = {}
# settings (modify in your bot's __init__) # settings (modify in your bot's __init__)
self.general_help = None # None -> does not respond to general help self.help_general = None # None -> does not respond to general help
self.specific_help = "No help available" self.help_specific = "No help available"
self.killable = True self.killable = True
self.kill_message = "/me *poof*" # how to respond to !kill, whether killable or not self.kill_message = "/me *poof*" # how to respond to !kill, whether killable or not
self.restartable = True self.restartable = True
@ -36,6 +38,21 @@ class Bot(Controller):
def register_callback(self, event, callback, specific=True): def register_callback(self, event, callback, specific=True):
self._callbacks.add((event, specific), callback) self._callbacks.add((event, specific), callback)
def add_help(self, topic, text, description=None):
info = (text, description) # TODO: make named tuple?
self._help_topics[topic] = info
def get_help(self, topic):
info = self._help_topics.get(topic, ("No help available", None))
return info[0]
def get_help_topics(self):
topics = []
for topic, info in sorted(self._help_topics.items()):
if info[1] is not None:
topics.append(f"{topic} - {info[1]}\n")
return "".join(topics)
async def restart(self): async def restart(self):
# After calling this, the bot is stopped, not yet restarted. # After calling this, the bot is stopped, not yet restarted.
self.restarting = True self.restarting = True
@ -59,9 +76,10 @@ class Bot(Controller):
) )
if len(args) > 0: if len(args) > 0:
mention = args[0] name = args[0]
args = args[1:] args = args[1:]
if mention[:1] == "@" and similar(mention[1:], self.nick): if name[:1] == "@" and similar(name[1:], self.nick):
logger.debug("Specific command!")
# specific callback (specific set to True) # specific callback (specific set to True)
await self._callbacks.call((command, True), message, args) await self._callbacks.call((command, True), message, args)
@ -82,6 +100,8 @@ class Bot(Controller):
argstr = match.group(2) argstr = match.group(2)
args = self.parse_args(argstr) args = self.parse_args(argstr)
logger.debug(f"Parsed command. command={command!r}, args={args!r}")
return command, args return command, args
def parse_args(self, text): def parse_args(self, text):
@ -170,13 +190,13 @@ class Bot(Controller):
@noargs # TODO: specific command help (!help @bot ping) @noargs # TODO: specific command help (!help @bot ping)
async def command_help(self, message): async def command_help(self, message):
if self.specific_help: if self.help_specific:
await self.room.send(self.specific_help, message.message_id) await self.room.send(self.help_specific, message.message_id)
@noargs @noargs
async def command_help_general(self, message): async def command_help_general(self, message):
if self.general_help is not None: if self.help_general is not None:
await self.room.send(self.general_help, message.message_id) await self.room.send(self.help_general, message.message_id)
@noargs @noargs
async def command_uptime(self, message): async def command_uptime(self, message):
@ -203,3 +223,16 @@ class Bot(Controller):
if self.restartable: if self.restartable:
await self.restart() await self.restart()
class Multibot(Bot):
def __init__(self, nick, keeper):
super().__init__(nick)
self.keeper = keeper
class MultibotKeeper():
def __init__(self, configfile):
# TODO: load configfile
# TODO: namedtuple botinfo (bot, task)
self._bots = {} # self._bots[roomname] = botinfo