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 BadUsageException(
"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")
