1

選択肢を現在のユーザーのセットに制限するには、次のようにする必要があります。

class ChoiceForm(forms.ModelForm):
    def __init__(self, user, *args, **kwargs):
        super(ChoiceForm, self).__init__(*args, **kwargs)
        self.fields['poll'].queryset = Poll.objects.filter(owner=user)

    class Meta:
        model = Choice

これは、プロジェクトのサイズによってはかなり繰り返しになる可能性があります。さらに、管理フォームに対して同じことをしなければならなくなります。これを回避することは可能ですか?

4

1 に答える 1

1

Use generic views and define a get_form method to add the user to the form's kwargs, then create a custom base form class which accepts an extra user kwarg, like so:

Form

class UserModelForm(ModelForm):
    """
    Base form class with user-based FK filtering.
    """
    def __init__(self, *args, **kwargs):
        """
        Filters the queryset on foreign keys to limit to the provided user.
        """
        # Pop the user from the kwargs
        user = kwargs.pop('user')

        # Delegate initilization to super using the remaining args and kwargs
        super(UserModelForm, self).__init__(*args, **kwargs)

        self.prepare_for_user(user)

    def prepare_for_user(self, user):
        """
        Handles user related setup - for use inside __init__ - abstracted
        from __init__ to make testing easier.
        """
        # Get the user's PK
        user_pk = user.pk

        # Filter foriegn key fields to records owned by the user

        # Make a generator of the form's FK fields
        fks = [f for f in self.fields.itervalues() if hasattr(f, 'queryset')]

        # Iterate the FK fields and filter them - you might want to check
        # for something like `if hasattr(field.queryset.model, 'user'):`, etc.
        for field in fks:
            field.queryset = field.queryset.filter(user_id=user_pk)

View

class MyView(object):
    """
    A mixin for views.
    """
    def get_form(self, form_class):
    """
    Returns the provided form class, instantiated with the form kwargs,
    adding a user kwarg.
    """
    # Create some form kwargs with the user included
    defaults = {'user': self.request.user}

    # Update the kwargs with the default form kwargs
    defaults.update(self.get_form_kwargs())

    return form_class(**defaults)
于 2013-05-30T00:53:24.150 に答える