build_test_tree.py

# Copyright Notice:
# Copyright 2017-2019 DMTF. All rights reserved.
# License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/Redfish-Test-Framework/blob/main/LICENSE.md

import argparse
import io
import os
import requests
import zipfile
import sys


def download_zip(subdir, zip_url):
    print('Extracting {} into {}'.format(zip_url, subdir if subdir is not None else '$PWD'))

    # Create subdirectory if specified
    if subdir is not None:
        try:
            subdir = os.path.abspath(subdir)
            if not os.path.isdir(subdir):
                os.mkdir(subdir)
        except OSError as e:
            print("Error creating target subdirectory {}, error: {}".format(subdir, e), file=sys.stderr)
            return
    else:
        subdir = os.getcwd()

    # Fetch zip
    try:
        r = requests.get(zip_url, stream=True)
        if r.status_code != requests.codes.ok:
            print("Unable to retrieve zip at {}. Status = {}, response = {}".format(zip_url, r.status_code, r.text),
                  file=sys.stderr)
            return
    except Exception as e:
        print('Unable to retrieve zip at {}. Exception is "{}"'.format(zip_url, e), file=sys.stderr)
        return

    # Extract zip
    try:
        z = zipfile.ZipFile(io.BytesIO(r.content), mode='r')
        z.extractall(path=subdir)
    except Exception as e:
        print('Unable to extract zip from {}. Exception is "{}"'.format(zip_url, e), file=sys.stderr)
        return


def main():
    arg_parser = argparse.ArgumentParser(
        description='Fetch Redfish tools and build test framework tree in current working directory')
    arg_parser.parse_args()

    download_list = [
        ['Schema-Validation', 'https://github.com/DMTF/Redfish-Service-Validator/archive/main.zip'],
        ['Other-Tests', 'https://github.com/DMTF/Redfish-Reference-Checker/archive/main.zip'],
        ['Other-Tests', 'https://github.com/DMTF/Redfish-Mockup-Creator/archive/main.zip'],
        ['Profile-Validation', 'https://github.com/DMTF/Redfish-Interop-Validator/archive/main.zip'],
        ['Protocol-Validation', 'https://github.com/DMTF/Redfish-Protocol-Validator/archive/main.zip'],
        [None, 'https://github.com/DMTF/Redfish-Usecase-Checkers/archive/main.zip']
    ]

    for subdir, zip_url in download_list:
        download_zip(subdir, zip_url)


if __name__ == "__main__":
    main()