def deploy_json_api_output(
intervention: Intervention,
area_result: pydantic.BaseModel,
output_dir: pathlib.Path,
filename_override=None,
):
if not output_dir.exists():
output_dir.mkdir(parents=True, exist_ok=True)
filename = filename_override or (area_result.output_key(intervention) +
".json")
output_path = output_dir / filename
output_path.write_text(area_result.json())
return area_result
def deploy_csv_api_output(
intervention: Intervention,
api_output: pydantic.BaseModel,
output_dir: pathlib.Path,
filename_override=None,
):
if not hasattr(api_output, "__root__"):
raise AssertionError("Missing root data")
if not output_dir.exists():
output_dir.mkdir(parents=True, exist_ok=True)
filename = filename_override or (api_output.output_key(intervention) +
".csv")
output_path = output_dir / filename
rows = dataset_deployer.remove_root_wrapper(api_output.dict())
dataset_deployer.write_nested_csv(rows, output_path)
