"""
cookiecutter.replay.
-------------------
"""
from __future__ import annotations
import json
import os
from typing import TYPE_CHECKING, Any
from cookiecutter.utils import make_sure_path_exists
if TYPE_CHECKING:
from pathlib import Path
[docs]
def get_file_name(replay_dir: Path | str, template_name: str) -> str:
"""Get the name of file."""
suffix = '.json' if not template_name.endswith('.json') else ''
file_name = f'{template_name}{suffix}'
return os.path.join(replay_dir, file_name)
[docs]
def dump(replay_dir: Path | str, template_name: str, context: dict[str, Any]) -> None:
"""Write json data to file."""
make_sure_path_exists(replay_dir)
if 'cookiecutter' not in context:
raise ValueError('Context is required to contain a cookiecutter key')
replay_file = get_file_name(replay_dir, template_name)
with open(replay_file, 'w', encoding="utf-8") as outfile:
json.dump(context, outfile, indent=2)
[docs]
def load(replay_dir: Path | str, template_name: str) -> dict[str, Any]:
"""Read json data from file."""
replay_file = get_file_name(replay_dir, template_name)
with open(replay_file, encoding="utf-8") as infile:
context: dict[str, Any] = json.load(infile)
if 'cookiecutter' not in context:
raise ValueError('Context is required to contain a cookiecutter key')
return context