def main()

in python/pip_install/extract_wheels/__init__.py [0:0]

• 58 lines of code
• 7 McCabe index (conditional complexity)

def main() -> None:
    """Main program.

    Exits zero on successful program termination, non-zero otherwise.
    """

    configure_reproducible_wheels()

    parser = argparse.ArgumentParser(
        description="Resolve and fetch artifacts transitively from PyPI"
    )
    parser.add_argument(
        "--requirements",
        action="store",
        required=True,
        help="Path to requirements.txt from where to install dependencies",
    )
    parser.add_argument(
        "--annotations",
        type=annotation.annotations_map_from_str_path,
        help="A json encoded file containing annotations for rendered packages.",
    )
    arguments.parse_common_args(parser)
    args = parser.parse_args()
    deserialized_args = dict(vars(args))
    arguments.deserialize_structured_args(deserialized_args)

    # Pip is run with the working directory changed to the folder containing the requirements.txt file, to allow for
    # relative requirements to be correctly resolved. The --wheel-dir is therefore required to be repointed back to the
    # current calling working directory (the repo root in .../external/name), where the wheel files should be written to
    pip_args = (
        [sys.executable, "-m", "pip"]
        + (["--isolated"] if args.isolated else [])
        + ["wheel", "-r", args.requirements]
        + ["--wheel-dir", os.getcwd()]
        + deserialized_args["extra_pip_args"]
    )

    env = os.environ.copy()
    env.update(deserialized_args["environment"])

    # Assumes any errors are logged by pip so do nothing. This command will fail if pip fails
    subprocess.run(
        pip_args,
        check=True,
        env=env,
        cwd=str(pathlib.Path(args.requirements).parent.resolve()),
    )

    extras = requirements.parse_extras(args.requirements)

    repo_label = "@%s" % args.repo

    # Locate all wheels
    wheels = [whl for whl in glob.glob("*.whl")]

    # Collect all annotations
    reqs = {whl: wheel.Wheel(whl).name for whl in wheels}
    annotations = args.annotations.collect(reqs.values())

    targets = [
        '"{}{}"'.format(
            repo_label,
            bazel.extract_wheel(
                wheel_file=whl,
                extras=extras,
                pip_data_exclude=deserialized_args["pip_data_exclude"],
                enable_implicit_namespace_pkgs=args.enable_implicit_namespace_pkgs,
                repo_prefix=args.repo_prefix,
                annotation=annotations.get(name),
            ),
        )
        for whl, name in reqs.items()
    ]

    with open("requirements.bzl", "w") as requirement_file:
        requirement_file.write(
            bazel.generate_requirements_file_contents(repo_label, targets)
        )