Added error handling to find_pr script (#14110)

pull/14240/head
Roy Lenferink 2019-05-08 23:12:36 +02:00 committed by Kubernetes Prow Robot
parent 017a224473
commit de62ee2d2b
1 changed files with 35 additions and 16 deletions

View File

@ -1,6 +1,7 @@
#!/usr/bin/env python
import os
import json
import click
import requests
@ -25,6 +26,11 @@ def main(tags, token, path, last_n_pr):
ex:
./find_pr.py --tags "language/fr" "content/fr/_index.html"
"""
if not token:
print("GitHub token not provided (required)")
exit(1)
query = Template("""
query {
repository(name: "website", owner: "kubernetes") {
@ -47,6 +53,8 @@ def main(tags, token, path, last_n_pr):
}
}
""").render(tags=tags, last_n_pr=last_n_pr)
try:
r = requests.post("https://api.github.com/graphql",
json={"query": query},
headers={
@ -54,6 +62,8 @@ def main(tags, token, path, last_n_pr):
"Accept": "application/vnd.github.ocelot-preview+json",
"Accept-Encoding": "gzip"
})
r.raise_for_status()
reply = r.json()
prs = reply['data']['repository']['pullRequests']['edges']
@ -65,6 +75,15 @@ def main(tags, token, path, last_n_pr):
print(pr["node"]["url"])
print("----------------")
except requests.exceptions.HTTPError as err:
gh_err_response = json.loads(err.response.text)
print("HTTP Error: %d %s" % (err.response.status_code, gh_err_response['message']))
except requests.exceptions.ConnectionError as err:
print("Error Connecting: %s" % err)
except requests.exceptions.Timeout as err:
print("Timeout Error: %s" % err)
except requests.exceptions.RequestException as err:
print("Oops, another error occurred: %s" % err)
if __name__ == '__main__':
main()