django-csp-advanced/csp_advanced/utils.py
2017-07-15 02:45:23 -04:00

41 lines
1 KiB
Python

from django.utils import six
def is_callable_csp_dict(data):
if callable(data):
return True
if not isinstance(data, dict):
return False
return any(callable(value) for value in six.itervalues(data))
def call_csp_dict(data, request, response):
if callable(data):
return data(request, response)
result = {}
for key, value in six.iteritems(data):
if callable(value):
result[key] = value(request, response)
else:
result[key] = value
return result
def merge_csp_dict(template, override):
result = template.copy()
for key, value in six.iteritems(override):
if key not in result:
result[key] = value
continue
orig = result[key]
if isinstance(orig, list):
result[key] = orig + list(value)
elif isinstance(orig, set):
result[key] = orig.union(value)
elif isinstance(orig, tuple):
result[key] = orig + tuple(value)
else:
result[key] = value
return result