54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
from django.apps import apps
|
|
from django.core import checks
|
|
from django.core.exceptions import ValidationError
|
|
from django.utils.functional import SimpleLazyObject
|
|
from django.utils.text import format_lazy
|
|
|
|
|
|
def prefix_validation_error(error, prefix, code, params):
|
|
"""
|
|
Prefix a validation error message while maintaining the existing
|
|
validation data structure.
|
|
"""
|
|
if error.error_list == [error]:
|
|
error_params = error.params or {}
|
|
return ValidationError(
|
|
# We can't simply concatenate messages since they might require
|
|
# their associated parameters to be expressed correctly which
|
|
# is not something `format_lazy` does. For example, proxied
|
|
# ngettext calls require a count parameter and are converted
|
|
# to an empty string if they are missing it.
|
|
message=format_lazy(
|
|
"{} {}",
|
|
SimpleLazyObject(lambda: prefix % params),
|
|
SimpleLazyObject(lambda: error.message % error_params),
|
|
),
|
|
code=code,
|
|
params={**error_params, **params},
|
|
)
|
|
return ValidationError(
|
|
[prefix_validation_error(e, prefix, code, params) for e in error.error_list]
|
|
)
|
|
|
|
|
|
class CheckPostgresInstalledMixin:
|
|
def _check_postgres_installed(self, *args):
|
|
# When subclassed by Index or BaseConstraint subclasses, args is
|
|
# (model, connection).
|
|
obj = args[0] if args else self
|
|
if not apps.is_installed("django.contrib.postgres"):
|
|
return [
|
|
checks.Error(
|
|
"'django.contrib.postgres' must be in INSTALLED_APPS in "
|
|
"order to use %s." % self.__class__.__name__,
|
|
obj=obj,
|
|
id="postgres.E005",
|
|
)
|
|
]
|
|
return []
|
|
|
|
def check(self, *args, **kwargs):
|
|
errors = super().check(*args, **kwargs)
|
|
errors.extend(self._check_postgres_installed(*args))
|
|
return errors
|