Filtrado fácil de clave foránea en una ficha de django


Es sabido que django mapea las claves foráneas a un ModelChoiceField en ModelForm, pero ¿qué ocurre cuando sólo queremos seleccionar un subconjunto de ellas?.

Por ejemplo, supongamos que estamos dando de alta un empleado y a la hora de seleccionar su categoría sólo queremos que aparezcan las que están asociadas a su sección.

Tenemos la ficha:


class AddForm(forms.ModelForm):
  class Meta:
   model=Empleado
    fields=('clave', 'descripcion', 'categoria', 'jefe_seccion', 'email',)


La propiedad queryset del campo categoría estará definida por defecto como Categoria.objects.all(), así que hacemos:

p=AddForm()
p.fields['categoria'].queryset=Categoria.objects.filter(seccion=self.seccion)

Con lo que en la ficha sólo nos saldrán las categorías de la sección actual.
Otra forma más complicada consiste en pasarle el parámetro al constructor de la clase pero hay que hacerlo con los kwargs por que si no tendremos problemas en cuando queramos usar la ficha para editar.

class AddForm(forms.ModelForm):
  categoria=forms.ModelChoiceField( label=u'Categoría', queryset = Categoria.objects.none())
  def __init__(self, *args, **kwargs):
   sec = kwargs.pop('sec', None)
   super(AddForm, self).__init__(*args, **kwargs)
   if sec:
    self.fields['categoria'].queryset = Categoria.objects.filter(seccion = sec)
  class Meta:
   model=Empleado
   fields=('clave', 'descripcion', 'categoria', 'jefe_seccion', 'email',)