24 lines
760 B
Python
24 lines
760 B
Python
|
from typing import Dict, Any
|
||
|
|
|
||
|
|
from dbt.clients import yaml_helper
|
||
|
|
from dbt.exceptions import raise_compiler_error, ValidationException
|
||
|
|
from dbt.logger import GLOBAL_LOGGER as logger
|
||
|
|
|
||
|
|
|
||
|
|
def parse_cli_vars(var_string: str) -> Dict[str, Any]:
|
||
|
|
try:
|
||
|
|
cli_vars = yaml_helper.load_yaml_text(var_string)
|
||
|
|
var_type = type(cli_vars)
|
||
|
|
if var_type is dict:
|
||
|
|
return cli_vars
|
||
|
|
else:
|
||
|
|
type_name = var_type.__name__
|
||
|
|
raise_compiler_error(
|
||
|
|
"The --vars argument must be a YAML dictionary, but was "
|
||
|
|
"of type '{}'".format(type_name))
|
||
|
|
except ValidationException:
|
||
|
|
logger.error(
|
||
|
|
"The YAML provided in the --vars argument is not valid.\n"
|
||
|
|
)
|
||
|
|
raise
|