92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
|
from argparse import ArgumentParser, Namespace
|
||
|
|
from pathlib import Path
|
||
|
|
from github.Repository import Repository
|
||
|
|
from github.GithubException import GithubException
|
||
|
|
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
from ryujinx_mako.commands._subcommand import GithubSubcommand
|
||
|
|
|
||
|
|
|
||
|
|
class UpdateReviewers(GithubSubcommand):
|
||
|
|
@staticmethod
|
||
|
|
def name() -> str:
|
||
|
|
return "update-reviewers"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def description() -> str:
|
||
|
|
return "Update reviewers for the specified PR"
|
||
|
|
|
||
|
|
def __init__(self, parser: ArgumentParser):
|
||
|
|
self._reviewers = set()
|
||
|
|
self._team_reviewers = set()
|
||
|
|
|
||
|
|
parser.add_argument("repo_path", type=str)
|
||
|
|
parser.add_argument("pr_number", type=int)
|
||
|
|
parser.add_argument("config_path", type=Path)
|
||
|
|
|
||
|
|
super().__init__(parser)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def reviewers_lower(self) -> list[str]:
|
||
|
|
return [x.lower() for x in self._reviewers]
|
||
|
|
|
||
|
|
def add_reviewers(self, new_entries: list[str]):
|
||
|
|
for reviewer in new_entries:
|
||
|
|
if reviewer.startswith("@"):
|
||
|
|
self._team_reviewers.add(reviewer[1:])
|
||
|
|
else:
|
||
|
|
self._reviewers.add(reviewer)
|
||
|
|
|
||
|
|
def update_reviewers(self, config, repo: Repository, pr_number: int) -> int:
|
||
|
|
pull_request = repo.get_pull(pr_number)
|
||
|
|
|
||
|
|
if not pull_request:
|
||
|
|
self.logger.error(f"Unknown PR #{pr_number}")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
if pull_request.draft:
|
||
|
|
self.logger.warning("Not assigning reviewers for draft PRs")
|
||
|
|
return 0
|
||
|
|
|
||
|
|
pull_request_author = pull_request.user.login
|
||
|
|
|
||
|
|
for label in pull_request.labels:
|
||
|
|
if label.name in config:
|
||
|
|
self.add_reviewers(config[label.name])
|
||
|
|
|
||
|
|
if "default" in config:
|
||
|
|
self.add_reviewers(config["default"])
|
||
|
|
|
||
|
|
if pull_request_author.lower() in self.reviewers_lower:
|
||
|
|
self._reviewers.remove(self.reviewers_lower.index(pull_request_author.lower()))
|
||
|
|
|
||
|
|
try:
|
||
|
|
reviewers = list(self._reviewers)
|
||
|
|
team_reviewers = list(self._team_reviewers)
|
||
|
|
self.logger.info(
|
||
|
|
f"Attempting to assign reviewers ({reviewers}) "
|
||
|
|
f"and team_reviewers ({team_reviewers})"
|
||
|
|
)
|
||
|
|
pull_request.create_review_request(reviewers, team_reviewers)
|
||
|
|
return 0
|
||
|
|
except GithubException:
|
||
|
|
self.logger.exception(f"Cannot assign review request for PR #{pr_number}")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
def run(self, args: Namespace):
|
||
|
|
repo = self.github.get_repo(args.repo_path)
|
||
|
|
|
||
|
|
if not repo:
|
||
|
|
self.logger.error("Repository not found!")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
if not args.config_path.exists():
|
||
|
|
self.logger.error(f"Config '{args.config_path}' not found!")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
with open(args.config_path, "r") as file:
|
||
|
|
config = yaml.safe_load(file)
|
||
|
|
|
||
|
|
exit(self.update_reviewers(config, repo, args.pr_number))
|