mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
from github import Github
|
|
from subprocess import check_output
|
|
import shlex
|
|
from tqdm import tqdm
|
|
import click
|
|
|
|
|
|
@click.command()
|
|
@click.option(
|
|
"--access-token",
|
|
required=True,
|
|
help="""
|
|
Github Access token that has repo:public_repo and user:read:user permission.
|
|
|
|
Create them at https://github.com/settings/tokens/new
|
|
""",
|
|
)
|
|
@click.option(
|
|
"--ray-path", required=True, help="File system path to the ray directory")
|
|
@click.option(
|
|
"--prev-branch",
|
|
required=True,
|
|
help="Previous version branch like ray-0.7.1")
|
|
@click.option(
|
|
"--curr-branch",
|
|
required=True,
|
|
help="Current version branch like ray-0.7.2")
|
|
def run(access_token, ray_path, prev_branch, curr_branch):
|
|
# Generate command
|
|
cmd = []
|
|
cmd.append(f"cd {ray_path}")
|
|
cmd.append(f'git log {prev_branch}..{curr_branch} --pretty=format:"%s" '
|
|
' | grep -Eo "#(\d+)"')
|
|
joined = " && ".join(cmd)
|
|
cmd = f"bash -c '{joined}'"
|
|
cmd = shlex.split(cmd)
|
|
print("Executing", cmd)
|
|
|
|
# Sort the PR numbers
|
|
pr_numbers = [
|
|
int(l.lstrip("#")) for l in check_output(cmd).decode().split()
|
|
]
|
|
print("PR numbers", pr_numbers)
|
|
|
|
# Use Github API to fetch the
|
|
g = Github(access_token)
|
|
ray_repo = g.get_repo("ray-project/ray")
|
|
logins = set()
|
|
for num in tqdm(pr_numbers):
|
|
try:
|
|
logins.add(ray_repo.get_pull(num).user.login)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
print()
|
|
print("Here's the list of contributors")
|
|
print("=" * 10)
|
|
print()
|
|
print("@" + ", @".join(logins))
|
|
print()
|
|
print("=" * 10)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|