django-localized-fields/localized_fields/fields/bleach_field.py

from django.conf import settings

from .field import LocalizedField


class LocalizedBleachField(LocalizedField):
    """Custom version of :see:BleachField that is actually a
    :see:LocalizedField."""

    def pre_save(self, instance, add: bool):
        """Ran just before the model is saved, allows us to built the slug.

        Arguments:
            instance:
                The model that is being saved.

            add:
                Indicates whether this is a new entry
                to the database or an update.
        """

        # the bleach library vendors dependencies and the html5lib
        # dependency is incompatible with python 3.9, until that's
        # fixed, you cannot use LocalizedBleachField with python 3.9
        # sympton:
        #   ImportError: cannot import name 'Mapping' from 'collections'
        try:
            import bleach

            from django_bleach.utils import get_bleach_default_options
        except ImportError:
            raise UserWarning(
                "LocalizedBleachField is not compatible with Python 3.9 yet."
            )

        localized_value = getattr(instance, self.attname)
        if not localized_value:
            return None

        for lang_code, _ in settings.LANGUAGES:
            value = localized_value.get(lang_code)
            if not value:
                continue

            localized_value.set(
                lang_code, bleach.clean(value, **get_bleach_default_options())
            )

        return localized_value