InterSIS/django-rest-encrypted-lookup

"""
Settings for rest_framework_encrypted_lookup
"""

from django.conf import settings

EXTRA_REQUIRED_SETTINGS = ('secret_key',)

# TODO: Refactor name to DEFAULT_ENCRYPTED_LOOKUP_SETTINGS on next major version upgrade
default_encrypted_lookup_settings = {  # pylint: disable=invalid-name
    'lookup_field_name': 'pk',
}

try:
    encrypted_lookup_settings = getattr(settings, 'ENCRYPTED_LOOKUP')  # pylint: disable=invalid-name
except AttributeError:
    raise AttributeError(
        "Django.conf.settings must include an ENCRYPTED_LOOKUP settings dictionary."
    )
else:
    for required_key in EXTRA_REQUIRED_SETTINGS:
        if required_key not in encrypted_lookup_settings:
            raise AttributeError(
                "ENCRYPTED_LOOKUP settings dictionary must include a '%s' key." % required_key
            )

    for key in encrypted_lookup_settings:
        if key not in default_encrypted_lookup_settings and key not in EXTRA_REQUIRED_SETTINGS:
            raise AttributeError("Unrecognized ENCRYPTED_LOOKUP setting key: '%s'" % key)

    for key, value in default_encrypted_lookup_settings.items():
        if key not in encrypted_lookup_settings:
            encrypted_lookup_settings[key] = value