Merge pull request #88 from ohenak/fix/ghe-get-401-error

[Chore] Provide token when getting 401 error
2022-09-22 03:59:00 -06:00
parent e64deeba68 667293445b
commit 415544869c
1 changed files with 1 additions and 1 deletions
--- a/pull-request.py
+++ b/pull-request.py
@@ -239,7 +239,7 @@ def find_default_branch():
    response = requests.get(REPO_URL)

    # Case 1: 404 might need a token
-    if response.status_code == 404:
+    if response.status_code in [401, 404]:
        response = requests.get(REPO_URL, headers=HEADERS)
    if response.status_code != 200:
        abort_if_fail(response, "Unable to retrieve default branch")