Fixes #3685 - Automatically add labels for issues reopened after ML triage

tests/unit/test_webhook_model.py

@@ -37,7 +37,8 @@
 outreach_comment_added = ('outreach generator url added', 200, {'Content-Type': 'text/plain'})  # noqa
 
 issue_info1 = {
-    'action': 'opened', 'state': 'open', 'milestoned_with': '',
+    'action': 'opened', 'state': 'open',
+    'milestoned_with': '', 'labeled_with': '',
     'milestone': '', 'body': '<!-- @browser: Firefox 55.0 -->\n<!-- @ua_header: Mozilla/5.0 (X11; Linux x86_64; rv:55.0) Gecko/20100101 Firefox/55.0 -->\n<!-- @reported_with: web -->\n<!-- @public_url: https://github.com/webcompat/webcompat-tests/issues/1  -->\n\n**URL**: https://www.netflix.com/',   # noqa
     'domain': 'www.netflix.com', 'number': 600,
     'original_labels': ['action-needsmoderation'],
@@ -49,7 +50,8 @@
 }
 
 issue_info2 = {
-    'action': 'milestoned', 'state': 'open', 'milestoned_with': 'accepted',
+    'action': 'milestoned', 'state': 'open',
+    'milestoned_with': 'accepted', 'labeled_with': '',
     'milestone': 'accepted', 'body': '<!-- @browser: Firefox 55.0 -->\n<!-- @ua_header: Mozilla/5.0 (X11; Linux x86_64; rv:55.0) Gecko/20100101 Firefox/55.0 -->\n<!-- @reported_with: web -->\n<!-- @public_url: https://github.com/webcompat/webcompat-tests/issues/1  -->\n\n**URL**: https://www.netflix.com/',  # noqa
     'domain': 'www.netflix.com', 'number': 600,
     'original_labels': ['action-needsmoderation'],

webcompat/webhooks/model.py

@@ -44,6 +44,7 @@ class WebHookIssue:
     original_labels: List[str]
     milestone: str
     milestoned_with: str
+    labeled_with: str
     host_reported_from: str
     html_url: str
 
@@ -67,6 +68,12 @@ def from_dict(cls, payload, host=None):
         milestoned_with = ''
         if payload.get('milestone'):
             milestoned_with = payload.get('milestone')['title']
+
+        # labeled/unlabeled action
+        labeled_with = ''
+        if payload.get('label'):
+            labeled_with = payload.get('label')['name']
+
         host_reported_from = ''
         if host:
             host_reported_from = host
@@ -78,7 +85,8 @@ def from_dict(cls, payload, host=None):
                    state=issue.get('state'), title=full_title,
                    original_labels=original_labels,
                    milestone=milestone, milestoned_with=milestoned_with,
-                   host_reported_from=host_reported_from, html_url=html_url)
+                   host_reported_from=host_reported_from, html_url=html_url,
+                   labeled_with=labeled_with)
 
     def close_private_issue(self):
         """Mark the private issue as closed."""
@@ -259,6 +267,11 @@ def classify(self):
             payload_request = {'milestone': AUTOCLOSED_MILESTONE_ID}
             make_request('patch', path, payload_request)
 
+    def add_bugbug_tracking_label(self, label_name):
+        payload = {'labels': [label_name]}
+        path = f'repos/{PUBLIC_REPO}/{self.number}/labels'
+        make_request('post', path, payload)
+
     def process_issue_action(self):
         """Route the actions and provide different responses.
 
@@ -300,6 +313,25 @@ def process_issue_action(self):
                 return oops()
             else:
                 return make_response('outreach generator url added', 200)
+        elif (self.action == 'milestoned' and scope == 'public' and
+              self.milestoned_with in ('needsdiagnosis', 'moved')):
+            try:
+                if 'bugbug-reopened' in self.original_labels:
+                    self.add_bugbug_tracking_label('bugbug-valid')
+            except HTTPError as e:
+                msg_log(f'bugbug-valid labeling failed ({e})', self.number)
+                return oops()
+            else:
+                return make_response('bugbug-valid label added', 200)
+        elif (self.action == 'unlabeled' and scope == 'public' and
+              self.labeled_with == 'bugbug-probability-high'):
+            try:
+                self.add_bugbug_tracking_label('bugbug-reopened')
+            except HTTPError as e:
+                msg_log(f'bugbug-reopen labeling failed ({e})', self.number)
+                return oops()
+            else:
+                return make_response('bugbug-reopen label added', 200)
         elif self.action == 'opened' and scope == 'private':
             # webcompat-bot needs to comment public URL of the issue
             # and we try to classify the issue using bugbug