diff --git a/README.md b/README.md index cd7b08e..87d036f 100644 --- a/README.md +++ b/README.md @@ -64,9 +64,15 @@ function signature as explained in this ### Accessing dependencies and products in the script Dependencies and products registered in the task function signature are used by pytask -to order tasks and track whether they are up-to-date. They are not automatically passed -to the Stata script. Use the `options` argument of the decorator to pass paths or other -values as command line arguments to your Stata executable. +to order tasks and track whether they are up-to-date. If `options` is not used, +pytask-stata writes all dependencies and products from the task function signature to a +YAML file and passes the path to this file as the first command line argument to the +do-file. + +#### Command Line Arguments + +Use the `options` argument of the decorator to keep the legacy interface and pass paths +or other values as command line arguments to your Stata executable. For example, pass the path of the product with @@ -112,6 +118,36 @@ def task_run_do_file(produces: Path = BLD / "auto.dta"): pass ``` +#### YAML Configuration Files + +By default, pytask-stata serializes all task keyword arguments and passes the path to +the generated YAML file as the first argument to the do-file. To read the file inside +Stata, install the user-written `yaml` package. + +```stata +ssc install yaml +``` + +Then read the configuration file in the Stata task. + +```python +@mark.stata(script=Path("script.do")) +def task_run_do_file(produces: Path = Path("auto.dta")): + pass +``` + +```do +args config +yaml read using "`config'", locals replace + +sysuse auto, clear +save "`r(yaml_produces)'" +``` + +Do not combine both interfaces. If `options` is supplied, pytask-stata assumes the +do-file receives all required values through command line arguments and does not create +a YAML configuration file. + ### Repeating tasks with different scripts or inputs You can also parametrize the execution of scripts, meaning executing multiple do-files diff --git a/justfile b/justfile index 483e4e3..d4dee13 100644 --- a/justfile +++ b/justfile @@ -12,7 +12,7 @@ test-ci: # Run type checking typing: - uv run --group typing --group test --isolated ty check + uv run --group typing --group test --group test-mock-stata --isolated ty check # Run linting and formatting lint: diff --git a/packages/stata_mock/src/stata_mock/cli.py b/packages/stata_mock/src/stata_mock/cli.py index 43370a6..9877fde 100644 --- a/packages/stata_mock/src/stata_mock/cli.py +++ b/packages/stata_mock/src/stata_mock/cli.py @@ -4,7 +4,10 @@ import re import sys +from dataclasses import dataclass +from dataclasses import field from pathlib import Path +from typing import Any INVALID_SYNTAX = 198 UNKNOWN_COMMAND = 199 @@ -44,33 +47,70 @@ def _parse_invocation(args: list[str]) -> tuple[Path, list[str], Path] | None: def _run_script(script: Path, options: list[str]) -> tuple[list[str], int | None]: lines = [f"running mock Stata for {script.name}"] - macros: dict[str, str] = {} + state = RuntimeState() - for raw_line in script.read_text().splitlines(): + for raw_line in _iter_stata_lines(script): line = raw_line.strip() if not line or line.startswith("*"): continue - line = _expand_local_macros(line, macros) + line = _expand_macros(line, state) lines.append(f". {line}") - error_code = _execute_line(line, macros, options) + error_code = _execute_line(line, state, options) if error_code is not None: return lines, error_code return lines, None -def _execute_line(line: str, macros: dict[str, str], options: list[str]) -> int | None: +def _iter_stata_lines(script: Path) -> list[str]: + lines: list[str] = [] + continued = "" + for raw_line in script.read_text().splitlines(): + stripped = raw_line.rstrip() + if stripped.endswith("///"): + continued += stripped.removesuffix("///").rstrip() + " " + continue + lines.append(continued + raw_line if continued else raw_line) + continued = "" + if continued: + lines.append(continued) + return lines + + +@dataclass +class YamlEntry: + """A flattened YAML key/value record.""" + + key: str + value: str + level: int + parent: str + type: str + + +@dataclass +class RuntimeState: + """Mutable state for a mock Stata do-file run.""" + + macros: dict[str, str] = field(default_factory=dict) + returns: dict[str, str] = field(default_factory=dict) + yaml_entries: dict[str, YamlEntry] = field(default_factory=dict) + + +def _execute_line(line: str, state: RuntimeState, options: list[str]) -> int | None: command, _, rest = line.partition(" ") command = command.lower() rest = rest.strip() if command == "args": - macros.update(dict(zip(rest.split(), options, strict=False))) + state.macros.update(dict(zip(rest.split(), options, strict=False))) elif command == "sysuse": return None elif command == "save": return _save_dataset(rest) + elif command == "yaml": + return _execute_yaml(rest, state) elif command in {"error", "exit"}: return int(rest.split()[0]) else: @@ -94,8 +134,267 @@ def _save_dataset(rest: str) -> int | None: return None -def _expand_local_macros(line: str, macros: dict[str, str]) -> str: - return re.sub(r"`([^']+)'", lambda match: macros.get(match.group(1), ""), line) +def _execute_yaml(rest: str, state: RuntimeState) -> int | None: + match = re.match(r"(?P\w+)(?P.*)", rest) + if match is None: + return INVALID_SYNTAX + subcommand = match.group("subcommand") + remaining = match.group("remaining") + subcommand = {"check": "validate", "desc": "describe"}.get( + subcommand.lower(), + subcommand.lower(), + ) + remaining = remaining.strip() + + if subcommand == "read": + return _yaml_read(remaining, state) + if subcommand == "get": + return _yaml_get(remaining, state) + if subcommand == "validate": + return _yaml_validate(remaining, state) + if subcommand in {"describe", "list", "dir", "frames", "clear"}: + return None + return UNKNOWN_COMMAND + + +def _yaml_read(rest: str, state: RuntimeState) -> int | None: + match = re.match( + r'^using\s+("?)(?P.+?)\1(?:\s*,\s*(?P.*))?$', + rest, + ) + if match is None: + return INVALID_SYNTAX + + path = Path(match.group("filename")) + if not path.is_absolute(): + path = Path.cwd() / path + try: + data = _parse_yaml_subset(path.read_text()) + except (OSError, ValueError): + return INVALID_SYNTAX + + entries = _flatten_yaml(data) + state.yaml_entries = {entry.key: entry for entry in entries} + options = _parse_options(match.group("options") or "") + prefix = options.get("prefix", "yaml_") + + state.returns.update( + { + "filename": path.as_posix(), + "n_keys": str(len(entries)), + "max_level": str(max((entry.level for entry in entries), default=0)), + "yaml_mode": "canonical", + "cache_hit": "0", + } + ) + if "locals" in options: + state.returns.update( + { + f"{prefix}{entry.key}": entry.value + for entry in entries + if entry.type != "parent" + } + ) + return None + + +def _yaml_get(rest: str, state: RuntimeState) -> int | None: + query, options = _split_command_options(rest) + query = query.strip() + if not query: + return INVALID_SYNTAX + + parent, _, key = query.partition(":") + if key: + search_key = f"{parent}_{key}" + result_name = key + else: + search_key = parent + result_name = parent.rsplit("_", maxsplit=1)[-1] + + entry = state.yaml_entries.get(search_key) + children = _get_yaml_child_attributes(search_key, state) + state.returns.update( + { + "key": key or parent, + "parent": parent if key else "", + "found": "1" if entry is not None or children else "0", + "n_attrs": str(len(children) or int(entry is not None)), + } + ) + if children: + attributes = str(options.get("attributes", "")).split() + allowed = set(attributes) if attributes else set(children) + state.returns.update( + {attr: value for attr, value in children.items() if attr in allowed} + ) + elif entry is not None: + state.returns[result_name] = entry.value + return None + + +def _get_yaml_child_attributes( + search_key: str, + state: RuntimeState, +) -> dict[str, str]: + prefix = f"{search_key}_" + attributes: dict[str, str] = {} + for key, entry in state.yaml_entries.items(): + if not key.startswith(prefix) or entry.type == "parent": + continue + attribute = key.removeprefix(prefix) + if "_" not in attribute: + attributes[attribute] = entry.value + return attributes + + +def _yaml_validate(rest: str, state: RuntimeState) -> int | None: + _, options = _split_command_options(rest) + valid = True + + required = str(options.get("required", "")).split() + for key in required: + valid = valid and key in state.yaml_entries + + for spec in str(options.get("types", "")).split(): + key, separator, expected_type = spec.partition(":") + if not separator: + return INVALID_SYNTAX + entry = state.yaml_entries.get(key) + valid = valid and entry is not None and entry.type == expected_type + + state.returns["valid"] = "1" if valid else "0" + return None if valid else INVALID_SYNTAX + + +def _parse_yaml_subset(text: str) -> dict[str, Any]: + if re.search(r"(^|\s)(---|\{|\}|\[[^\]]*\]|&\w+|\*\w+)", text): + msg = "Unsupported YAML syntax." + raise ValueError(msg) + + root: dict[str, Any] = {} + stack: list[tuple[int, dict[str, Any]]] = [(-1, root)] + + for raw_line in text.splitlines(): + if not raw_line.strip() or raw_line.lstrip().startswith("#"): + continue + indent = len(raw_line) - len(raw_line.lstrip(" ")) + if indent % 2: + msg = "Only two-space indentation is supported." + raise ValueError(msg) + + stripped = raw_line.strip() + if stripped.startswith("- "): + msg = "Top-level list items are not supported." + raise ValueError(msg) + key, separator, value = stripped.partition(":") + if not separator or not key or " " in key: + msg = "Invalid YAML mapping key." + raise ValueError(msg) + + while stack and indent <= stack[-1][0]: + stack.pop() + parent = stack[-1][1] + if value.strip() == "": + child: dict[str, Any] = {} + parent[key] = child + stack.append((indent, child)) + else: + parent[key] = _parse_yaml_scalar(value.strip()) + + return root + + +def _parse_yaml_scalar(value: str) -> str | int | float | bool | None: + value = value.strip("\"'") + if value.lower() in {"true", "false"}: + return value.lower() == "true" + if value.lower() in {"null", "~"}: + return None + try: + return int(value) + except ValueError: + try: + return float(value) + except ValueError: + return value + + +def _flatten_yaml(data: dict[str, Any]) -> list[YamlEntry]: + entries: list[YamlEntry] = [] + + def recurse(value: Any, prefix: str, level: int, parent: str) -> None: + if isinstance(value, dict): + if prefix: + entries.append(YamlEntry(prefix, "", level, parent, "parent")) + for key, child in value.items(): + child_key = f"{prefix}_{key}" if prefix else key + recurse(child, child_key, level + 1, prefix) + return + + entries.append( + YamlEntry( + key=prefix, + value=_format_yaml_value(value), + level=level, + parent=parent, + type=_yaml_value_type(value), + ) + ) + + recurse(data, "", 0, "") + return entries + + +def _format_yaml_value(value: Any) -> str: + if isinstance(value, bool): + return "true" if value else "false" + if value is None: + return "" + return str(value) + + +def _yaml_value_type(value: Any) -> str: + if isinstance(value, bool): + return "boolean" + if isinstance(value, int | float): + return "numeric" + if value is None: + return "null" + return "string" + + +def _split_command_options(rest: str) -> tuple[str, dict[str, str | bool]]: + command, separator, options = rest.partition(",") + return command.strip(), _parse_options(options if separator else "") + + +def _parse_options(options: str) -> dict[str, str | bool]: + parsed: dict[str, str | bool] = {} + for option in _split_options(options): + if match := re.match(r"(?P\w+)\((?P.*)\)$", option): + parsed[_normalize_option_name(match.group("name"))] = match.group("value") + elif option: + parsed[_normalize_option_name(option)] = True + return parsed + + +def _split_options(options: str) -> list[str]: + return re.findall(r"\w+\([^)]*\)|\w+", options) + + +def _normalize_option_name(name: str) -> str: + return {"l": "locals", "s": "scalars", "q": "quiet"}.get(name, name) + + +def _expand_macros(line: str, state: RuntimeState) -> str: + def replace(match: re.Match[str]) -> str: + name = match.group(1) + if r_match := re.fullmatch(r"r\(([^)]+)\)", name): + return state.returns.get(r_match.group(1), "") + return state.macros.get(name, "") + + return re.sub(r"`([^']+)'", replace, line) def _parse_save_target(rest: str) -> str: diff --git a/pyproject.toml b/pyproject.toml index afb95b8..2012da9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,7 +15,7 @@ classifiers = [ "Programming Language :: Python :: 3 :: Only", ] requires-python = ">=3.10" -dependencies = ["click>=8.1.8,!=8.2.0", "pytask>=0.5.2"] +dependencies = ["click>=8.1.8,!=8.2.0", "pytask>=0.5.2", "pyyaml"] dynamic = ["version"] [project.readme] diff --git a/src/pytask_stata/collect.py b/src/pytask_stata/collect.py index 0041849..ad13642 100644 --- a/src/pytask_stata/collect.py +++ b/src/pytask_stata/collect.py @@ -24,6 +24,8 @@ from pytask import parse_products_from_task_function from pytask import remove_marks +from pytask_stata.serialization import SERIALIZERS +from pytask_stata.serialization import create_path_to_serialized from pytask_stata.shared import convert_task_id_to_name_of_log_file from pytask_stata.shared import stata @@ -32,11 +34,21 @@ def run_stata_script( _executable: str, _script: Path, _options: list[str], + _serialized: Path | None, _log_name: str, _cwd: Path, ) -> None: - """Run an R script.""" - cmd = [_executable, "-e", "do", _script.as_posix(), *_options, f"-{_log_name}"] + """Run a Stata do-file.""" + serialized_option = [] if _serialized is None else [_serialized.as_posix()] + cmd = [ + _executable, + "-e", + "do", + _script.as_posix(), + *_options, + *serialized_option, + f"-{_log_name}", + ] print("Executing " + " ".join(cmd) + ".") # noqa: T201 subprocess.run(cmd, cwd=_cwd, check=True) # noqa: S603 @@ -63,7 +75,8 @@ def pytask_collect_task( raise ValueError(msg) mark = _parse_stata_mark(mark=marks[0]) - script, options = stata(**marks[0].kwargs) + script = mark.kwargs["script"] + options = mark.kwargs["options"] or [] cast("Any", obj).pytask_meta.markers.append(mark) # Collect the nodes in @pytask.mark.julia and validate them. @@ -182,12 +195,73 @@ def pytask_collect_task( ) task.depends_on["_log_name"] = log_name_node + suffix = mark.kwargs["suffix"] + serialized = None if suffix is None else create_path_to_serialized(task, suffix) + task.depends_on["_serialized"] = _collect_serialized_node( + session=session, + path_nodes=path_nodes, + node_info=NodeInfo( + arg_name="_serialized", + path=(), + value=PythonNode(value=serialized), + task_path=path, + task_name=name, + ), + ) + return task return None +def _collect_serialized_node( + *, + session: Session, + path_nodes: Path, + node_info: NodeInfo, +) -> PythonNode: + """Collect the node for the serialized task configuration.""" + serialized_node = session.hook.pytask_collect_node( + session=session, + path=path_nodes, + node_info=node_info, + ) + return cast("PythonNode", serialized_node) + + def _parse_stata_mark(mark: Mark) -> Mark: """Parse a Stata mark.""" - script, options = stata(**mark.kwargs) - parsed_kwargs = {"script": script or None, "options": options or []} + script, options, serializer, suffix = stata(**mark.kwargs) + + if options is not None and (serializer is not None or suffix is not None): + msg = ( + "The @pytask.mark.stata interfaces cannot be mixed. Use either " + "'options' for command line arguments or 'serializer'/'suffix' for " + "serialized task arguments." + ) + raise ValueError(msg) + + if options is None and serializer is None: + serializer = "yaml" + + if isinstance(serializer, str) and serializer not in SERIALIZERS: + msg = ( + f"Serializer {serializer!r} is not known. Available serializers are " + f"{sorted(SERIALIZERS)}." + ) + raise ValueError(msg) + + proposed_suffix = ( + SERIALIZERS[serializer]["suffix"] + if isinstance(serializer, str) and serializer in SERIALIZERS and suffix is None + else suffix + ) + if serializer is not None and proposed_suffix is None: + msg = "Missing suffix for serialized Stata task." + raise ValueError(msg) + parsed_kwargs = { + "script": script or None, + "options": options, + "serializer": serializer, + "suffix": proposed_suffix, + } return Mark("stata", (), parsed_kwargs) diff --git a/src/pytask_stata/execute.py b/src/pytask_stata/execute.py index 07b4925..8ceaf22 100644 --- a/src/pytask_stata/execute.py +++ b/src/pytask_stata/execute.py @@ -4,16 +4,20 @@ import re from pathlib import Path +from typing import Any from typing import cast from pytask import PathNode +from pytask import PPathNode from pytask import PTask from pytask import PTaskWithPath from pytask import PythonNode from pytask import Session from pytask import has_mark from pytask import hookimpl +from pytask.tree_util import tree_map +from pytask_stata.serialization import serialize_keyword_arguments from pytask_stata.shared import STATA_COMMANDS @@ -29,6 +33,47 @@ def pytask_execute_task_setup(session: Session, task: PTask) -> None: ) raise RuntimeError(msg) + marks = list(task.markers) + stata_marks = [mark for mark in marks if mark.name == "stata"] + if stata_marks: + serializer = stata_marks[0].kwargs.get("serializer") + if serializer is None: + return + + serialized_node = task.depends_on["_serialized"] + if not isinstance(serialized_node, PythonNode) or not isinstance( + serialized_node.value, Path + ): + msg = ( + "Expected '_serialized' dependency to be a PythonNode " + "with a pathlib.Path value." + ) + raise TypeError(msg) + + path_to_serialized = serialized_node.value + path_to_serialized.parent.mkdir(parents=True, exist_ok=True) + kwargs = _collect_stata_keyword_arguments(task) + serialize_keyword_arguments(serializer, path_to_serialized, kwargs) + + +def _collect_stata_keyword_arguments(task: PTask) -> dict[str, Any]: + """Collect keyword arguments passed to the Stata config file.""" + kwargs: dict[str, Any] = { + **tree_map( + lambda node: ( + node.path.as_posix() if isinstance(node, PPathNode) else node.value + ), + task.depends_on, # ty: ignore[invalid-argument-type] + ), + **tree_map( + lambda node: ( + node.path.as_posix() if isinstance(node, PPathNode) else node.value + ), + task.produces, # ty: ignore[invalid-argument-type] + ), + } + return {key: value for key, value in kwargs.items() if not key.startswith("_")} + @hookimpl def pytask_execute_task_teardown(session: Session, task: PTask) -> None: diff --git a/src/pytask_stata/serialization.py b/src/pytask_stata/serialization.py new file mode 100644 index 0000000..245156c --- /dev/null +++ b/src/pytask_stata/serialization.py @@ -0,0 +1,80 @@ +"""Serialize task keyword arguments for Stata do-files.""" + +from __future__ import annotations + +import uuid +from collections.abc import Callable +from pathlib import Path +from typing import Any +from typing import TypedDict + +import yaml +from pytask import PTask +from pytask import PTaskWithPath + +__all__ = ["SERIALIZERS", "create_path_to_serialized", "serialize_keyword_arguments"] + +_HIDDEN_FOLDER = ".pytask/pytask-stata" + +SerializerFunc = Callable[..., str] + + +class SerializerEntry(TypedDict): + """Describe a serializer function and its output suffix.""" + + serializer: SerializerFunc + suffix: str + + +def _dump_yaml(kwargs: dict[str, Any]) -> str: + """Serialize task keyword arguments as YAML.""" + return yaml.safe_dump(kwargs, sort_keys=False) + + +SERIALIZERS: dict[str, SerializerEntry] = { + "yaml": {"serializer": _dump_yaml, "suffix": ".yaml"}, + "yml": {"serializer": _dump_yaml, "suffix": ".yml"}, +} + + +def create_path_to_serialized(task: PTask, suffix: str) -> Path: + """Create path to serialized task data.""" + return ( + (task.path.parent if isinstance(task, PTaskWithPath) else Path.cwd()) + .joinpath(_HIDDEN_FOLDER, str(uuid.uuid4())) + .with_suffix(suffix) + ) + + +def serialize_keyword_arguments( + serializer: str | SerializerFunc, + path_to_serialized: Path, + kwargs: dict[str, Any], +) -> None: + """Serialize keyword arguments.""" + if isinstance(serializer, str): + if serializer not in SERIALIZERS: + msg = f"Serializer {serializer!r} is not known." + raise ValueError(msg) + serializer_func = SERIALIZERS[serializer]["serializer"] + elif callable(serializer): + serializer_func = serializer + else: + msg = f"Serializer {serializer!r} is not known." + raise TypeError(msg) + + serialized = serializer_func(_normalize_for_stata(kwargs)) + path_to_serialized.write_text(serialized) + + +def _normalize_for_stata(value: Any) -> Any: + """Normalize values to objects supported by common config serializers.""" + if isinstance(value, Path): + return value.as_posix() + if isinstance(value, dict): + return {str(key): _normalize_for_stata(item) for key, item in value.items()} + if isinstance(value, list): + return [_normalize_for_stata(item) for item in value] + if isinstance(value, tuple): + return tuple(_normalize_for_stata(item) for item in value) + return value diff --git a/src/pytask_stata/shared.py b/src/pytask_stata/shared.py index bf8d1d3..8d09403 100644 --- a/src/pytask_stata/shared.py +++ b/src/pytask_stata/shared.py @@ -3,6 +3,7 @@ from __future__ import annotations import sys +from collections.abc import Callable from collections.abc import Iterable from collections.abc import Sequence from typing import TYPE_CHECKING @@ -44,21 +45,40 @@ STATA_COMMANDS = [] +_DEFAULT_OPTIONS = object() + + def stata( *, script: str | Path, - options: str | Iterable[str] | None = None, -) -> tuple[str | Path | None, str | Iterable[str] | None]: + options: str | Iterable[str] | None | object = _DEFAULT_OPTIONS, + serializer: str | Callable[..., str] | None = None, + suffix: str | None = None, +) -> tuple[ + str | Path | None, + list[str] | None, + str | Callable[..., str] | None, + str | None, +]: """Specify command line options for Stata. Parameters ---------- + script : str | Path + The path to the Stata do-file which is executed. options : str | Iterable[str] | None One or multiple command line options passed to Stata. + serializer : str | Callable[..., str] | None + A function to serialize data for the task. If ``None``, task data is not + serialized. + suffix : str | None + A suffix for the serialized file. If ``None``, infer it from known serializers. """ - options = [] if options is None else list(map(str, _to_list(options))) - return script, options + parsed_options = ( + None if options is _DEFAULT_OPTIONS else list(map(str, _to_list(options))) + ) + return script, parsed_options, serializer, suffix def convert_task_id_to_name_of_log_file(task: PTask) -> str: diff --git a/tests/test_collect.py b/tests/test_collect.py index 4a7ca3b..60e1ff7 100644 --- a/tests/test_collect.py +++ b/tests/test_collect.py @@ -16,13 +16,19 @@ (), {"script": "script.do", "options": "--option"}, does_not_raise(), - ("script.do", ["--option"]), + ("script.do", ["--option"], None, None), ), ( (), {"script": "script.do", "options": [1]}, does_not_raise(), - ("script.do", ["1"]), + ("script.do", ["1"], None, None), + ), + ( + (), + {"script": "script.do"}, + does_not_raise(), + ("script.do", None, None, None), ), ], ) @@ -38,7 +44,30 @@ def test_stata(args, kwargs, expectation, expected): ( Mark("stata", (), {"script": "script.do"}), does_not_raise(), + Mark( + "stata", + (), + { + "script": "script.do", + "options": None, + "serializer": "yaml", + "suffix": ".yaml", + }, + ), + ), + ( Mark("stata", (), {"script": "script.do", "options": []}), + does_not_raise(), + Mark( + "stata", + (), + { + "script": "script.do", + "options": [], + "serializer": None, + "suffix": None, + }, + ), ), ], ) @@ -50,3 +79,41 @@ def test_parse_stata_mark( with expectation: out = _parse_stata_mark(mark) assert out == expected + + +def test_parse_stata_mark_with_yaml(): + mark = Mark("stata", (), {"script": "script.do", "serializer": "yaml"}) + + out = _parse_stata_mark(mark) + + assert out == Mark( + "stata", + (), + { + "script": "script.do", + "options": None, + "serializer": "yaml", + "suffix": ".yaml", + }, + ) + + +def test_parse_stata_mark_raises_for_json(): + mark = Mark("stata", (), {"script": "script.do", "serializer": "json"}) + + with pytest.raises(ValueError, match="Serializer 'json' is not known"): + _parse_stata_mark(mark) + + +@pytest.mark.parametrize( + "kwargs", + [ + {"script": "script.do", "options": [], "serializer": "yaml"}, + {"script": "script.do", "options": [], "suffix": ".yaml"}, + ], +) +def test_parse_stata_mark_raises_for_mixed_interfaces(kwargs): + mark = Mark("stata", (), kwargs) + + with pytest.raises(ValueError, match="interfaces cannot be mixed"): + _parse_stata_mark(mark) diff --git a/tests/test_execute.py b/tests/test_execute.py index 15c5158..6a168e5 100644 --- a/tests/test_execute.py +++ b/tests/test_execute.py @@ -8,6 +8,7 @@ import pytest from pytask import ExitCode from pytask import Mark +from pytask import PythonNode from pytask import Session from pytask import Task from pytask import build @@ -40,6 +41,69 @@ def test_pytask_execute_task_setup_raise_error(stata, platform, expectation): pytask_execute_task_setup(session, task) +def test_pytask_execute_task_setup_serializes_keyword_arguments(tmp_path): + path_to_serialized = tmp_path / "config.txt" + product = tmp_path / "out.dta" + + def serializer(kwargs): + return f"produces={kwargs['produces']}" + + task = Task( + base_name="task_example", + path=tmp_path / "task_example.py", + function=lambda: None, + depends_on={ + "_serialized": PythonNode(value=path_to_serialized), + "_script": PythonNode(value=tmp_path / "script.do"), + "_options": PythonNode(value=[]), + "_log_name": PythonNode(value=""), + "produces": PythonNode(value=product), + }, + markers=[ + Mark( + "stata", + (), + { + "script": "script.do", + "serializer": serializer, + "suffix": ".txt", + }, + ) + ], + ) + session = Session(config={"stata": "stata", "platform": sys.platform}) + + pytask_execute_task_setup(session, task) + + assert path_to_serialized.read_text() == f"produces={product.as_posix()}" + + +@needs_stata +def test_run_do_file_with_yaml_config(runner, tmp_path): + task_source = """ + import pytask + from pathlib import Path + + @pytask.mark.stata(script="script.do") + def task_run_do_file(produces=Path("auto.dta")): + pass + """ + tmp_path.joinpath("task_example.py").write_text(textwrap.dedent(task_source)) + + do_file = """ + args config + yaml read using "`config'", locals replace + sysuse auto, clear + save "`r(yaml_produces)'" + """ + tmp_path.joinpath("script.do").write_text(textwrap.dedent(do_file)) + + result = runner.invoke(cli, [tmp_path.as_posix()]) + + assert result.exit_code == ExitCode.OK + assert tmp_path.joinpath("auto.dta").exists() + + @needs_stata def test_run_do_file(runner, tmp_path): task_source = """ diff --git a/tests/test_stata_mock.py b/tests/test_stata_mock.py new file mode 100644 index 0000000..405828a --- /dev/null +++ b/tests/test_stata_mock.py @@ -0,0 +1,228 @@ +from __future__ import annotations + +import sys +import textwrap + +from stata_mock.cli import main + + +def test_yaml_read_with_locals_exposes_r_macros_and_saves_product( + tmp_path, monkeypatch +): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text( + textwrap.dedent( + """ + produces: out.dta + database: + host: localhost + port: 5432 + debug: true + """ + ) + ) + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", locals replace + save "`r(yaml_produces)'" + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert (tmp_path / "out.dta").exists() + assert "end of mock do-file" in (tmp_path / "mock.log").read_text() + + +def test_yaml_read_supports_custom_prefix_for_r_macros(tmp_path, monkeypatch): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text("produces: prefixed.dta\n") + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", locals prefix(cfg_) replace + save "`r(cfg_produces)'" + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert (tmp_path / "prefixed.dta").exists() + + +def test_yaml_get_returns_attributes_for_nested_mapping(tmp_path, monkeypatch): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text( + textwrap.dedent( + """ + database: + host: localhost + port: 5432 + """ + ) + ) + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", replace + yaml get database:host, quiet + save "`r(host)'" + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert (tmp_path / "localhost.dta").exists() + + +def test_yaml_get_returns_child_attributes_for_parent_key(tmp_path, monkeypatch): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text( + textwrap.dedent( + """ + indicators: + CME_MRY0T4: + label: mortality + unit: deaths + dataflow: CME + """ + ) + ) + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", replace + yaml get indicators:CME_MRY0T4, attributes(label unit) quiet + save "`r(label)'" + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert (tmp_path / "mortality.dta").exists() + + +def test_yaml_validate_required_and_types_success(tmp_path, monkeypatch): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text( + textwrap.dedent( + """ + name: project + version: 1 + debug: false + """ + ) + ) + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", replace + yaml validate, required(name version debug) /// + types(version:numeric debug:boolean) + sysuse auto, clear + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert "end of mock do-file" in (tmp_path / "mock.log").read_text() + + +def test_yaml_validate_required_failure_returns_stata_syntax_error( + tmp_path, monkeypatch +): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text("name: project\n") + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", replace + yaml validate, required(name version) + sysuse auto, clear + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert "r(198)" in (tmp_path / "mock.log").read_text() + + +def test_yaml_read_rejects_unsupported_yaml_syntax(tmp_path, monkeypatch): + monkeypatch.chdir(tmp_path) + config = tmp_path / "config.yaml" + config.write_text("produces: &target out.dta\ncopy: *target\n") + script = tmp_path / "script.do" + script.write_text( + textwrap.dedent( + """ + args config + yaml read using "`config'", locals replace + save "`r(yaml_produces)'" + """ + ) + ) + + monkeypatch.setattr( + sys, + "argv", + ["Stata", "-e", "do", script.as_posix(), config.as_posix(), "-mock"], + ) + + assert main() == 0 + assert "r(198)" in (tmp_path / "mock.log").read_text() + assert not (tmp_path / "out.dta").exists() diff --git a/uv.lock b/uv.lock index c0d910a..ce66a35 100644 --- a/uv.lock +++ b/uv.lock @@ -511,6 +511,7 @@ source = { editable = "." } dependencies = [ { name = "click" }, { name = "pytask" }, + { name = "pyyaml" }, ] [package.dev-dependencies] @@ -531,6 +532,7 @@ typing = [ requires-dist = [ { name = "click", specifier = ">=8.1.8,!=8.2.0" }, { name = "pytask", specifier = ">=0.5.2" }, + { name = "pyyaml" }, ] [package.metadata.requires-dev] @@ -590,6 +592,70 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/ca/31/d4e37e9e550c2b92a9cbc2e4d0b7420a27224968580b5a447f420847c975/pytest_xdist-3.8.0-py3-none-any.whl", hash = "sha256:202ca578cfeb7370784a8c33d6d05bc6e13b4f25b5053c30a152269fd10f0b88", size = 46396, upload-time = "2025-07-01T13:30:56.632Z" }, ] +[[package]] +name = "pyyaml" +version = "6.0.3" +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/05/8e/961c0007c59b8dd7729d542c61a4d537767a59645b82a0b521206e1e25c2/pyyaml-6.0.3.tar.gz", hash = "sha256:d76623373421df22fb4cf8817020cbb7ef15c725b9d5e45f17e189bfc384190f", size = 130960, upload-time = "2025-09-25T21:33:16.546Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/f4/a0/39350dd17dd6d6c6507025c0e53aef67a9293a6d37d3511f23ea510d5800/pyyaml-6.0.3-cp310-cp310-macosx_10_13_x86_64.whl", hash = "sha256:214ed4befebe12df36bcc8bc2b64b396ca31be9304b8f59e25c11cf94a4c033b", size = 184227, upload-time = "2025-09-25T21:31:46.04Z" }, + { url = "https://files.pythonhosted.org/packages/05/14/52d505b5c59ce73244f59c7a50ecf47093ce4765f116cdb98286a71eeca2/pyyaml-6.0.3-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:02ea2dfa234451bbb8772601d7b8e426c2bfa197136796224e50e35a78777956", size = 174019, upload-time = "2025-09-25T21:31:47.706Z" }, + { url = "https://files.pythonhosted.org/packages/43/f7/0e6a5ae5599c838c696adb4e6330a59f463265bfa1e116cfd1fbb0abaaae/pyyaml-6.0.3-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:b30236e45cf30d2b8e7b3e85881719e98507abed1011bf463a8fa23e9c3e98a8", size = 740646, upload-time = "2025-09-25T21:31:49.21Z" }, + { url = "https://files.pythonhosted.org/packages/2f/3a/61b9db1d28f00f8fd0ae760459a5c4bf1b941baf714e207b6eb0657d2578/pyyaml-6.0.3-cp310-cp310-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:66291b10affd76d76f54fad28e22e51719ef9ba22b29e1d7d03d6777a9174198", size = 840793, upload-time = "2025-09-25T21:31:50.735Z" }, + { url = "https://files.pythonhosted.org/packages/7a/1e/7acc4f0e74c4b3d9531e24739e0ab832a5edf40e64fbae1a9c01941cabd7/pyyaml-6.0.3-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:9c7708761fccb9397fe64bbc0395abcae8c4bf7b0eac081e12b809bf47700d0b", size = 770293, upload-time = "2025-09-25T21:31:51.828Z" }, + { url = "https://files.pythonhosted.org/packages/8b/ef/abd085f06853af0cd59fa5f913d61a8eab65d7639ff2a658d18a25d6a89d/pyyaml-6.0.3-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:418cf3f2111bc80e0933b2cd8cd04f286338bb88bdc7bc8e6dd775ebde60b5e0", size = 732872, upload-time = "2025-09-25T21:31:53.282Z" }, + { url = "https://files.pythonhosted.org/packages/1f/15/2bc9c8faf6450a8b3c9fc5448ed869c599c0a74ba2669772b1f3a0040180/pyyaml-6.0.3-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:5e0b74767e5f8c593e8c9b5912019159ed0533c70051e9cce3e8b6aa699fcd69", size = 758828, upload-time = "2025-09-25T21:31:54.807Z" }, + { url = "https://files.pythonhosted.org/packages/a3/00/531e92e88c00f4333ce359e50c19b8d1de9fe8d581b1534e35ccfbc5f393/pyyaml-6.0.3-cp310-cp310-win32.whl", hash = "sha256:28c8d926f98f432f88adc23edf2e6d4921ac26fb084b028c733d01868d19007e", size = 142415, upload-time = "2025-09-25T21:31:55.885Z" }, + { url = "https://files.pythonhosted.org/packages/2a/fa/926c003379b19fca39dd4634818b00dec6c62d87faf628d1394e137354d4/pyyaml-6.0.3-cp310-cp310-win_amd64.whl", hash = "sha256:bdb2c67c6c1390b63c6ff89f210c8fd09d9a1217a465701eac7316313c915e4c", size = 158561, upload-time = "2025-09-25T21:31:57.406Z" }, + { url = "https://files.pythonhosted.org/packages/6d/16/a95b6757765b7b031c9374925bb718d55e0a9ba8a1b6a12d25962ea44347/pyyaml-6.0.3-cp311-cp311-macosx_10_13_x86_64.whl", hash = "sha256:44edc647873928551a01e7a563d7452ccdebee747728c1080d881d68af7b997e", size = 185826, upload-time = "2025-09-25T21:31:58.655Z" }, + { url = "https://files.pythonhosted.org/packages/16/19/13de8e4377ed53079ee996e1ab0a9c33ec2faf808a4647b7b4c0d46dd239/pyyaml-6.0.3-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:652cb6edd41e718550aad172851962662ff2681490a8a711af6a4d288dd96824", size = 175577, upload-time = "2025-09-25T21:32:00.088Z" }, + { url = "https://files.pythonhosted.org/packages/0c/62/d2eb46264d4b157dae1275b573017abec435397aa59cbcdab6fc978a8af4/pyyaml-6.0.3-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:10892704fc220243f5305762e276552a0395f7beb4dbf9b14ec8fd43b57f126c", size = 775556, upload-time = "2025-09-25T21:32:01.31Z" }, + { url = "https://files.pythonhosted.org/packages/10/cb/16c3f2cf3266edd25aaa00d6c4350381c8b012ed6f5276675b9eba8d9ff4/pyyaml-6.0.3-cp311-cp311-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:850774a7879607d3a6f50d36d04f00ee69e7fc816450e5f7e58d7f17f1ae5c00", size = 882114, upload-time = "2025-09-25T21:32:03.376Z" }, + { url = "https://files.pythonhosted.org/packages/71/60/917329f640924b18ff085ab889a11c763e0b573da888e8404ff486657602/pyyaml-6.0.3-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:b8bb0864c5a28024fac8a632c443c87c5aa6f215c0b126c449ae1a150412f31d", size = 806638, upload-time = "2025-09-25T21:32:04.553Z" }, + { url = "https://files.pythonhosted.org/packages/dd/6f/529b0f316a9fd167281a6c3826b5583e6192dba792dd55e3203d3f8e655a/pyyaml-6.0.3-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:1d37d57ad971609cf3c53ba6a7e365e40660e3be0e5175fa9f2365a379d6095a", size = 767463, upload-time = "2025-09-25T21:32:06.152Z" }, + { url = "https://files.pythonhosted.org/packages/f2/6a/b627b4e0c1dd03718543519ffb2f1deea4a1e6d42fbab8021936a4d22589/pyyaml-6.0.3-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:37503bfbfc9d2c40b344d06b2199cf0e96e97957ab1c1b546fd4f87e53e5d3e4", size = 794986, upload-time = "2025-09-25T21:32:07.367Z" }, + { url = "https://files.pythonhosted.org/packages/45/91/47a6e1c42d9ee337c4839208f30d9f09caa9f720ec7582917b264defc875/pyyaml-6.0.3-cp311-cp311-win32.whl", hash = "sha256:8098f252adfa6c80ab48096053f512f2321f0b998f98150cea9bd23d83e1467b", size = 142543, upload-time = "2025-09-25T21:32:08.95Z" }, + { url = "https://files.pythonhosted.org/packages/da/e3/ea007450a105ae919a72393cb06f122f288ef60bba2dc64b26e2646fa315/pyyaml-6.0.3-cp311-cp311-win_amd64.whl", hash = "sha256:9f3bfb4965eb874431221a3ff3fdcddc7e74e3b07799e0e84ca4a0f867d449bf", size = 158763, upload-time = "2025-09-25T21:32:09.96Z" }, + { url = "https://files.pythonhosted.org/packages/d1/33/422b98d2195232ca1826284a76852ad5a86fe23e31b009c9886b2d0fb8b2/pyyaml-6.0.3-cp312-cp312-macosx_10_13_x86_64.whl", hash = "sha256:7f047e29dcae44602496db43be01ad42fc6f1cc0d8cd6c83d342306c32270196", size = 182063, upload-time = "2025-09-25T21:32:11.445Z" }, + { url = "https://files.pythonhosted.org/packages/89/a0/6cf41a19a1f2f3feab0e9c0b74134aa2ce6849093d5517a0c550fe37a648/pyyaml-6.0.3-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:fc09d0aa354569bc501d4e787133afc08552722d3ab34836a80547331bb5d4a0", size = 173973, upload-time = "2025-09-25T21:32:12.492Z" }, + { url = "https://files.pythonhosted.org/packages/ed/23/7a778b6bd0b9a8039df8b1b1d80e2e2ad78aa04171592c8a5c43a56a6af4/pyyaml-6.0.3-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:9149cad251584d5fb4981be1ecde53a1ca46c891a79788c0df828d2f166bda28", size = 775116, upload-time = "2025-09-25T21:32:13.652Z" }, + { url = "https://files.pythonhosted.org/packages/65/30/d7353c338e12baef4ecc1b09e877c1970bd3382789c159b4f89d6a70dc09/pyyaml-6.0.3-cp312-cp312-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:5fdec68f91a0c6739b380c83b951e2c72ac0197ace422360e6d5a959d8d97b2c", size = 844011, upload-time = "2025-09-25T21:32:15.21Z" }, + { url = "https://files.pythonhosted.org/packages/8b/9d/b3589d3877982d4f2329302ef98a8026e7f4443c765c46cfecc8858c6b4b/pyyaml-6.0.3-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:ba1cc08a7ccde2d2ec775841541641e4548226580ab850948cbfda66a1befcdc", size = 807870, upload-time = "2025-09-25T21:32:16.431Z" }, + { url = "https://files.pythonhosted.org/packages/05/c0/b3be26a015601b822b97d9149ff8cb5ead58c66f981e04fedf4e762f4bd4/pyyaml-6.0.3-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:8dc52c23056b9ddd46818a57b78404882310fb473d63f17b07d5c40421e47f8e", size = 761089, upload-time = "2025-09-25T21:32:17.56Z" }, + { url = "https://files.pythonhosted.org/packages/be/8e/98435a21d1d4b46590d5459a22d88128103f8da4c2d4cb8f14f2a96504e1/pyyaml-6.0.3-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:41715c910c881bc081f1e8872880d3c650acf13dfa8214bad49ed4cede7c34ea", size = 790181, upload-time = "2025-09-25T21:32:18.834Z" }, + { url = "https://files.pythonhosted.org/packages/74/93/7baea19427dcfbe1e5a372d81473250b379f04b1bd3c4c5ff825e2327202/pyyaml-6.0.3-cp312-cp312-win32.whl", hash = "sha256:96b533f0e99f6579b3d4d4995707cf36df9100d67e0c8303a0c55b27b5f99bc5", size = 137658, upload-time = "2025-09-25T21:32:20.209Z" }, + { url = "https://files.pythonhosted.org/packages/86/bf/899e81e4cce32febab4fb42bb97dcdf66bc135272882d1987881a4b519e9/pyyaml-6.0.3-cp312-cp312-win_amd64.whl", hash = "sha256:5fcd34e47f6e0b794d17de1b4ff496c00986e1c83f7ab2fb8fcfe9616ff7477b", size = 154003, upload-time = "2025-09-25T21:32:21.167Z" }, + { url = "https://files.pythonhosted.org/packages/1a/08/67bd04656199bbb51dbed1439b7f27601dfb576fb864099c7ef0c3e55531/pyyaml-6.0.3-cp312-cp312-win_arm64.whl", hash = "sha256:64386e5e707d03a7e172c0701abfb7e10f0fb753ee1d773128192742712a98fd", size = 140344, upload-time = "2025-09-25T21:32:22.617Z" }, + { url = "https://files.pythonhosted.org/packages/d1/11/0fd08f8192109f7169db964b5707a2f1e8b745d4e239b784a5a1dd80d1db/pyyaml-6.0.3-cp313-cp313-macosx_10_13_x86_64.whl", hash = "sha256:8da9669d359f02c0b91ccc01cac4a67f16afec0dac22c2ad09f46bee0697eba8", size = 181669, upload-time = "2025-09-25T21:32:23.673Z" }, + { url = "https://files.pythonhosted.org/packages/b1/16/95309993f1d3748cd644e02e38b75d50cbc0d9561d21f390a76242ce073f/pyyaml-6.0.3-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:2283a07e2c21a2aa78d9c4442724ec1eb15f5e42a723b99cb3d822d48f5f7ad1", size = 173252, upload-time = "2025-09-25T21:32:25.149Z" }, + { url = "https://files.pythonhosted.org/packages/50/31/b20f376d3f810b9b2371e72ef5adb33879b25edb7a6d072cb7ca0c486398/pyyaml-6.0.3-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:ee2922902c45ae8ccada2c5b501ab86c36525b883eff4255313a253a3160861c", size = 767081, upload-time = "2025-09-25T21:32:26.575Z" }, + { url = "https://files.pythonhosted.org/packages/49/1e/a55ca81e949270d5d4432fbbd19dfea5321eda7c41a849d443dc92fd1ff7/pyyaml-6.0.3-cp313-cp313-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:a33284e20b78bd4a18c8c2282d549d10bc8408a2a7ff57653c0cf0b9be0afce5", size = 841159, upload-time = "2025-09-25T21:32:27.727Z" }, + { url = "https://files.pythonhosted.org/packages/74/27/e5b8f34d02d9995b80abcef563ea1f8b56d20134d8f4e5e81733b1feceb2/pyyaml-6.0.3-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:0f29edc409a6392443abf94b9cf89ce99889a1dd5376d94316ae5145dfedd5d6", size = 801626, upload-time = "2025-09-25T21:32:28.878Z" }, + { url = "https://files.pythonhosted.org/packages/f9/11/ba845c23988798f40e52ba45f34849aa8a1f2d4af4b798588010792ebad6/pyyaml-6.0.3-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:f7057c9a337546edc7973c0d3ba84ddcdf0daa14533c2065749c9075001090e6", size = 753613, upload-time = "2025-09-25T21:32:30.178Z" }, + { url = "https://files.pythonhosted.org/packages/3d/e0/7966e1a7bfc0a45bf0a7fb6b98ea03fc9b8d84fa7f2229e9659680b69ee3/pyyaml-6.0.3-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:eda16858a3cab07b80edaf74336ece1f986ba330fdb8ee0d6c0d68fe82bc96be", size = 794115, upload-time = "2025-09-25T21:32:31.353Z" }, + { url = "https://files.pythonhosted.org/packages/de/94/980b50a6531b3019e45ddeada0626d45fa85cbe22300844a7983285bed3b/pyyaml-6.0.3-cp313-cp313-win32.whl", hash = "sha256:d0eae10f8159e8fdad514efdc92d74fd8d682c933a6dd088030f3834bc8e6b26", size = 137427, upload-time = "2025-09-25T21:32:32.58Z" }, + { url = "https://files.pythonhosted.org/packages/97/c9/39d5b874e8b28845e4ec2202b5da735d0199dbe5b8fb85f91398814a9a46/pyyaml-6.0.3-cp313-cp313-win_amd64.whl", hash = "sha256:79005a0d97d5ddabfeeea4cf676af11e647e41d81c9a7722a193022accdb6b7c", size = 154090, upload-time = "2025-09-25T21:32:33.659Z" }, + { url = "https://files.pythonhosted.org/packages/73/e8/2bdf3ca2090f68bb3d75b44da7bbc71843b19c9f2b9cb9b0f4ab7a5a4329/pyyaml-6.0.3-cp313-cp313-win_arm64.whl", hash = "sha256:5498cd1645aa724a7c71c8f378eb29ebe23da2fc0d7a08071d89469bf1d2defb", size = 140246, upload-time = "2025-09-25T21:32:34.663Z" }, + { url = "https://files.pythonhosted.org/packages/9d/8c/f4bd7f6465179953d3ac9bc44ac1a8a3e6122cf8ada906b4f96c60172d43/pyyaml-6.0.3-cp314-cp314-macosx_10_13_x86_64.whl", hash = "sha256:8d1fab6bb153a416f9aeb4b8763bc0f22a5586065f86f7664fc23339fc1c1fac", size = 181814, upload-time = "2025-09-25T21:32:35.712Z" }, + { url = "https://files.pythonhosted.org/packages/bd/9c/4d95bb87eb2063d20db7b60faa3840c1b18025517ae857371c4dd55a6b3a/pyyaml-6.0.3-cp314-cp314-macosx_11_0_arm64.whl", hash = "sha256:34d5fcd24b8445fadc33f9cf348c1047101756fd760b4dacb5c3e99755703310", size = 173809, upload-time = "2025-09-25T21:32:36.789Z" }, + { url = "https://files.pythonhosted.org/packages/92/b5/47e807c2623074914e29dabd16cbbdd4bf5e9b2db9f8090fa64411fc5382/pyyaml-6.0.3-cp314-cp314-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:501a031947e3a9025ed4405a168e6ef5ae3126c59f90ce0cd6f2bfc477be31b7", size = 766454, upload-time = "2025-09-25T21:32:37.966Z" }, + { url = "https://files.pythonhosted.org/packages/02/9e/e5e9b168be58564121efb3de6859c452fccde0ab093d8438905899a3a483/pyyaml-6.0.3-cp314-cp314-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:b3bc83488de33889877a0f2543ade9f70c67d66d9ebb4ac959502e12de895788", size = 836355, upload-time = "2025-09-25T21:32:39.178Z" }, + { url = "https://files.pythonhosted.org/packages/88/f9/16491d7ed2a919954993e48aa941b200f38040928474c9e85ea9e64222c3/pyyaml-6.0.3-cp314-cp314-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:c458b6d084f9b935061bc36216e8a69a7e293a2f1e68bf956dcd9e6cbcd143f5", size = 794175, upload-time = "2025-09-25T21:32:40.865Z" }, + { url = "https://files.pythonhosted.org/packages/dd/3f/5989debef34dc6397317802b527dbbafb2b4760878a53d4166579111411e/pyyaml-6.0.3-cp314-cp314-musllinux_1_2_aarch64.whl", hash = "sha256:7c6610def4f163542a622a73fb39f534f8c101d690126992300bf3207eab9764", size = 755228, upload-time = "2025-09-25T21:32:42.084Z" }, + { url = "https://files.pythonhosted.org/packages/d7/ce/af88a49043cd2e265be63d083fc75b27b6ed062f5f9fd6cdc223ad62f03e/pyyaml-6.0.3-cp314-cp314-musllinux_1_2_x86_64.whl", hash = "sha256:5190d403f121660ce8d1d2c1bb2ef1bd05b5f68533fc5c2ea899bd15f4399b35", size = 789194, upload-time = "2025-09-25T21:32:43.362Z" }, + { url = "https://files.pythonhosted.org/packages/23/20/bb6982b26a40bb43951265ba29d4c246ef0ff59c9fdcdf0ed04e0687de4d/pyyaml-6.0.3-cp314-cp314-win_amd64.whl", hash = "sha256:4a2e8cebe2ff6ab7d1050ecd59c25d4c8bd7e6f400f5f82b96557ac0abafd0ac", size = 156429, upload-time = "2025-09-25T21:32:57.844Z" }, + { url = "https://files.pythonhosted.org/packages/f4/f4/a4541072bb9422c8a883ab55255f918fa378ecf083f5b85e87fc2b4eda1b/pyyaml-6.0.3-cp314-cp314-win_arm64.whl", hash = "sha256:93dda82c9c22deb0a405ea4dc5f2d0cda384168e466364dec6255b293923b2f3", size = 143912, upload-time = "2025-09-25T21:32:59.247Z" }, + { url = "https://files.pythonhosted.org/packages/7c/f9/07dd09ae774e4616edf6cda684ee78f97777bdd15847253637a6f052a62f/pyyaml-6.0.3-cp314-cp314t-macosx_10_13_x86_64.whl", hash = "sha256:02893d100e99e03eda1c8fd5c441d8c60103fd175728e23e431db1b589cf5ab3", size = 189108, upload-time = "2025-09-25T21:32:44.377Z" }, + { url = "https://files.pythonhosted.org/packages/4e/78/8d08c9fb7ce09ad8c38ad533c1191cf27f7ae1effe5bb9400a46d9437fcf/pyyaml-6.0.3-cp314-cp314t-macosx_11_0_arm64.whl", hash = "sha256:c1ff362665ae507275af2853520967820d9124984e0f7466736aea23d8611fba", size = 183641, upload-time = "2025-09-25T21:32:45.407Z" }, + { url = "https://files.pythonhosted.org/packages/7b/5b/3babb19104a46945cf816d047db2788bcaf8c94527a805610b0289a01c6b/pyyaml-6.0.3-cp314-cp314t-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:6adc77889b628398debc7b65c073bcb99c4a0237b248cacaf3fe8a557563ef6c", size = 831901, upload-time = "2025-09-25T21:32:48.83Z" }, + { url = "https://files.pythonhosted.org/packages/8b/cc/dff0684d8dc44da4d22a13f35f073d558c268780ce3c6ba1b87055bb0b87/pyyaml-6.0.3-cp314-cp314t-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:a80cb027f6b349846a3bf6d73b5e95e782175e52f22108cfa17876aaeff93702", size = 861132, upload-time = "2025-09-25T21:32:50.149Z" }, + { url = "https://files.pythonhosted.org/packages/b1/5e/f77dc6b9036943e285ba76b49e118d9ea929885becb0a29ba8a7c75e29fe/pyyaml-6.0.3-cp314-cp314t-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:00c4bdeba853cc34e7dd471f16b4114f4162dc03e6b7afcc2128711f0eca823c", size = 839261, upload-time = "2025-09-25T21:32:51.808Z" }, + { url = "https://files.pythonhosted.org/packages/ce/88/a9db1376aa2a228197c58b37302f284b5617f56a5d959fd1763fb1675ce6/pyyaml-6.0.3-cp314-cp314t-musllinux_1_2_aarch64.whl", hash = "sha256:66e1674c3ef6f541c35191caae2d429b967b99e02040f5ba928632d9a7f0f065", size = 805272, upload-time = "2025-09-25T21:32:52.941Z" }, + { url = "https://files.pythonhosted.org/packages/da/92/1446574745d74df0c92e6aa4a7b0b3130706a4142b2d1a5869f2eaa423c6/pyyaml-6.0.3-cp314-cp314t-musllinux_1_2_x86_64.whl", hash = "sha256:16249ee61e95f858e83976573de0f5b2893b3677ba71c9dd36b9cf8be9ac6d65", size = 829923, upload-time = "2025-09-25T21:32:54.537Z" }, + { url = "https://files.pythonhosted.org/packages/f0/7a/1c7270340330e575b92f397352af856a8c06f230aa3e76f86b39d01b416a/pyyaml-6.0.3-cp314-cp314t-win_amd64.whl", hash = "sha256:4ad1906908f2f5ae4e5a8ddfce73c320c2a1429ec52eafd27138b7f1cbe341c9", size = 174062, upload-time = "2025-09-25T21:32:55.767Z" }, + { url = "https://files.pythonhosted.org/packages/f1/12/de94a39c2ef588c7e6455cfbe7343d3b2dc9d6b6b2f40c4c6565744c873d/pyyaml-6.0.3-cp314-cp314t-win_arm64.whl", hash = "sha256:ebc55a14a21cb14062aa4162f906cd962b28e2e9ea38f9b4391244cd8de4ae0b", size = 149341, upload-time = "2025-09-25T21:32:56.828Z" }, +] + [[package]] name = "rich" version = "14.1.0"