From de62ee2d2b239510162044319894506c5c3145ba Mon Sep 17 00:00:00 2001
From: Roy Lenferink <lenferinkroy@gmail.com>
Date: Wed, 8 May 2019 23:12:36 +0200
Subject: [PATCH] Added error handling to find_pr script (#14110)

---
 scripts/find_pr.py | 51 +++++++++++++++++++++++++++++++---------------
 1 file changed, 35 insertions(+), 16 deletions(-)

diff --git a/scripts/find_pr.py b/scripts/find_pr.py
index 503cf3121cabc..f05a790d759b6 100755
--- a/scripts/find_pr.py
+++ b/scripts/find_pr.py
@@ -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,24 +53,37 @@ def main(tags, token, path, last_n_pr):
       }
     }
     """).render(tags=tags, last_n_pr=last_n_pr)
-    r = requests.post("https://api.github.com/graphql",
-                      json={"query": query},
-                      headers={
-                          "Authorization": "token %s" % token,
-                          "Accept": "application/vnd.github.ocelot-preview+json",
-                          "Accept-Encoding": "gzip"
-                      })
-    reply = r.json()
-    prs = reply['data']['repository']['pullRequests']['edges']
 
-    for pr in prs:
-        files = pr["node"]["files"]["edges"]
-        for f in files:
-            if path == f["node"]["path"]:
-                print("%s (%s)" % (pr["node"]["title"], pr["node"]["state"]))
-                print(pr["node"]["url"])
-                print("----------------")
+    try:
+        r = requests.post("https://api.github.com/graphql",
+                          json={"query": query},
+                          headers={
+                              "Authorization": "token %s" % token,
+                              "Accept": "application/vnd.github.ocelot-preview+json",
+                              "Accept-Encoding": "gzip"
+                          })
+        r.raise_for_status()
+
+        reply = r.json()
+        prs = reply['data']['repository']['pullRequests']['edges']
+
+        for pr in prs:
+            files = pr["node"]["files"]["edges"]
+            for f in files:
+                if path == f["node"]["path"]:
+                    print("%s (%s)" % (pr["node"]["title"], pr["node"]["state"]))
+                    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()