123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
# Usage: python3 rerun-apps-version-check.py -t <github token> -r <repo>
|
|
#
|
|
# Default repo is emqx/emqx
|
|
#
|
|
import requests
|
|
import http.client
|
|
import json
|
|
import os
|
|
import sys
|
|
import time
|
|
import math
|
|
import inspect
|
|
from optparse import OptionParser
|
|
from urllib.parse import urlparse, parse_qs
|
|
from requests.adapters import HTTPAdapter
|
|
from requests.packages.urllib3.util.retry import Retry
|
|
|
|
user_agent = sys.argv[0].split('/')[-1]
|
|
|
|
def query(owner, repo):
|
|
return """
|
|
query {
|
|
repository(owner: "%s", name: "%s") {
|
|
pullRequests(first: 25, states: OPEN, orderBy: {field:CREATED_AT, direction:DESC}) {
|
|
nodes {
|
|
url
|
|
commits(last: 1) {
|
|
nodes {
|
|
commit {
|
|
checkSuites(first: 25) {
|
|
nodes {
|
|
url
|
|
checkRuns(first: 1, filterBy: {checkName: "check_apps_version"}) {
|
|
nodes {
|
|
url
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
""" % (owner, repo)
|
|
|
|
|
|
def get_headers(token: str):
|
|
return {'Accept': 'application/vnd.github+json',
|
|
'Authorization': f'Bearer {token}',
|
|
'X-GitHub-Api-Version': '2022-11-28',
|
|
'User-Agent': f'{user_agent}'
|
|
}
|
|
|
|
def get_session():
|
|
session = requests.Session()
|
|
|
|
retries = Retry(total=10,
|
|
backoff_factor=1, # 1s
|
|
allowed_methods=None,
|
|
status_forcelist=[ 429, 500, 502, 503, 504 ]) # Retry on these status codes
|
|
|
|
session.mount('https://', HTTPAdapter(max_retries=retries))
|
|
|
|
return session
|
|
|
|
def get_check_suite_ids(token: str, repo: str):
|
|
session = get_session()
|
|
url = f'https://api.github.com/graphql'
|
|
[repo_owner, repo_repo] = repo.split('/')
|
|
r = session.post(url, headers=get_headers(token), json={'query': query(repo_owner, repo_repo)})
|
|
if r.status_code == 200:
|
|
resp = r.json()
|
|
if not 'data' in resp:
|
|
print(f'Failed to fetch check runs: {r.status_code}\n{r.json()}')
|
|
sys.exit(1)
|
|
result = []
|
|
for pr in resp['data']['repository']['pullRequests']['nodes']:
|
|
if not pr['commits']['nodes']:
|
|
continue
|
|
if not pr['commits']['nodes'][0]['commit']['checkSuites']['nodes']:
|
|
continue
|
|
for node in pr['commits']['nodes'][0]['commit']['checkSuites']['nodes']:
|
|
if node['checkRuns']['nodes']:
|
|
url_parsed = urlparse(node['url'])
|
|
params = parse_qs(url_parsed.query)
|
|
check_suite_id = params['check_suite_id'][0]
|
|
result.extend([(check_suite_id, pr['url'], node['checkRuns']['nodes'][0]['url'])])
|
|
return result
|
|
else:
|
|
print(f'Failed to fetch check runs: {r.status_code}\n{r.text}')
|
|
sys.exit(1)
|
|
|
|
def rerequest_check_suite(token: str, repo: str, check_suite_id: str):
|
|
session = get_session()
|
|
url = f'https://api.github.com/repos/{repo}/check-suites/{check_suite_id}/rerequest'
|
|
r = session.post(url, headers=get_headers(token))
|
|
if r.status_code == 201:
|
|
print(f'Successfully triggered {url}')
|
|
else:
|
|
print(f'Failed to trigger {url}: {r.status_code}\n{r.text}')
|
|
|
|
def main():
|
|
parser = OptionParser()
|
|
parser.add_option("-r", "--repo", dest="repo",
|
|
help="github repo", default="emqx/emqx")
|
|
parser.add_option("-t", "--token", dest="gh_token",
|
|
help="github API token")
|
|
(options, args) = parser.parse_args()
|
|
|
|
# Get github token from env var if provided, else use the one from command line.
|
|
# The token must be exported in the env from ${{ secrets.GITHUB_TOKEN }} in the workflow.
|
|
token = os.environ['GITHUB_TOKEN'] if 'GITHUB_TOKEN' in os.environ else options.gh_token
|
|
for id, pr_url, check_run_url in get_check_suite_ids(token, options.repo):
|
|
print(f'Attempting to re-request {check_run_url} for {pr_url}')
|
|
rerequest_check_suite(token, options.repo, id)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|