149 lines
5.0 KiB
Python
149 lines
5.0 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.db import models
|
|
from django.utils.html import escape
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from .classes import Filter
|
|
from .models import UserLocaleProfile
|
|
from .utils import return_attrib
|
|
from .widgets import DetailSelectMultiple, DisableableSelectWidget, PlainWidget
|
|
|
|
|
|
class DetailForm(forms.ModelForm):
|
|
def __init__(self, extra_fields=None, *args, **kwargs):
|
|
super(DetailForm, self).__init__(*args, **kwargs)
|
|
if extra_fields:
|
|
for extra_field in extra_fields:
|
|
result = return_attrib(self.instance, extra_field['field'])
|
|
label = 'label' in extra_field and extra_field['label'] or None
|
|
# TODO: Add others result types <=> Field types
|
|
if isinstance(result, models.query.QuerySet):
|
|
self.fields[extra_field['field']] = \
|
|
forms.ModelMultipleChoiceField(
|
|
queryset=result, label=label)
|
|
else:
|
|
self.fields[extra_field['field']] = forms.CharField(
|
|
label=extra_field['label'],
|
|
initial=escape(
|
|
return_attrib(
|
|
self.instance,
|
|
extra_field['field'], None
|
|
)
|
|
),
|
|
widget=PlainWidget)
|
|
|
|
for field_name, field in self.fields.items():
|
|
if isinstance(field.widget, forms.widgets.SelectMultiple):
|
|
self.fields[field_name].widget = DetailSelectMultiple(
|
|
choices=field.widget.choices,
|
|
attrs=field.widget.attrs,
|
|
queryset=getattr(field, 'queryset', None),
|
|
)
|
|
self.fields[field_name].help_text = ''
|
|
elif isinstance(field.widget, forms.widgets.Select):
|
|
self.fields[field_name].widget = DetailSelectMultiple(
|
|
choices=field.widget.choices,
|
|
attrs=field.widget.attrs,
|
|
queryset=getattr(field, 'queryset', None),
|
|
)
|
|
self.fields[field_name].help_text = ''
|
|
|
|
for field_name, field in self.fields.items():
|
|
self.fields[field_name].widget.attrs.update(
|
|
{'readonly': 'readonly'}
|
|
)
|
|
|
|
|
|
class ChoiceForm(forms.Form):
|
|
"""
|
|
Form to be used in side by side templates used to add or remove
|
|
items from a many to many field
|
|
"""
|
|
def __init__(self, *args, **kwargs):
|
|
choices = kwargs.pop('choices', [])
|
|
label = kwargs.pop('label', _('Selection'))
|
|
help_text = kwargs.pop('help_text', None)
|
|
disabled_choices = kwargs.pop('disabled_choices', ())
|
|
super(ChoiceForm, self).__init__(*args, **kwargs)
|
|
self.fields['selection'].choices = choices
|
|
self.fields['selection'].label = label
|
|
self.fields['selection'].help_text = help_text
|
|
self.fields['selection'].widget.disabled_choices = disabled_choices
|
|
self.fields['selection'].widget.attrs.update(
|
|
{'size': 14, 'class': 'choice_form'}
|
|
)
|
|
|
|
selection = forms.MultipleChoiceField(widget=DisableableSelectWidget())
|
|
|
|
|
|
class UserForm_view(DetailForm):
|
|
"""
|
|
Form used to display an user's public details
|
|
"""
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = (
|
|
'username', 'first_name', 'last_name', 'email', 'is_staff',
|
|
'is_superuser', 'last_login', 'date_joined', 'groups'
|
|
)
|
|
|
|
|
|
class UserForm(forms.ModelForm):
|
|
"""
|
|
Form used to edit an user's mininal fields by the user himself
|
|
"""
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ('username', 'first_name', 'last_name', 'email')
|
|
|
|
|
|
class LocaleProfileForm(forms.ModelForm):
|
|
class Meta:
|
|
model = UserLocaleProfile
|
|
fields = ('language', 'timezone')
|
|
|
|
|
|
class LocaleProfileForm_view(DetailForm):
|
|
class Meta:
|
|
model = UserLocaleProfile
|
|
fields = ('language', 'timezone')
|
|
|
|
|
|
class FileDisplayForm(forms.Form):
|
|
text = forms.CharField(
|
|
label='',
|
|
widget=forms.widgets.Textarea(
|
|
attrs={'cols': 40, 'rows': 20, 'readonly': 'readonly'}
|
|
)
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(FileDisplayForm, self).__init__(*args, **kwargs)
|
|
changelog_path = os.path.join(
|
|
settings.BASE_DIR, os.sep.join(self.DIRECTORY), self.FILENAME
|
|
)
|
|
fd = open(changelog_path)
|
|
self.fields['text'].initial = fd.read()
|
|
fd.close()
|
|
|
|
|
|
class FilterForm(forms.Form):
|
|
filter_slug = forms.ChoiceField(label=_('Field'))
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(FilterForm, self).__init__(*args, **kwargs)
|
|
self.fields['filter_slug'].choices = Filter.all().items()
|
|
|
|
|
|
class LicenseForm(FileDisplayForm):
|
|
FILENAME = 'LICENSE'
|
|
DIRECTORY = []
|