Make yapf happy

pull/17505/head
Lidi Zheng 6 years ago
parent 0f0822d53f
commit 5a38d1956f
  1. 78
      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))

Loading…
Cancel
Save