Webhook to add closed-by-bot label

astropy_bot/closedbybotlabel.py

@@ -0,0 +1,29 @@
+from baldrick.blueprints.github import github_webhook_handler
+from baldrick.github.github_api import IssueHandler, PullRequestHandler
+
+
+@github_webhook_handler
+def handle_closed_by_bot_label(repo_handler, payload, headers):
+    """
+    Auto-apply closed-by-bot label when bot closes issue or pull request.
+    """
+    event = headers['X-GitHub-Event']
+
+    if event not in ('pull_request', 'issues') or payload['action'] != 'closed':
+        return "Not a closed pull_request or issues event"
+
+    repository = repo_handler.repo
+    installation = repo_handler.installation
+    closed_by = payload['sender']['login']  # TODO: Is this correct???
+
+    if event == 'pull_request':
+        number = payload['pull_request']['number']
+        handler = PullRequestHandler(repository, number, installation)
+    elif event == 'issues':
+        number = payload['issue']['number']
+        handler = IssueHandler(repository, number, installation)
+    else:
+        return "Not an issue or pull request"
+
+    if closed_by == 'astropy-bot':
+        handler.set_labels('closed-by-bot')

astropy_bot/tests/test_closedbybotlabel.py

@@ -0,0 +1 @@
+# TODO: Add tests here