# -*- coding: utf-8 -*-
from django import forms

from captcha.fields import CaptchaField

from tinymce.widgets import TinyMCE

from .models import Entry, Category


class EntryForm(forms.ModelForm):
    """Форма редактирования нового сообщения."""
    captcha = CaptchaField(label=u'Код с картинки', error_messages={'required': u'Введите код с картинки.'})

    class Meta:
        model = Entry
        fields = ['name', 'email', 'message']

    def __init__(self, *args, **kwargs):
        super(EntryForm, self).__init__(*args, **kwargs)
        self.fields['name'].error_messages['required'] = u'Укажите имя.'
        self.fields['message'].error_messages['required'] = u'Введите текст сообщения.'

    def clean_name(self):
        return self.cleaned_data.get('name', '').strip()

    def clean_email(self):
        return self.cleaned_data.get('email', '').strip()

    def clean_message(self):
        return self.cleaned_data.get('message', '').strip()


class CategoryAdminForm(forms.ModelForm):
    class Meta:
        model = Category
        widgets = {
            'description': TinyMCE,
        }