26 lines
702 B
Python
26 lines
702 B
Python
from django import forms
|
|
from django.utils.translation import ugettext as _
|
|
|
|
from 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))
|
|
self.fields['title'].required = False
|
|
self.fields['title'].label = _(u'New folder')
|
|
|
|
class Meta:
|
|
model = Folder
|
|
fields = ('title',)
|
|
|
|
|