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