43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
from plugin import BasePlugin
|
||
import common
|
||
from common import parse_command_args_to_alias as parse
|
||
|
||
class Plugin(BasePlugin):
|
||
def init(self):
|
||
self.add_command('alias', self.command_alias, '/alias <alias> <command> <args>\nAlias: create an alias command')
|
||
self.add_command('unalias', self.command_unalias, '/unalias <alias>\nUnalias: remove a previously created alias')
|
||
self.commands = {}
|
||
|
||
def command_alias(self, line):
|
||
arg = common.shell_split(line)
|
||
if len(arg) < 2:
|
||
self.core.information('Alias: Not enough parameters', 'Error')
|
||
return
|
||
alias = arg[0]
|
||
tmp_args = common.shell_split(arg[1])
|
||
command = tmp_args.pop(0)
|
||
tmp_args = arg[1][len(command)+1:]
|
||
|
||
if alias in self.core.commands or alias in self.commands:
|
||
self.core.information('Alias: command already exists', 'Error')
|
||
return
|
||
self.commands[alias] = lambda arg: self.get_command(command)(parse(arg, tmp_args))
|
||
self.add_command(alias, self.commands[alias], 'This command is an alias for /%s %s' %( command, tmp_args))
|
||
self.core.information('Alias /%s successfuly created' % alias, 'Info')
|
||
|
||
def command_unalias(self, alias):
|
||
if alias in self.commands:
|
||
del self.commands[alias]
|
||
self.del_command(alias)
|
||
self.core.information('Alias /%s successfuly deleted' % alias, 'Info')
|
||
|
||
def get_command(self, name):
|
||
"""Returns the function associated with a command"""
|
||
def dummy(args):
|
||
"""Dummy function called if the command doesn’t exist"""
|
||
pass
|
||
if name in self.core.commands:
|
||
return self.core.commands[name][0]
|
||
elif name in self.core.current_tab().commands:
|
||
return self.core.current_tab().commands[name][0]
|
||
return dummy
|