Django: заставить ModelChoiceField оценивать набор запросов во время выполнения - PullRequest
Новая
       14

Django: заставить ModelChoiceField оценивать набор запросов во время выполнения

1 голос
/ 18 марта 2011

Я переопределил диспетчер по умолчанию для моих моделей, чтобы отображать только разрешенные элементы в соответствии с зарегистрированным пользователем (своего рода разрешение для конкретного объекта): 

class User_manager(models.Manager):
    def get_query_set(self):
        """ Filter results according to logged user """
        #Compose a filter dictionary with current user (stored in a middleware method)
        user_filter = middleware.get_user_filter() 
        return super(User_manager, self).get_query_set().filter(**user_filter)

class Foo(models.Model):
    objects = User_manager()
    ...

Таким образом, всякий раз, когдаЯ использую Foo.objects, текущий пользователь извлекается, и фильтр применяется к набору запросов по умолчанию, чтобы показать только разрешенные записи.

Затем у меня есть модель с ForeignKey для Foo: 

class Bar(models.Model):
    foo = models.ForeignKey(Foo)

class BarForm(form.ModelForm):
    class Meta:
        model = Bar

Когда я сочиняю BarForm, я ожидаю увидеть только экземпляры Foo фильтра, но фильтр не применяется.Я думаю, это потому, что набор запросов оценивается и кэшируется при запуске Django, когда ни один пользователь не вошел в систему и не применен фильтр.

Есть ли способ заставить Django оценивать набор запросов ModelChoice во время выполнения,без необходимости делать это явным в определении формы?(несмотря на все проблемы с производительностью ...)

РЕДАКТИРОВАТЬ Я нашел, где оценивается набор запросов (django \ db \ models \ fields \ related.py: 887): 

def formfield(self, **kwargs):
    db = kwargs.pop('using', None)
    defaults = {
        'form_class': forms.ModelChoiceField,
        'queryset': self.rel.to._default_manager.using(db).complex_filter(self.rel.limit_choices_to),
        'to_field_name': self.rel.field_name,
    }
    defaults.update(kwargs)
    return super(ForeignKey, self).formfield(**defaults)

Есть подсказка?

Ответы [ 3 ]

2 голосов
/ 30 сентября 2013

Была именно эта проблема - нужно было заполнить форму выбора пользовательскими объектами из группы, но ответ fun_vit неверен (по крайней мере, для django 1.5)

Во-первых, вы не хотите перезаписывать поле ['somefield']. Choices object - это объект ModelChoiceIterator, а не набор запросов. Во-вторых, комментарий в django.forms.BaseForm предупреждает вас о переопределении base_fields: 

    # The base_fields class attribute is the *class-wide* definition of
    # fields. Because a particular *instance* of the class might want to
    # alter self.fields, we create self.fields here by copying base_fields.
    # Instances should always modify self.fields; they should not modify
    # self.base_fields.

Это сработало для меня (django 1.5): 

class MyForm(ModelForm):
    users = ModelMultipleChoiceField(queryset=User.objects.none())

    def __init__(self, *args, **kwargs):
        super(MyForm, self).__init__(*args,**kwargs)

        site = Site.objects.get_current()
        self.fields['users'].queryset = site.user_group.user_set.all()

    class Meta:
        model = MyModel
1 голос
/ 08 ноября 2011

я использую init пользовательской формы: 

class BT_Form(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(BT_Form, self).__init__(*args, **kwargs)

        #prepare new values
        cities = [(u'',u'------')] #default value
        cities.extend([
            (
                c.pk, 
                c.__unicode__()
            ) for c in City.objects.filter(enabled=True).all()
        ])

        self.fields['fly_from_city'].choices = cities #renew values
0 голосов
/ 22 апреля 2011

Нет: мне пришлось переписать определение набора запросов (которое оценивается при запуске)

...