Python remove_root_wrapper示例

示例#1

文件： api_v2_pipeline.py 项目： epius/covid-data-model

def deploy_csv_api_output(api_output: pydantic.BaseModel,
                          output_path: pathlib.Path,
                          columns: List[str]) -> None:
    if not hasattr(api_output, "__root__"):
        raise AssertionError("Missing root data")

    data = _model_to_dict(api_output.__dict__)
    rows = dataset_deployer.remove_root_wrapper(data)
    dataset_deployer.write_nested_csv(rows, output_path, header=columns)

示例#2

文件： api_v2_pipeline.py 项目： andresmmujica/covid-data-model

def deploy_csv_api_output(
    api_output: pydantic.BaseModel,
    output_path: pathlib.Path,
    keys_to_skip: Optional[List[str]] = None,
) -> None:
    if not hasattr(api_output, "__root__"):
        raise AssertionError("Missing root data")

    rows = dataset_deployer.remove_root_wrapper(api_output.dict())
    dataset_deployer.write_nested_csv(rows,
                                      output_path,
                                      keys_to_skip=keys_to_skip)

示例#3

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)

示例#4

文件： test_can_predictions.py 项目： andresmmujica/covid-data-model

def test_remove_root():
    value = AggregateRegionSummaryWithTimeseries(__root__=[]).dict()
    assert value == {"__root__": []}
    result = dataset_deployer.remove_root_wrapper(value)
    assert result == []

示例#5

文件： test_can_predictions.py 项目： fc327/covid-data-model

def test_remove_root():
    value = CovidActNowBulkTimeseries(__root__=[]).dict()
    assert value == {"__root__": []}
    result = dataset_deployer.remove_root_wrapper(value)
    assert result == []