from django import forms from django.forms import ModelForm from django.contrib.auth.models import User, Group from todo.models import Item, List class AddListForm(ModelForm): # The picklist showing allowable groups to which a new list can be added # determines which groups the user belongs to. This queries the form object # to derive that list. def __init__(self, user, *args, **kwargs): super(AddListForm, self).__init__(*args, **kwargs) self.fields['group'].queryset = Group.objects.filter(user=user) class Meta: model = List exclude = [] class AddItemForm(ModelForm): # The picklist showing the users to which a new task can be assigned # must find other members of the groups the current list belongs to. def __init__(self, task_list, *args, **kwargs): super(AddItemForm, self).__init__(*args, **kwargs) # print dir(self.fields['list']) # print self.fields['list'].initial self.fields['assigned_to'].queryset = User.objects.filter(groups__in=[task_list.group]) self.fields['assigned_to'].label_from_instance = lambda obj: "%s (%s)" % ( obj.get_full_name(), obj.username) due_date = forms.DateField( required=False, widget=forms.DateTimeInput(attrs={'class': 'due_date_picker'}) ) title = forms.CharField( widget=forms.widgets.TextInput(attrs={'size': 35}) ) # note = forms.CharField( # widget=forms.TextArea(attrs={'size': 35}) # ) note = forms.CharField(widget=forms.Textarea()) class Meta: model = Item exclude = [] class EditItemForm(ModelForm): # The picklist showing the users to which a new task can be assigned # must find other members of the groups the current list belongs to. def __init__(self, *args, **kwargs): super(EditItemForm, self).__init__(*args, **kwargs) self.fields['assigned_to'].queryset = User.objects.filter(groups__in=[self.instance.list.group]) class Meta: model = Item exclude = ('created_date', 'created_by',) class AddExternalItemForm(ModelForm): """Form to allow users who are not part of the GTD system to file a ticket.""" title = forms.CharField( widget=forms.widgets.TextInput(attrs={'size': 35}) ) note = forms.CharField( widget=forms.widgets.Textarea(), help_text='Foo', ) class Meta: model = Item exclude = ('list', 'created_date', 'due_date', 'created_by', 'assigned_to',) class SearchForm(ModelForm): """Search.""" q = forms.CharField( widget=forms.widgets.TextInput(attrs={'size': 35}) )