Extract priorities from redmine responses appropriately.

GothenburgBitFactory · Nov 19, 2014 · 6dccc13 · 6dccc13
1 parent 0f11061
commit 6dccc13
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/bugwarrior/services/redmine.py b/bugwarrior/services/redmine.py
@@ -55,6 +55,14 @@ class RedMineIssue(Issue):
     }
     UNIQUE_KEY = (URL, )
 
+    PRIORITY_MAP = {
+        0: 'L',
+        1: 'L',
+        2: 'M',
+        3: 'H',
+        4: 'H',
+    }
+
     def to_taskwarrior(self):
         return {
             'project': self.get_project_name(),
@@ -65,6 +73,12 @@ def to_taskwarrior(self):
             self.ID: self.record['id']
         }
 
+    def get_priority(self):
+        return self.PRIORITY_MAP.get(
+            self.record.get('priority', {}).get('id'),
+            self.origin['default_priority']
+        )
+
     def get_issue_url(self):
         return (
             self.origin['url'] + "/issues/" + six.text_type(self.record["id"])