Commit 9b855ec7 authored by MinchinWeb's avatar MinchinWeb

Priority command: allow lowercase priorities to be input

parent 93544521
...@@ -91,6 +91,16 @@ class PriorityCommandTest(CommandTest): ...@@ -91,6 +91,16 @@ class PriorityCommandTest(CommandTest):
self.assertEqual(self.output, "Priority changed from A to C\n| 1| (C) Foo\nPriority set to C.\n| 2| (C) Bar\n") self.assertEqual(self.output, "Priority changed from A to C\n| 1| (C) Foo\nPriority set to C.\n| 2| (C) Bar\n")
self.assertEqual(self.errors, "") self.assertEqual(self.errors, "")
def test_set_prio7(self):
""" Allow lowercase priority to be set. """
command = PriorityCommand(["Foo", "2", "c"], self.todolist, self.out,
self.error)
command.execute()
self.assertTrue(self.todolist.is_dirty())
self.assertEqual(self.output, "Priority changed from A to C\n| 1| (C) Foo\nPriority set to C.\n| 2| (C) Bar\n")
self.assertEqual(self.errors, "")
def test_expr_prio1(self): def test_expr_prio1(self):
command = PriorityCommand(["-e", "@test", "C"], self.todolist, command = PriorityCommand(["-e", "@test", "C"], self.todolist,
self.out, self.error, None) self.out, self.error, None)
......
...@@ -22,7 +22,7 @@ from topydo.lib.Utils import is_valid_priority ...@@ -22,7 +22,7 @@ from topydo.lib.Utils import is_valid_priority
class PriorityCommand(MultiCommand): class PriorityCommand(MultiCommand):
def __init__(self, p_args, p_todolist, #pragma: no branch def __init__(self, p_args, p_todolist, # pragma: no branch
p_out=lambda a: None, p_out=lambda a: None,
p_err=lambda a: None, p_err=lambda a: None,
p_prompt=lambda a: None): p_prompt=lambda a: None):
...@@ -33,7 +33,7 @@ class PriorityCommand(MultiCommand): ...@@ -33,7 +33,7 @@ class PriorityCommand(MultiCommand):
def _execute_multi_specific(self): def _execute_multi_specific(self):
def normalize_priority(p_priority): def normalize_priority(p_priority):
match = re.search(r'\b([A-Z])\b', p_priority) match = re.search(r'\b([A-Z])\b', p_priority.upper())
return match.group(1) if match else p_priority return match.group(1) if match else p_priority
priority = normalize_priority(self.args[-1]) priority = normalize_priority(self.args[-1])
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment