Label PRs with modified addons

Author: sbidoul

src/oca_github_bot/config.py

@@ -59,7 +59,7 @@ def func_wrapper(*args, **kwargs):
 # Available tasks:
 #  delete_branch,tag_approved,tag_ready_to_merge,gen_addons_table,
 #  gen_addons_readme,gen_addons_icon,setuptools_odoo,merge_bot,tag_needs_review,
-#  migration_issue_bot,whool_init,gen_metapackage
+#  migration_issue_bot,whool_init,gen_metapackage,label_modified_addons
 BOT_TASKS = os.environ.get("BOT_TASKS", "all").split(",")
 
 BOT_TASKS_DISABLED = os.environ.get("BOT_TASKS_DISABLED", "").split(",")

src/oca_github_bot/tasks/__init__.py

@@ -3,6 +3,7 @@
 
 from . import (
     heartbeat,
+    label_modified_addons,
     main_branch_bot,
     mention_maintainer,
     migration_issue_bot,

src/oca_github_bot/tasks/label_modified_addons.py

@@ -0,0 +1,39 @@
+# Copyright (c) ACSONE SA/NV 2024
+# Distributed under the MIT License (http://opensource.org/licenses/MIT).
+
+from .. import github
+from ..config import switchable
+from ..manifest import git_modified_addons
+from ..process import check_call
+from ..queue import task
+from ..version_branch import is_main_branch_bot_branch
+
+
+def _label_modified_addons(gh, org, repo, pr, dry_run):
+    gh_pr = gh.pull_request(org, repo, pr)
+    target_branch = gh_pr.base.ref
+    pr_branch = f"tmp-pr-{pr}"
+    with github.temporary_clone(org, repo, target_branch) as clone_dir:
+        check_call(
+            ["git", "fetch", "origin", f"pull/{pr}/head:{pr_branch}"],
+            cwd=clone_dir,
+        )
+        check_call(["git", "checkout", pr_branch], cwd=clone_dir)
+        modified_addons, _ = git_modified_addons(clone_dir, target_branch)
+        if not modified_addons:
+            return
+        gh_issue = github.gh_call(gh_pr.issue)
+        new_labels = {f"addon:{modified_addon}" for modified_addon in modified_addons}
+        if is_main_branch_bot_branch(target_branch):
+            new_labels.add(f"series:{target_branch}")
+        new_labels = new_labels - {label.name for label in gh_issue.labels()}
+        if new_labels and not dry_run:
+            for new_label in new_labels:
+                github.gh_call(gh_issue.add_labels, new_label)
+
+
+@task()
+@switchable("label_modified_addons")
+def label_modified_addons(org, repo, pr, dry_run):
+    with github.login() as gh:
+        _label_modified_addons(gh, org, repo, pr, dry_run)

src/oca_github_bot/version_branch.py

@@ -33,7 +33,7 @@ def is_main_branch_bot_branch(branch_name):
 
 
 def is_protected_branch(branch_name):
-    if branch_name == "master":
+    if branch_name in ("master", "main"):
         return True
     return bool(ODOO_VERSION_RE.match(branch_name))
 

src/oca_github_bot/webhooks/on_pr_label_modified_addons.py

@@ -0,0 +1,21 @@
+# Copyright (c) ACSONE SA/NV 2024
+# Distributed under the MIT License (http://opensource.org/licenses/MIT).
+
+import logging
+
+from ..router import router
+from ..tasks.label_modified_addons import label_modified_addons
+
+_logger = logging.getLogger(__name__)
+
+
+@router.register("pull_request", action="opened")
+@router.register("pull_request", action="reopened")
+@router.register("pull_request", action="synchronize")
+async def on_pr_label_modified_addons(event, *args, **kwargs):
+    """
+    Whenever a PR is opened, add labels based on modified addons.
+    """
+    org, repo = event.data["repository"]["full_name"].split("/")
+    pr = event.data["pull_request"]["number"]
+    label_modified_addons.delay(org, repo, pr)

tests/cassettes/test_label_modified_addons.yaml

@@ -0,0 +1,10 @@
+# Copyright ACSONE SA/NV 2024
+# Distributed under the MIT License (http://opensource.org/licenses/MIT).
+import pytest
+
+from oca_github_bot.tasks.label_modified_addons import _label_modified_addons
+
+
+@pytest.mark.vcr()
+def test_label_modified_addons(gh):
+    _label_modified_addons(gh, "OCA", "mis-builder", "610", dry_run=False)