build_packages.py

# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------

import argparse
import os
from subprocess import check_call
from pathlib import Path
from six import text_type


try:
    from packaging.version import parse as Version, InvalidVersion
except ImportError:  # Should not happen, but at worst in most case this is the same
    from pip._vendor.packaging.version import parse as Version, InvalidVersion

DEFAULT_DESTINATION_FOLDER = os.path.join("..", "dist")
package_list = ["Kqlmagic"]


def create_package(name, dest_folder=DEFAULT_DESTINATION_FOLDER):
    absdirpath = os.path.abspath(name)
    check_call(["python", "setup.py", "bdist_wheel", "-d", dest_folder], cwd=absdirpath)
    check_call(["python", "setup.py", "sdist", "-d", dest_folder], cwd=absdirpath)


def travis_build_package():
    """Assumed called on Travis, to prepare a package to be deployed
    This method prints on stdout for Travis.
    Return is obj to pass to sys.exit() directly
    """
    travis_tag = os.environ.get("TRAVIS_TAG")
    if not travis_tag:
        print("TRAVIS_TAG environment variable is not present")
        return "TRAVIS_TAG environment variable is not present"

    try:
        version = Version(travis_tag)
    except InvalidVersion:
        failure = "Version must be a valid PEP440 version (version is: {})".format(version)
        print(failure)
        return failure

    abs_dist_path = Path(os.environ["TRAVIS_BUILD_DIR"], "dist")
    [create_package(package, text_type(abs_dist_path)) for package in package_list]

    print("Produced:\n{}".format(list(abs_dist_path.glob("*"))))

    pattern = "*{}*".format(version)
    packages = list(abs_dist_path.glob(pattern))
    if not packages:
        return "Package version does not match tag {}, abort".format(version)
    pypi_server = os.environ.get("PYPI_SERVER", "default PyPI server")
    print("Package created as expected and will be pushed to {}".format(pypi_server))


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Build Azure package.")
    parser.add_argument("name", help="The package name")
    parser.add_argument(
        "--dest",
        "-d",
        default=DEFAULT_DESTINATION_FOLDER,
        help="Destination folder. Relative to the package dir. [default: %(default)s]",
    )

    args = parser.parse_args()
    if args.name == "all":
        for package in package_list:
            create_package(package, args.dest)
    else:
        create_package(args.name, args.dest)