diff options
author | Lidi Zheng <lidiz@google.com> | 2018-12-14 10:08:22 -0800 |
---|---|---|
committer | Lidi Zheng <lidiz@google.com> | 2018-12-14 10:08:22 -0800 |
commit | 5a38d1956fb7007108a310e0280df58c9c824cec (patch) | |
tree | 43bf0f624a45172e05fa54c7d5bafd20cf1fdf7f /tools/run_tests | |
parent | 0f0822d53f37ea9da8bf838e5b3c62b05afe2f10 (diff) |
Make yapf happy
Diffstat (limited to 'tools/run_tests')
-rw-r--r-- | tools/run_tests/python_utils/check_on_pr.py | 78 |
1 files changed, 49 insertions, 29 deletions
diff --git a/tools/run_tests/python_utils/check_on_pr.py b/tools/run_tests/python_utils/check_on_pr.py index 96418bf2d3..935d59713e 100644 --- a/tools/run_tests/python_utils/check_on_pr.py +++ b/tools/run_tests/python_utils/check_on_pr.py @@ -25,47 +25,57 @@ _GITHUB_API_PREFIX = 'https://api.github.com' _GITHUB_REPO = 'lidizheng/grpc' _GITHUB_APP_ID = 22288 _INSTALLATION_ID = 516307 -_GITHUB_APP_KEY = open(os.environ['HOME'] + '/.ssh/google-grpc-checker.2018-12-13.private-key.pem', 'r').read() +_GITHUB_APP_KEY = open( + os.environ['HOME'] + '/.ssh/google-grpc-checker.2018-12-13.private-key.pem', + 'r').read() _ACCESS_TOKEN_CACHE = None + def _jwt_token(): - return jwt.encode({ - 'iat': int(time.time()), - 'exp': int(time.time() + 60 * 10), # expire in 10 minutes - 'iss': _GITHUB_APP_ID, - }, _GITHUB_APP_KEY, algorithm='RS256') + return jwt.encode( + { + 'iat': int(time.time()), + 'exp': int(time.time() + 60 * 10), # expire in 10 minutes + 'iss': _GITHUB_APP_ID, + }, + _GITHUB_APP_KEY, + algorithm='RS256') + def _access_token(): global _ACCESS_TOKEN_CACHE if _ACCESS_TOKEN_CACHE == None or _ACCESS_TOKEN_CACHE['exp'] < time.time(): resp = requests.post( - url='https://api.github.com/app/installations/%s/access_tokens' % _INSTALLATION_ID, + url='https://api.github.com/app/installations/%s/access_tokens' % + _INSTALLATION_ID, headers={ 'Authorization': 'Bearer %s' % _jwt_token().decode('ASCII'), 'Accept': 'application/vnd.github.machine-man-preview+json', - } - ) - _ACCESS_TOKEN_CACHE = {'token': resp.json()['token'], 'exp': time.time()+60} + }) + _ACCESS_TOKEN_CACHE = { + 'token': resp.json()['token'], + 'exp': time.time() + 60 + } return _ACCESS_TOKEN_CACHE['token'] + def _call(url, method='GET', json=None): if not url.startswith('https://'): url = _GITHUB_API_PREFIX + url - headers={ + headers = { 'Authorization': 'Bearer %s' % _access_token(), 'Accept': 'application/vnd.github.antiope-preview+json', } - return requests.request( - method=method, - url=url, - headers=headers, - json=json) + return requests.request(method=method, url=url, headers=headers, json=json) + def _latest_commit(): - resp = _call('/repos/%s/pulls/%s/commits' % (_GITHUB_REPO, os.environ['ghprbPullId'])) + resp = _call('/repos/%s/pulls/%s/commits' % (_GITHUB_REPO, + os.environ['ghprbPullId'])) return resp.json()[-1] + def check_on_pr(name, summary, success=True): """Create/Update a check on current pull request. @@ -84,15 +94,25 @@ def check_on_pr(name, summary, success=True): print('Missing ghprbPullId env var: not commenting') return commit = _latest_commit() - resp = _call('/repos/%s/check-runs' % _GITHUB_REPO, method='POST', json={ - 'name': name, - 'head_sha': commit['sha'], - 'status': 'completed', - 'completed_at': '%sZ' % datetime.datetime.utcnow().replace(microsecond=0).isoformat(), - 'conclusion': 'success' if success else 'failure', - 'output': { - 'title': name, - 'summary': summary, - } - }) - print('Result of Creating/Updating Check on PR:', json.dumps(resp.json(), indent=2)) + resp = _call( + '/repos/%s/check-runs' % _GITHUB_REPO, + method='POST', + json={ + 'name': + name, + 'head_sha': + commit['sha'], + 'status': + 'completed', + 'completed_at': + '%sZ' % + datetime.datetime.utcnow().replace(microsecond=0).isoformat(), + 'conclusion': + 'success' if success else 'failure', + 'output': { + 'title': name, + 'summary': summary, + } + }) + print('Result of Creating/Updating Check on PR:', + json.dumps(resp.json(), indent=2)) |