lib.itmens/common/forms.py

from django import forms
import django.contrib.postgres.forms as postgres
from django.utils import formats
from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _
import json


class KeyValueInput(forms.Widget):
    template_name = 'widgets/key_value.html'

    def get_context(self, name, value, attrs):
        """ called when rendering """
        context = {}
        context['widget'] = {
            'name': name,
            'is_hidden': self.is_hidden,
            'required': self.is_required,
            'value': self.format_value(value),
            'attrs': self.build_attrs(self.attrs, attrs),
            'template_name': self.template_name,
            'keyvalue_pairs': {},
        }
        if context['widget']['value']:
            key_value_pairs = json.loads(context['widget']['value'])
            # for kv in key_value_pairs:
            context['widget']['keyvalue_pairs'] = key_value_pairs
        return context

    class Media:
        js = ('js/key_value_input.js',)


class JSONField(postgres.JSONField):
    widget = KeyValueInput
    def to_python(self, value):
        if not value:
            return None
        json = {}
        if isinstance(value, dict):
            json = value
        else:
            pairs = eval(value)
            if isinstance(pairs, dict):
                json = pairs
            else:
                # list or tuple
                for pair in pairs:
                    json = {**json, **pair}
        return super().to_python(json)


class RadioBooleanField(forms.ChoiceField):
    widget = forms.RadioSelect

    def to_python(self, value):
        """Return a Python boolean object."""
        # Explicitly check for the string 'False', which is what a hidden field
        # will submit for False. Also check for '0', since this is what
        # RadioSelect will provide. Because bool("True") == bool('1') == True,
        # we don't need to handle that explicitly.
        if isinstance(value, str) and value.lower() in ('false', '0'):
            value = False
        else:
            value = bool(value)
        return value


class RatingValidator:
    """ empty value is not validated """
    def __call__(self, value):
        if not isinstance(value, int):
            raise ValidationError(
                _('%(value)s is not an integer'),
                params={'value': value},
            )
        if not str(value) in [str(i) for i in range(1, 11)]:
            raise ValidationError(
                _('%(value)s is not an integer in range 1-10'),
                params={'value': value},
            )


class PreviewImageInput(forms.FileInput):
    template_name = 'widgets/image.html'
    def format_value(self, value):
        """
        Return the file object if it has a defined url attribute.
        """
        if self.is_initial(value):
            if value.url:
                return value.url
            else:
                return

    def is_initial(self, value):
        """
        Return whether value is considered to be initial value.
        """
        return bool(value and getattr(value, 'url', False))


class TagInput(forms.TextInput):
    """
    Dump tag queryset into tag list
    """
    template_name = 'widgets/tag.html'
    def format_value(self, value):
        if value == '' or value is None or len(value) == 0:
            return ''
        tag_list = []
        try:
            tag_list = [t['content'] for t in value]
        except TypeError:
            tag_list = [t.content for t in value]
        # return ','.join(tag_list)
        return tag_list

    class Media:
        css = {
            'all': ('lib/css/tag-input.css',)
        }
        js = ('lib/js/tag-input.js',)
        

class TagField(forms.CharField):
    """
    Split comma connected string into tag list
    """
    widget = TagInput
    def to_python(self, value):
        value = super().to_python(value)
        if not value:
            return
        return [t.strip() for t in value.split(',')]


class MultiSelect(forms.SelectMultiple):
    template_name = 'widgets/multi_select.html'

    class Media:
        css = {
            'all': ('lib/css/multiple-select.min.css',)
        }
        js = ('lib/js/multiple-select.min.js',)


class HstoreInput(forms.Widget):
    template_name = 'widgets/hstore.html'

    def format_value(self, value):
        """
        Return a value as it should appear when rendered in a template.
        """
        if value == '' or value is None:
            return None
        if self.is_localized:
            return formats.localize_input(value)
        return value

    class Media:
        js = ('js/key_value_input.js',)


class HstoreField(forms.CharField):
    widget = HstoreInput
    def to_python(self, value):
        if not value:
            return None
        # already in python types
        if isinstance(value, list):
            return value
        pairs = eval(value)
        if len(pairs) == 1:
            pairs = (pairs,)
        return pairs


#############################
# Form
#############################

class MarkForm(forms.ModelForm):
    IS_PRIVATE_CHOICES = [
        (True, _("仅关注者")),
        (False, _("公开")),
    ]
    
    id = forms.IntegerField(required=False, widget=forms.HiddenInput())
    share_to_mastodon = forms.BooleanField(
        label=_("分享到长毛象"), initial=True, required=False)
    rating = forms.IntegerField(
        validators=[RatingValidator()], widget=forms.HiddenInput(), required=False)
    is_private = RadioBooleanField(
        label=_("可见性"),
        initial=True,
        choices=IS_PRIVATE_CHOICES
    )
    tags = TagField(
        required=False,
        widget=TagInput(attrs={'placeholder': _("回车增加标签")}),
        label=_("标签")
    )
    text = forms.CharField(
        required=False,
        widget=forms.Textarea(
            attrs={
                "placeholder": _("最多只能写360字哦~"),
                "maxlength": 360
            }
        ),

        label=_("短评"),
    )

class ReviewForm(forms.ModelForm):
    IS_PRIVATE_CHOICES = [
        (True, _("仅关注者")),
        (False, _("公开")),
    ]
    share_to_mastodon = forms.BooleanField(
        label=_("分享到长毛象"), initial=True, required=False)
    id = forms.IntegerField(required=False, widget=forms.HiddenInput())
    is_private = RadioBooleanField(
        label=_("可见性"),
        initial=True,
        choices=IS_PRIVATE_CHOICES
    )