简体   繁体   中英

Custom Filter in Django Admin returns SuspiciousOperation

I am trying to implement a new version of this snippet , making it compatible for Django 1.4 and 1.5

Quite a bit has evolved, the code below almost working excepts that Django returns me a SuspiciousOperation error. I see how I could hack it, but I would prefer not to touch the core of Django. If you have some suggestions, there are welcome :

Here is my filter :

class RelatedNullFilterSpec(FieldListFilter):
    def __init__(self, field, request, params, model, model_admin, field_path):
        field_root, field_name = field_path.rsplit('__', 1)
        self.lookup_title = field.verbose_name
        self.title = self.lookup_title
        self.null_lookup_kwarg = '%s__isnull' % field_root
        self.null_lookup_val = request.GET.get(self.null_lookup_kwarg, None)
        self.lookup_kwarg = '%s__exact' % (field_path)
        self.lookup_val = request.GET.get(self.lookup_kwarg, None)
        if isinstance(field, models.fields.BooleanField):
            self.lookup_choices = (
                # (None, _('All')),
                ('1', _('Yes')),
                ('0', _('No')))
        else:
            self.lookup_choices = field.get_choices(include_blank=False)
        super(RelatedNullFilterSpec, self).__init__(field, request, params, model, model_admin, field_path)

    def expected_parameters(self):
        return [self.lookup_kwarg, self.null_lookup_kwarg]

    def choices(self, cl):
        yield {'selected': self.lookup_val is None and self.null_lookup_val is None,
               'query_string': cl.get_query_string({}, [self.lookup_kwarg,self.null_lookup_kwarg]),
               'display': _('All')}
        yield {'selected': self.lookup_val is None and self.null_lookup_val=="True",
               'query_string': cl.get_query_string({self.null_lookup_kwarg:True},[self.lookup_kwarg]),
               'display': _('Null')}
        yield {'selected': self.lookup_val is None and self.null_lookup_val=="False",
               'query_string': cl.get_query_string({self.null_lookup_kwarg:False},[self.lookup_kwarg]),
               'display': _('Not Null')}
        for pk_val, val in self.lookup_choices:
            yield {'selected': self.lookup_val == smart_unicode(pk_val),
                   'query_string': cl.get_query_string({self.lookup_kwarg: pk_val},[self.null_lookup_kwarg]),
                   'display': val}

Then in my admin, I have the following:

list_filter = ('time_added', 'time_modified', ('model1__model2__property', RelatedNullFilterSpec),)

And I always got this error from the lookup_allowed method of the Django BaseModelAdmin class...

In django.db.models.options, I could implement a hack to overwrite or extend the self.related_fkey_lookups but it is a bit too hacky to my taste.

EDIT: Notice that the following almost standard filter also returns the same error: ('venue__eat_venue', BooleanFieldListFilter)

In general, my aim is that I would like a filter that allows me to sort the objects by presence/absence of the model2 related field (Null/Not-Null) and by the values of the property (in case the model2 related field exists). It would be quite handy and I don't think too specific.

And finally, yes, everything works when I do not request this custom filter for my model1__model2__property :-)

This appears to be a bug in Django, when you have double-underscores for the filter path (first element of the tuple)... in your case 'model1__model2__property'

NOTE: now fixed in Django 1.7

More details here: https://code.djangoproject.com/ticket/19182

It's quite ugly but so far the only workaround I found was to paste this fixed and overridden method into the ModelAdmin where you want to use that list_filter :

def lookup_allowed(self, lookup, value):
    """
    Copy and pasted from ModelAdmin to fix bug in Django
    https://code.djangoproject.com/ticket/19182
    """
    from django.db.models.constants import LOOKUP_SEP
    from django.db.models.sql.constants import QUERY_TERMS
    from django.contrib.admin import FieldListFilter
    from django.contrib.admin import widgets

    model = self.model
    # Check FKey lookups that are allowed, so that popups produced by
    # ForeignKeyRawIdWidget, on the basis of ForeignKey.limit_choices_to,
    # are allowed to work.
    for l in model._meta.related_fkey_lookups:
        for k, v in widgets.url_params_from_lookup_dict(l).items():
            if k == lookup and v == value:
                return True

    parts = lookup.split(LOOKUP_SEP)

    # Last term in lookup is a query term (__exact, __startswith etc)
    # This term can be ignored.
    if len(parts) > 1 and parts[-1] in QUERY_TERMS:
        parts.pop()

    # Special case -- foo__id__exact and foo__id queries are implied
    # if foo has been specificially included in the lookup list; so
    # drop __id if it is the last part. However, first we need to find
    # the pk attribute name.
    rel_name = None
    for part in parts[:-1]:
        try:
            field, _, _, _ = model._meta.get_field_by_name(part)
        except FieldDoesNotExist:
            # Lookups on non-existants fields are ok, since they're ignored
            # later.
            return True
        if hasattr(field, 'rel'):
            model = field.rel.to
            rel_name = field.rel.get_related_field().name
        elif isinstance(field, RelatedObject):
            model = field.model
            rel_name = model._meta.pk.name
        else:
            rel_name = None
    if rel_name and len(parts) > 1 and parts[-1] == rel_name:
        parts.pop()

    if len(parts) == 1:
        return True
    clean_lookup = LOOKUP_SEP.join(parts)
    // FIX BEGINS:
    valid_lookups = [self.date_hierarchy]
    for filter_item in self.list_filter:
        if callable(filter_item):
            valid_lookups.append(filter_item.parameter_name)
        elif isinstance(filter_item, (list, tuple)):
            valid_lookups.append(filter_item[0])
        else:
            valid_lookups.append(filter_item)
    return clean_lookup in valid_lookups

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM