mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
4a0d96e496
To ease oncall burden for the docathon PR reviewers and ensure all PRs are correctly labeled, adding this GH action that will look for the issue number in the PR and if that issue has a docathon-h1-2024 label, then it would propagate the labels from the issues into the PR. It should not conflict with the existing labelers because we use ``pull_request.add_to_labels`` - credit @kit1980. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127569 Approved by: https://github.com/kit1980
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import os
|
|
import re
|
|
import sys
|
|
|
|
from github import Github
|
|
|
|
|
|
def main() -> None:
|
|
token = os.environ.get("GITHUB_TOKEN")
|
|
|
|
repo_owner = "pytorch"
|
|
repo_name = "pytorch"
|
|
pull_request_number = int(sys.argv[1])
|
|
|
|
g = Github(token)
|
|
repo = g.get_repo(f"{repo_owner}/{repo_name}")
|
|
pull_request = repo.get_pull(pull_request_number)
|
|
pull_request_body = pull_request.body
|
|
# PR without description
|
|
if pull_request_body is None:
|
|
return
|
|
|
|
# get issue number from the PR body
|
|
if not re.search(r"#\d{1,6}", pull_request_body):
|
|
print("The pull request does not mention an issue.")
|
|
return
|
|
issue_number = int(re.findall(r"#(\d{1,6})", pull_request_body)[0])
|
|
issue = repo.get_issue(issue_number)
|
|
issue_labels = issue.labels
|
|
docathon_label_present = any(
|
|
label.name == "docathon-h1-2024" for label in issue_labels
|
|
)
|
|
|
|
# if the issue has a docathon label, add all labels from the issue to the PR.
|
|
if not docathon_label_present:
|
|
print("The 'docathon-h1-2024' label is not present in the issue.")
|
|
return
|
|
pull_request_labels = pull_request.get_labels()
|
|
pull_request_label_names = [label.name for label in pull_request_labels]
|
|
issue_label_names = [label.name for label in issue_labels]
|
|
labels_to_add = [
|
|
label for label in issue_label_names if label not in pull_request_label_names
|
|
]
|
|
if not labels_to_add:
|
|
print("The pull request already has the same labels.")
|
|
return
|
|
pull_request.add_to_labels(*labels_to_add)
|
|
print("Labels added to the pull request!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|