[llvm] [llvm][GitHub] Move PR project status to Done once backport PR is made (PR #126374)

[Tom Stellard via llvm-commits]

https://github.com/tstellar created https://github.com/llvm/llvm-project/pull/126374

It's common to use the /cherry-pick command on a PR to create a backport request.  However, this creates a lot of clutter in the LLVM Release Status project, because we end up with two items in the project, one for the original PR and one for the new PR.

This change will set the status of the original PR to Done once the new PR (for the release branch) is created.  This will save release managers a lot of work of having to manually updated the status for PRs that contain backport requests.

>From 3182851273a5c703b21e6f4da7f082bda1371c75 Mon Sep 17 00:00:00 2001
From: Tom Stellard <tstellar at redhat.com>
Date: Sat, 8 Feb 2025 08:27:21 -0800
Subject: [PATCH] [llvm][GitHub] Move PR project status to Done once backport
 PR is made

It's common to use the /cherry-pick command on a PR to create a backport
request.  However, this creates a lot of clutter in the LLVM Release
Status project, because we end up with two items in the project, one
for the original PR and one for the new PR.

This change will set the status of the original PR to Done once the new
PR (for the release branch) is created.  This will save release managers
a lot of work of having to manually updated the status for PRs that
contain backport requests.
---
 llvm/utils/git/github-automation.py | 94 +++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)

diff --git a/llvm/utils/git/github-automation.py b/llvm/utils/git/github-automation.py
index 6978da51ac64574..a9f8a34514ad3b8 100755
--- a/llvm/utils/git/github-automation.py
+++ b/llvm/utils/git/github-automation.py
@@ -472,6 +472,99 @@ def release_branch_for_issue(self) -> Optional[str]:
             return m.group(1)
         return None
 
+    def update_issue_project_status(self) -> None:
+        """
+        A common workflow is to merge a PR and then use the /cherry-pick
+        command in a pull request comment to backport the change to the
+        release branch.  In this case, once the new PR for the backport has
+        been created, we want to mark the project status for the original PR
+        as Done.  This is becuase we can track progress now on the new PR
+        which is targeting the release branch.
+
+        """
+
+        pr = self.issue.as_pull_request()
+        if not pr:
+            return
+
+        if pr.state != "closed":
+            return
+
+        gh = github.Github(login_or_token=self.token)
+        query = """
+            query($node_id: ID!) {
+              node(id: $node_id) {
+                ... on PullRequest {
+                  url
+                  projectItems(first:100){
+                    nodes {
+                        id
+                        project {
+                          id
+                          number
+                          field(name: "Status") {
+                            ... on ProjectV2SingleSelectField {
+                              id
+                              name
+                              options {
+                                id
+                                name
+                              }
+                            }
+                          }
+                        }
+                    }
+                  }
+                }
+              }
+            }
+        """
+        variables = {
+            "node_id" : pr.node_id
+        }
+        res_header, res_data = gh._Github__requester.graphql_query(
+            query=query, variables=variables
+        )
+        print(res_header)
+
+        for item in res_data['data']['node']['projectItems']['nodes']:
+            project = item['project']
+            if project['number'] != 3:
+                continue
+            status_field = project['field']
+            for option in status_field['options']:
+                if option['name'] != "Done":
+                    continue
+                variables = {
+                    "project" : project["id"],
+                    "item" : item["id"],
+                    "status_field" : status_field["id"],
+                    "status_value" : option["id"],
+                }
+
+                query = """
+                    mutation($project: ID!, $item: ID!, $status_field: ID!, $status_value: String!) {
+                      set_status:
+                      updateProjectV2ItemFieldValue(input: {
+                        projectId: $project
+                        itemId: $item
+                        fieldId: $status_field
+                        value: {
+                          singleSelectOptionId: $status_value
+                        }
+                      }) {
+                        projectV2Item {
+                          id
+                        }
+                      }
+                    }
+                """
+
+                res_header, res_data = gh._Github__requester.graphql_query(
+                    query=query, variables=variables
+                )
+                print(res_header)
+
     def print_release_branch(self) -> None:
         print(self.release_branch_for_issue)
 
@@ -664,6 +757,7 @@ def create_pull_request(
 
         self.issue_notify_pull_request(pull)
         self.issue_remove_cherry_pick_failed_label()
+        self.update_issue_project_status()
 
         # TODO(tstellar): Do you really want to always return True?
         return True