27 lines
759 B
Python
27 lines
759 B
Python
from django import forms
|
|
from django.utils.translation import ugettext as _
|
|
|
|
from folders.models import Folder
|
|
|
|
|
|
class FolderForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Folder
|
|
fields = ('title',)
|
|
|
|
|
|
class AddDocumentForm(forms.ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
user = kwargs.pop('user', None)
|
|
super(AddDocumentForm, self).__init__(*args, **kwargs)
|
|
self.fields['existing_folder'] = forms.ModelChoiceField(
|
|
required=False,
|
|
queryset=Folder.objects.filter(user=user),
|
|
label=_(u'Existing folders'))
|
|
self.fields['title'].required = False
|
|
self.fields['title'].label = _(u'New folder')
|
|
|
|
class Meta:
|
|
model = Folder
|
|
fields = ('title',)
|