def render_raw_permissions(query, parameters):
def _flatten_dict(p, parent_key=''):
new_p = {}
for k, v in deepcopy(p).items():
new_key = f'{parent_key}_{k}' if parent_key else k
new_p[new_key] = v
if isinstance(v, list):
v = {idx: elt for idx, elt in enumerate(v)}
if isinstance(v, dict):
new_p.update(_flatten_dict(v, new_key))
elif isinstance(v, str):
new_p.update({new_key: f'"{v}"'})
else:
new_p.update({new_key: v})
return new_p
if parameters is None:
return query
# Flag params to keep type if not complex (no quotes or condition)
for pattern in RE_JINJA_ALONE_IN_STRING:
query = re.sub(pattern, RE_SET_KEEP_TYPE, query)
p_keep_type = re.findall(RE_GET_KEEP_TYPE, query)
for key in p_keep_type:
query = query.replace(key, slugify(key, separator='_'))
if len(p_keep_type):
# Add a version of parameters flatten + with quotes for string
p_keep_type = _flatten_dict(parameters, parent_key='__keep_type_')
parameters.update(p_keep_type)
return Template(query).render(parameters)
def apply_query_parameters(query: str, parameters: dict) -> str:
"""
Apply parameters to query
Interpolate the query, which is a Jinja templates, with the provided parameters.
"""
def _flatten_dict(p, parent_key=''):
new_p = {}
for k, v in deepcopy(p).items():
new_key = f'{parent_key}_{k}' if parent_key else k
new_p[new_key] = v
if isinstance(v, list):
v = {idx: elt for idx, elt in enumerate(v)}
if isinstance(v, dict):
new_p.update(_flatten_dict(v, new_key))
elif isinstance(v, str):
new_p.update({new_key: f'"{v}"'})
else:
new_p.update({new_key: v})
return new_p
if parameters is None:
return query
# Flag params to keep type if not complex (no quotes or condition)
for pattern in RE_JINJA_ALONE_IN_STRING:
query = re.sub(pattern, RE_SET_KEEP_TYPE, query)
p_keep_type = re.findall(RE_GET_KEEP_TYPE, query)
for key in p_keep_type:
query = query.replace(key, slugify(key, separator='_'))
if len(p_keep_type):
# Add a version of parameters flatten + with quotes for string
p_keep_type = _flatten_dict(parameters, parent_key='__keep_type_')
parameters.update(p_keep_type)
logging.getLogger(__name__).debug(
f'Render query: {query} with parameters {parameters}')
return Template(query).render(parameters)