22 lines
504 B
Python
22 lines
504 B
Python
|
from dbt.dataclass_schema import StrEnum
|
||
|
|
import json
|
||
|
|
|
||
|
|
from typing import Union, Dict, Any
|
||
|
|
|
||
|
|
|
||
|
|
class ModelHookType(StrEnum):
|
||
|
|
PreHook = 'pre-hook'
|
||
|
|
PostHook = 'post-hook'
|
||
|
|
|
||
|
|
|
||
|
|
def get_hook_dict(source: Union[str, Dict[str, Any]]) -> Dict[str, Any]:
|
||
|
|
"""From a source string-or-dict, get a dictionary that can be passed to
|
||
|
|
Hook.from_dict
|
||
|
|
"""
|
||
|
|
if isinstance(source, dict):
|
||
|
|
return source
|
||
|
|
try:
|
||
|
|
return json.loads(source)
|
||
|
|
except ValueError:
|
||
|
|
return {'sql': source}
|