ceph-csi/scripts/get_github_labels.py
Niels de Vos aa062513c0 ci: report connections failures with GitHub API
On occasion the scripts that use the GitHub API fail without logging any
useful error:

    ./scripts/get_github_labels.py --id=1568 --has-label=ci/skip/e2e
    Traceback (most recent call last):
      File "./scripts/get_github_labels.py", line 71, in <module>
        main()
      File "./scripts/get_github_labels.py", line 55, in main
        names = get_names(json)
      File "./scripts/get_github_labels.py", line 40, in get_names
        names.append(label['name'])
    TypeError: string indices must be integers

While debugging, it seems that the limit of API calls is reached:

    403 Client Error: rate limit exceeded

It is useful to have failure messages reported in the output of the CI
jobs for future potential troubleshooting.

Signed-off-by: Niels de Vos <ndevos@redhat.com>
2020-10-13 12:05:57 +00:00

86 lines
2.2 KiB
Python
Executable File

#!/usr/bin/python
'''
Fetches the labels of an Issue or Pull-Request from GitHub.
Parameters:
--id=<id>: the number of the the GitHub Issue or Pull-Request
--has-label=<label>: label to check (exit 0 if set, 2 unset), without
--has-label, all labels of the Issue or Pull-Request
get printed.
Exit codes:
0: success
1: any unexpected failure
2: --has-label=<label> was passed, <label> is not set on the PR
'''
import argparse
import sys
import requests
LABEL_URL_FMT = 'https://api.github.com/repos/ceph/ceph-csi/issues/%s/labels'
'''
Formatted URL for fetching the labels, replace '%s' with the number of the
Issue or Pull-Request.
'''
def get_json_labels(gh_id):
'''
Fetch the labels from GitHub, return the full JSON structures that were
obtained. These include several ID's, color, decription, name and more.
'''
url = LABEL_URL_FMT % gh_id
headers = {'Accept': 'application/vnd.github.v3+json'}
res = requests.get(url, headers=headers)
# if "res.status_code != requests.codes.ok", raise an exception
res.raise_for_status()
return res.json()
def get_names(gh_labels):
'''
Take the JSON formatted labels, and return a list of the name for each label.
'''
names = list()
for label in gh_labels:
names.append(label['name'])
return names
def main():
'''
main() function to parse arguments and run the actions.
'''
parser = argparse.ArgumentParser()
parser.add_argument('--id', type=int, required=True, help='ID of the Issue or Pull-Request')
parser.add_argument('--has-label', help='check if the Issue or Pull-Request has the label set')
args = parser.parse_args()
# get the labels for the issue
try:
json = get_json_labels(args.id)
except Exception as err:
print('Error: %s' % err)
sys.exit(1)
names = get_names(json)
# in case --has-label is passed, exit with 0 or 2
if args.has_label:
if args.has_label in names:
sys.exit(0)
else:
sys.exit(2)
# --has-label was not passed, list all labels
else:
for name in names:
print(name)
sys.exit(0)
if __name__ == '__main__':
main()