kodo/simditor/fields.py

# -- coding: utf-8 --
"""simditor fields."""
from django import forms
from django.db import models

from .widgets import SimditorWidget


class RichTextFormField(forms.fields.CharField):
    """RichTextFormField."""

    def __init__(self, *args, **kwargs):
        kwargs.update(
            {
                'widget': SimditorWidget()
            }
        )
        super(RichTextFormField, self).__init__(*args, **kwargs)


class RichTextField(models.TextField):
    """RichTextField."""

    def __init__(self, *args, **kwargs):
        super(RichTextField, self).__init__(*args, **kwargs)

    def formfield(self, **kwargs):
        defaults = {
            'form_class': self._get_form_class()
        }
        defaults.update(kwargs)
        return super(RichTextField, self).formfield(**defaults)

    @staticmethod
    def _get_form_class():
        return RichTextFormField