from django import forms

from django.db import models

class RichTextFormField(forms.fields.CharField):

def __init__(self, *args, **kwargs):

self.cleanse = kwargs.pop("cleanse", None)

super().__init__(*args, **kwargs)

css_class = self.widget.attrs.get("class", "")

css_class += " item-richtext"

self.widget.attrs["class"] = css_class

def clean(self, value):

value = super().clean(value)

if self.cleanse:

value = self.cleanse(value)

return value

class RichTextField(models.TextField):

"""

Drop-in replacement for Django's ``models.TextField`` which allows editing

rich text instead of plain text in the item editor.

"""

def __init__(self, *args, **kwargs):

self.cleanse = kwargs.pop("cleanse", None)

super().__init__(*args, **kwargs)

def formfield(self, form_class=RichTextFormField, **kwargs):

return super().formfield(form_class=form_class, cleanse=self.cleanse, **kwargs)