def do_a_remove(self, line):
"""Remove an alias"""
if line in self.aliases:
del self.aliases[line]
aliases = Config.selectBy(name="ALIASES")[0]
aliases.value = repr(self.aliases)
else:
tui.error("No alias with that name. Use a_list to display all aliases")
def do_a_add(self, line):
"""Add an alias on a command
Ex. create an alias 'la' for 't_list -a':
a_add la t_list -a"""
tokens = line.split()
if len(tokens)<2:
raise YokadiException("You should provide an alias name and a command")
name=tokens[0]
command=" ".join(tokens[1:])
self.aliases.update({ name : command})
try:
aliases = Config.selectBy(name="ALIASES")[0]
except IndexError:
# Config entry does not exist. Create it.
aliases = Config(name="ALIASES", value="{}", system=True, desc="User command aliases")
aliases.value = repr(self.aliases)
def testAdd(self):
self.cmd.do_a_add("l t_list")
alias = Config.selectBy(name="ALIASES")[0]
self.assertEqual(eval(alias.value)["l"], "t_list")
