Adding new UDAs to github and gerrit

These show the draft/wip status for both services Signed-off-by: Jorge Gallegos <[email protected]>
GothenburgBitFactory · Feb 15, 2024 · a33839e · a33839e
1 parent 8fbc9c0
commit a33839e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/bugwarrior/services/gerrit.py b/bugwarrior/services/gerrit.py
@@ -24,6 +24,8 @@ class GerritIssue(Issue):
     FOREIGN_ID = 'gerritid'
     BRANCH = 'gerritbranch'
     TOPIC = 'gerrittopic'
+    STATUS = 'gerritstatus'
+    WIP = 'gerritwip'
 
     UDAS = {
         SUMMARY: {
@@ -46,6 +48,14 @@ class GerritIssue(Issue):
             'type': 'string',
             'label': 'Gerrit Topic',
         },
+        STATUS: {
+            'type': 'string',
+            'label': 'Gerrit Status',
+        },
+        WIP: {
+            'type': 'string',
+            'label': 'Gerrit Work in Progress',
+        },
     }
     UNIQUE_KEY = (URL, )
 
@@ -61,6 +71,8 @@ def to_taskwarrior(self):
             self.SUMMARY: self.record['subject'],
             self.BRANCH: self.record['branch'],
             self.TOPIC: self.record.get('topic', 'notopic'),
+            self.STATUS: self.record.get('status', ''),
+            self.WIP: 'wip' if self.record.get('work_in_progress', False) else '',
         }
 
     def get_default_description(self):

diff --git a/bugwarrior/services/github.py b/bugwarrior/services/github.py
@@ -207,6 +207,7 @@ class GithubIssue(Issue):
     USER = 'githubuser'
     NAMESPACE = 'githubnamespace'
     STATE = 'githubstate'
+    DRAFT = 'githubdraft'
 
     UDAS = {
         TITLE: {
@@ -260,7 +261,11 @@ class GithubIssue(Issue):
         STATE: {
             'type': 'string',
             'label': 'GitHub State',
-        }
+        },
+        DRAFT: {
+            'type': 'string',
+            'label': 'GitHub Draft',
+        },
     }
     UNIQUE_KEY = (URL, TYPE,)
 
@@ -293,7 +298,8 @@ def to_taskwarrior(self):
             self.UPDATED_AT: updated,
             self.CLOSED_AT: closed,
             self.NAMESPACE: self.extra['namespace'],
-            self.STATE: self.record.get('state', '')
+            self.STATE: self.record.get('state', ''),
+            self.DRAFT: 'draft' if self.record.get('draft', False) else '',
         }
 
     def get_tags(self):