mirror of https://github.com/apache/druid.git
61 lines
2.6 KiB
Python
61 lines
2.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
||
|
|
# contributor license agreements. See the NOTICE file distributed with
|
||
|
|
# this work for additional information regarding copyright ownership.
|
||
|
|
# The ASF licenses this file to You under the Apache License, Version 2.0
|
||
|
|
# (the "License"); you may not use this file except in compliance with
|
||
|
|
# the License. You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
if len(sys.argv) != 5:
|
||
|
|
sys.stderr.write('usage: program <github-username> <previous-release-commit> <new-release-commit> <milestone-number-to-tag>\n')
|
||
|
|
sys.stderr.write(" e.g., program myusername 75c70c2ccc 29f3a328da 30\n")
|
||
|
|
sys.stderr.write(" It is also necessary to set a GIT_TOKEN environment variable containing a personal access token.\n")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
github_username = sys.argv[1]
|
||
|
|
previous_release_commit = sys.argv[2]
|
||
|
|
new_release_commit = sys.argv[3]
|
||
|
|
milestone = sys.argv[4]
|
||
|
|
milestone_json = {'milestone': milestone}
|
||
|
|
|
||
|
|
# Find all commits between that commit and the current release branch
|
||
|
|
command = "git rev-list {}..{}".format(previous_release_commit, new_release_commit)
|
||
|
|
all_commits = subprocess.check_output(command, shell=True).decode('UTF-8')
|
||
|
|
|
||
|
|
for sha in all_commits.splitlines():
|
||
|
|
try:
|
||
|
|
url = "https://api.github.com/repos/apache/incubator-druid/commits/{}/pulls".format(sha)
|
||
|
|
headers = {'Accept': 'application/vnd.github.groot-preview+json'}
|
||
|
|
pull_requests = requests.get(url, headers=headers, auth=(github_username, os.environ["GIT_TOKEN"])).json()
|
||
|
|
|
||
|
|
print("Retrieved {} pull requests associated to commit {}".format(len(pull_requests), sha))
|
||
|
|
for pr in pull_requests:
|
||
|
|
pr_number = pr['number']
|
||
|
|
if pr['milestone'] is None:
|
||
|
|
print("Tagging Pull Request {} with milestone {}".format(pr_number, milestone))
|
||
|
|
url = "https://api.github.com/repos/apache/incubator-druid/issues/{}".format(pr_number)
|
||
|
|
requests.patch(url, json=milestone_json, auth=(github_username, os.environ["GIT_TOKEN"]))
|
||
|
|
else:
|
||
|
|
print("Skipping Pull Request {} since it's already tagged with milestone {}".format(pr_number, milestone))
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print("Got exception for commit: {} ex: {}".format(sha, e))
|
||
|
|
continue
|
||
|
|
|