Verified Commit 7eae6969 authored by Dom Sekotill's avatar Dom Sekotill
Browse files

Add a --builder argument

parent 319de624

build_wheel/__main__.py

+5 −1
Original line number Diff line number Diff line
@@ -19,8 +19,9 @@ PLATFORM_OPTS = ["all", *PLATFORMS] 
@click.command

@click.option("--output", type=click.Path(file_okay=False), default="dist")

@click.option("--platform", "platforms", type=click.Choice(PLATFORM_OPTS), multiple=True, default=["all"])

@click.option("--builder")

@click.argument("packages", nargs=-1, required=True)

def cli(output: str, platforms: Sequence[str], packages: Sequence[str]) -> None:

def cli(output: str, platforms: Sequence[str], packages: Sequence[str], builder: str|None) -> None:

	config_resource = resources.files(__package__) / "bake.hcl"



	if "all" in platforms:
@@ -31,6 +32,9 @@ def cli(output: str, platforms: Sequence[str], packages: Sequence[str]) -> None: 
		*(f"--set=*.platform={platform}" for platform in platforms),

	]



	if builder:

		cmd.append(f"--builder={builder}")



	with resources.as_file(config_resource) as config:

		cmd += ["--file", config]

		env = os.environ.copy()