Django-1.2.1/django/contrib/databrowse/plugins/fieldchoices.py

   1 from django import http
   2 from django.db import models
   3 from django.contrib.databrowse.datastructures import EasyModel
   4 from django.contrib.databrowse.sites import DatabrowsePlugin
   5 from django.shortcuts import render_to_response
   6 from django.utils.text import capfirst
   7 from django.utils.encoding import smart_str, force_unicode
   8 from django.utils.safestring import mark_safe
   9 import urllib
  10
  11 class FieldChoicePlugin(DatabrowsePlugin):
  12     def __init__(self, field_filter=None):
  13         # If field_filter is given, it should be a callable that takes a
  14         # Django database Field instance and returns True if that field should
  15         # be included. If field_filter is None, that all fields will be used.
  16         self.field_filter = field_filter
  17
  18     def field_dict(self, model):
  19         """
  20         Helper function that returns a dictionary of all fields in the given
  21         model. If self.field_filter is set, it only includes the fields that
  22         match the filter.
  23         """
  24         if self.field_filter:
  25             return dict([(f.name, f) for f in model._meta.fields if self.field_filter(f)])
  26         else:
  27             return dict([(f.name, f) for f in model._meta.fields if not f.rel and not f.primary_key and not f.unique and not isinstance(f, (models.AutoField, models.TextField))])
  28
  29     def model_index_html(self, request, model, site):
  30         fields = self.field_dict(model)
  31         if not fields:
  32             return u''
  33         return mark_safe(u'<p class="filter"><strong>View by:</strong> %s</p>' % \
  34             u', '.join(['<a href="fields/%s/">%s</a>' % (f.name, force_unicode(capfirst(f.verbose_name))) for f in fields.values()]))
  35
  36     def urls(self, plugin_name, easy_instance_field):
  37         if easy_instance_field.field in self.field_dict(easy_instance_field.model.model).values():
  38             field_value = smart_str(easy_instance_field.raw_value)
  39             return [mark_safe(u'%s%s/%s/%s/' % (
  40                 easy_instance_field.model.url(),
  41                 plugin_name, easy_instance_field.field.name,
  42                 urllib.quote(field_value, safe='')))]
  43
  44     def model_view(self, request, model_databrowse, url):
  45         self.model, self.site = model_databrowse.model, model_databrowse.site
  46         self.fields = self.field_dict(self.model)
  47
  48         # If the model has no fields with choices, there's no point in going
  49         # further.
  50         if not self.fields:
  51             raise http.Http404('The requested model has no fields.')
  52
  53         if url is None:
  54             return self.homepage_view(request)
  55         url_bits = url.split('/', 1)
  56         if self.fields.has_key(url_bits[0]):
  57             return self.field_view(request, self.fields[url_bits[0]], *url_bits[1:])
  58
  59         raise http.Http404('The requested page does not exist.')
  60
  61     def homepage_view(self, request):
  62         easy_model = EasyModel(self.site, self.model)
  63         field_list = self.fields.values()
  64         field_list.sort(lambda x, y: cmp(x.verbose_name, y.verbose_name))
  65         return render_to_response('databrowse/fieldchoice_homepage.html', {'root_url': self.site.root_url, 'model': easy_model, 'field_list': field_list})
  66
  67     def field_view(self, request, field, value=None):
  68         easy_model = EasyModel(self.site, self.model)
  69         easy_field = easy_model.field(field.name)
  70         if value is not None:
  71             obj_list = easy_model.objects(**{field.name: value})
  72             return render_to_response('databrowse/fieldchoice_detail.html', {'root_url': self.site.root_url, 'model': easy_model, 'field': easy_field, 'value': value, 'object_list': obj_list})
  73         obj_list = [v[field.name] for v in self.model._default_manager.distinct().order_by(field.name).values(field.name)]
  74         return render_to_response('databrowse/fieldchoice_list.html', {'root_url': self.site.root_url, 'model': easy_model, 'field': easy_field, 'object_list': obj_list})