GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2025-02-17 14:01:03 +01:00
Code Issues Projects Releases Wiki Activity

add alias argument to add_cmd()

Browse Source
This commit is contained in:
Mitchell Cooper 2017-07-09 21:18:45 -04:00
parent b7466327db
commit 2299204efa
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
utils.py
View File

@ -217,6 +217,9 @@ class ServiceBot():
# List of command names to "feature" # List of command names to "feature"
self.featured_cmds = set() self.featured_cmds = set()
# Maps command aliases to the respective primary commands
self.alias_cmds = {}
if default_help: if default_help:
self.add_cmd(self.help) self.add_cmd(self.help)
@ -337,7 +340,7 @@ class ServiceBot():
log.exception('Unhandled exception caught in command %r', cmd) log.exception('Unhandled exception caught in command %r', cmd)
self.reply(irc, 'Uncaught exception in command %r: %s: %s' % (cmd, type(e).__name__, str(e))) self.reply(irc, 'Uncaught exception in command %r: %s: %s' % (cmd, type(e).__name__, str(e)))
def add_cmd(self, func, name=None, featured=False): def add_cmd(self, func, name=None, featured=False, alias=None):
"""Binds an IRC command function to the given command name.""" """Binds an IRC command function to the given command name."""
if name is None: if name is None:
name = func.__name__ name = func.__name__
@ -346,6 +349,10 @@ class ServiceBot():
# Mark as a featured command if requested to do so. # Mark as a featured command if requested to do so.
if featured: if featured:
self.featured_cmds.add(name) self.featured_cmds.add(name)
# If this is an alias, store primary command
if alias is not None:
self.alias_cmds[name] = alias
self.commands[name].append(func) self.commands[name].append(func)
return func return func