274 lines
7.6 KiB
Python
274 lines
7.6 KiB
Python
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):
|
|
"""
|
|
Input widget for Json field
|
|
"""
|
|
template_name = 'widgets/hstore.html'
|
|
|
|
def get_context(self, name, value, attrs):
|
|
context = super().get_context(name, value, attrs)
|
|
data = None
|
|
if context['widget']['value'] is not None:
|
|
data = json.loads(context['widget']['value'])
|
|
context['widget']['value'] = [ {p[0]: p[1]} for p in data.items()] if data else []
|
|
return context
|
|
|
|
class Media:
|
|
js = ('js/key_value_input.js',)
|
|
|
|
|
|
class HstoreInput(forms.Widget):
|
|
"""
|
|
Input widget for Hstore field
|
|
"""
|
|
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)
|
|
# do not return str
|
|
return value
|
|
|
|
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(0, 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': ('https://cdn.jsdelivr.net/npm/multiple-select@1.5.2/dist/multiple-select.min.css',)
|
|
}
|
|
js = ('https://cdn.jsdelivr.net/npm/multiple-select@1.5.2/dist/multiple-select.min.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
|
|
|
|
|
|
class DurationInput(forms.TextInput):
|
|
"""
|
|
HH:mm:ss input widget
|
|
"""
|
|
input_type = "time"
|
|
|
|
def get_context(self, name, value, attrs):
|
|
context = super().get_context(name, value, attrs)
|
|
# context['widget']['type'] = self.input_type
|
|
context['widget']['attrs']['step'] = "1"
|
|
return context
|
|
|
|
def format_value(self, value):
|
|
"""
|
|
Given `value` is an integer in ms
|
|
"""
|
|
ms = value
|
|
if not ms:
|
|
return super().format_value(None)
|
|
x = ms // 1000
|
|
seconds = x % 60
|
|
x //= 60
|
|
if x == 0:
|
|
return super().format_value(f"00:00:{seconds:0>2}")
|
|
minutes = x % 60
|
|
x //= 60
|
|
if x == 0:
|
|
return super().format_value(f"00:{minutes:0>2}:{seconds:0>2}")
|
|
hours = x % 24
|
|
return super().format_value(f"{hours:0>2}:{minutes:0>2}:{seconds:0>2}")
|
|
|
|
|
|
class DurationField(forms.TimeField):
|
|
widget = DurationInput
|
|
def to_python(self, value):
|
|
|
|
# empty value
|
|
if value is None or value == '':
|
|
return
|
|
|
|
# if value is integer in ms
|
|
if isinstance(value, int):
|
|
return value
|
|
|
|
# if value is string in time format
|
|
h, m, s = value.split(':')
|
|
return (int(h) * 3600 + int(m) * 60 + int(s)) * 1000
|
|
|
|
|
|
#############################
|
|
# 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
|
|
)
|