Alphabetic filter for admin

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# Authors: Marinho Brandao <marinho at gmail.com>
#          Guilherme M. Gondim (semente) <semente at taurinus.org>
# File: <your project>/admin/filterspecs.py

from django.db import models
from django.contrib.admin.filterspecs import FilterSpec, ChoicesFilterSpec
from django.utils.encoding import smart_unicode
from django.utils.translation import ugettext as _

class AlphabeticFilterSpec(ChoicesFilterSpec):
    """
    Adds filtering by first char (alphabetic style) of values in the admin
    filter sidebar. Set the alphabetic filter in the model field attribute
    'alphabetic_filter'.

    my_model_field.alphabetic_filter = True
    """

    def __init__(self, f, request, params, model, model_admin):
        super(AlphabeticFilterSpec, self).__init__(f, request, params, model,
                                                   model_admin)
        self.lookup_kwarg = '%s__istartswith' % f.name
        self.lookup_val = request.GET.get(self.lookup_kwarg, None)
        values_list = model.objects.values_list(f.name, flat=True)
        # getting the first char of values
        self.lookup_choices = list(set(val[0] for val in values_list if val))
        self.lookup_choices.sort()

    def choices(self, cl):
        yield {'selected': self.lookup_val is None,
                'query_string': cl.get_query_string({}, [self.lookup_kwarg]),
                'display': _('All')}
        for val in self.lookup_choices:
            yield {'selected': smart_unicode(val) == self.lookup_val,
                    'query_string': cl.get_query_string({self.lookup_kwarg: val}),
                    'display': val.upper()}
    def title(self):
        return _('%(field_name)s that starts with') % \
            {'field_name': self.field.verbose_name}

# registering the filter
FilterSpec.filter_specs.insert(0, (lambda f: getattr(f, 'alphabetic_filter', False),
                                   AlphabeticFilterSpec))

More like this

  1. Exists Filter OneToOneField in Admin by davidvaz 1 year, 6 months ago
  2. Optimistic locking in Admin by Taifu 1 year, 9 months ago
  3. AlphabeticFilterSpec by marinho 5 years ago
  4. Filter by taggit tags in the admin by flupke 1 year, 7 months ago
  5. Group results by a range of values in admin sidebar by wgollino 5 years, 3 months ago

Comments

magicrebirth (on April 23, 2009):

super!

#

magicrebirth (on May 13, 2009):

Line 55 is better changed like this:

self.lookup_choices = list(set(val[0] for val in values_list if val))

BY adding 'if val' it prevents the routine to crash when indexing items that are blank!

#

semente (on May 29, 2009):

snippet updated.

magicrebirth, thanks for the tip.

#

urosh (on August 12, 2009):

Request Method: GET Request URL: http://vi/voip/phonebook/ Exception Type: TypeError Exception Value:

'long' object is unsubscriptable

#

bradmontgomery (on January 14, 2010):

Thanks! This is very helpful.

#

luc_j (on November 10, 2010):

Very helpful. I've made the following changes. I hope it helps

Avoid duplicates for lower and uppercase

self.lookup_choices = list(set(val.upper()[0] for val in values_list if val))

The register must be called explicitely

@classmethod
def register_filterspec(cls):
    """register the filter. To be called in the models.py"""
    FilterSpec.filter_specs.insert(0,
        (lambda f: getattr(f, 'alphabetic_filter', False), cls)
    )

#

ohnoimdead (on May 15, 2011):

Need to add field_path to init:

def __init__(self, f, request, params, model, model_admin, field_path=None):
    super(AlphabeticFilterSpec, self).__init__(f, request, params, model,
                                               model_admin,
                                               field_path=field_path)

#

martinvanbuuren (on October 10, 2011):

Solution to "'long' object is unsubscriptable"

Change

26 self.lookup_choices = list(set(val[0] for val in values_list if val))

to

26 self.lookup_choices = list(set(str(val)[0] for val in values_list if val))

This issue rises at ForeignKeys. The solution returns the first character of the id's rather than an error message.

#

(Forgotten your password?)