Task: Bail out if more than one task has been matched on refresh

[etc/taskwarrior.git] / tasklib / task.py

diff --git a/tasklib/task.py b/tasklib/task.py
index ba95662de1736a2d53e2d041ff4f1faa199f68a0..cbc11925495a908a8d12a1a88b725ed732856958 100644 (file)
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -677,7 +677,13 @@ class Task(TaskResource):
                                            "task: %s" % '\n'.join(id_lines))
 
             # Circumvent the ID storage, since ID is considered read-only
-            self._data['id'] = int(id_lines[0].split(' ')[2].rstrip('.'))
+            identifier = id_lines[0].split(' ')[2].rstrip('.')
+
+            # Identifier can be either ID or UUID for completed tasks
+            try:
+                self._data['id'] = int(identifier)
+            except ValueError:
+                self._data['uuid'] = identifier
 
         # Refreshing is very important here, as not only modification time
         # is updated, but arbitrary attribute may have changed due hooks
@@ -747,6 +753,14 @@ class Task(TaskResource):
         # of newly saved tasks. Any other place in the code is fine
         # with using UUID only.
         args = [self['uuid'] or self['id'], 'export']
+        output = self.warrior.execute_command(args)
+
+        # If more than 1 task has been matched, raise exception
+        if len(output) > 1:
+            raise TaskWarriorException(
+                "Unique identifier {0} matches multiple tasks: {1}".format(
+                    self['uuid'] or self['id'], output))
+
         new_data = json.loads(self.warrior.execute_command(args)[0])
         if only_fields:
             to_update = dict(