import argparse
|
import sys
|
|
from ._implementation import resolve
|
from ._toml_compat import tomllib
|
|
|
def main() -> None:
|
if tomllib is None:
|
print(
|
"Usage error: dependency-groups CLI requires tomli or Python 3.11+",
|
file=sys.stderr,
|
)
|
raise SystemExit(2)
|
|
parser = argparse.ArgumentParser(
|
description=(
|
"A dependency-groups CLI. Prints out a resolved group, newline-delimited."
|
)
|
)
|
parser.add_argument(
|
"GROUP_NAME", nargs="*", help="The dependency group(s) to resolve."
|
)
|
parser.add_argument(
|
"-f",
|
"--pyproject-file",
|
default="pyproject.toml",
|
help="The pyproject.toml file. Defaults to trying in the current directory.",
|
)
|
parser.add_argument(
|
"-o",
|
"--output",
|
help="An output file. Defaults to stdout.",
|
)
|
parser.add_argument(
|
"-l",
|
"--list",
|
action="store_true",
|
help="List the available dependency groups",
|
)
|
args = parser.parse_args()
|
|
with open(args.pyproject_file, "rb") as fp:
|
pyproject = tomllib.load(fp)
|
|
dependency_groups_raw = pyproject.get("dependency-groups", {})
|
|
if args.list:
|
print(*dependency_groups_raw.keys())
|
return
|
if not args.GROUP_NAME:
|
print("A GROUP_NAME is required", file=sys.stderr)
|
raise SystemExit(3)
|
|
content = "\n".join(resolve(dependency_groups_raw, *args.GROUP_NAME))
|
|
if args.output is None or args.output == "-":
|
print(content)
|
else:
|
with open(args.output, "w", encoding="utf-8") as fp:
|
print(content, file=fp)
|
|
|
if __name__ == "__main__":
|
main()
|
