symposion_app/symposion/schedule/forms.py
2012-09-13 23:17:54 -06:00

22 lines
717 B
Python

from django import forms
from django.db.models import Q
from symposion.schedule.models import Presentation
class SlotEditForm(forms.Form):
presentation = forms.ModelChoiceField(
queryset=Presentation.objects.all(),
required=True,
)
def __init__(self, *args, **kwargs):
presentation = kwargs.get("initial", {}).get("presentation")
super(SlotEditForm, self).__init__(*args, **kwargs)
queryset = self.fields["presentation"].queryset
if presentation:
queryset = queryset.filter(Q(slot=None) | Q(pk=presentation.pk))
else:
queryset = queryset.filter(slot=None)
self.fields["presentation"].queryset = queryset