Django-1.2.1/build/lib.linux-i686-2.6/django/contrib/databrowse/plugins/calendars.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 force_unicode
   8 from django.utils.safestring import mark_safe
   9 from django.views.generic import date_based
  10 from django.utils import datetime_safe
  11
  12 class CalendarPlugin(DatabrowsePlugin):
  13     def __init__(self, field_names=None):
  14         self.field_names = field_names
  15
  16     def field_dict(self, model):
  17         """
  18         Helper function that returns a dictionary of all DateFields or
  19         DateTimeFields in the given model. If self.field_names is set, it takes
  20         take that into account when building the dictionary.
  21         """
  22         if self.field_names is None:
  23             return dict([(f.name, f) for f in model._meta.fields if isinstance(f, models.DateField)])
  24         else:
  25             return dict([(f.name, f) for f in model._meta.fields if isinstance(f, models.DateField) and f.name in self.field_names])
  26
  27     def model_index_html(self, request, model, site):
  28         fields = self.field_dict(model)
  29         if not fields:
  30             return u''
  31         return mark_safe(u'<p class="filter"><strong>View calendar by:</strong> %s</p>' % \
  32             u', '.join(['<a href="calendars/%s/">%s</a>' % (f.name, force_unicode(capfirst(f.verbose_name))) for f in fields.values()]))
  33
  34     def urls(self, plugin_name, easy_instance_field):
  35         if isinstance(easy_instance_field.field, models.DateField):
  36             d = easy_instance_field.raw_value
  37             return [mark_safe(u'%s%s/%s/%s/%s/%s/' % (
  38                 easy_instance_field.model.url(),
  39                 plugin_name, easy_instance_field.field.name,
  40                 str(d.year),
  41                 datetime_safe.new_date(d).strftime('%b').lower(),
  42                 d.day))]
  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 DateFields, there's no point in going further.
  49         if not self.fields:
  50             raise http.Http404('The requested model has no calendars.')
  51
  52         if url is None:
  53             return self.homepage_view(request)
  54         url_bits = url.split('/')
  55         if self.fields.has_key(url_bits[0]):
  56             return self.calendar_view(request, self.fields[url_bits[0]], *url_bits[1:])
  57
  58         raise http.Http404('The requested page does not exist.')
  59
  60     def homepage_view(self, request):
  61         easy_model = EasyModel(self.site, self.model)
  62         field_list = self.fields.values()
  63         field_list.sort(lambda x, y: cmp(x.verbose_name, y.verbose_name))
  64         return render_to_response('databrowse/calendar_homepage.html', {'root_url': self.site.root_url, 'model': easy_model, 'field_list': field_list})
  65
  66     def calendar_view(self, request, field, year=None, month=None, day=None):
  67         easy_model = EasyModel(self.site, self.model)
  68         queryset = easy_model.get_query_set()
  69         extra_context = {'root_url': self.site.root_url, 'model': easy_model, 'field': field}
  70         if day is not None:
  71             return date_based.archive_day(request, year, month, day, queryset, field.name,
  72                 template_name='databrowse/calendar_day.html', allow_empty=False, allow_future=True,
  73                 extra_context=extra_context)
  74         elif month is not None:
  75             return date_based.archive_month(request, year, month, queryset, field.name,
  76                 template_name='databrowse/calendar_month.html', allow_empty=False, allow_future=True,
  77                 extra_context=extra_context)
  78         elif year is not None:
  79             return date_based.archive_year(request, year, queryset, field.name,
  80                 template_name='databrowse/calendar_year.html', allow_empty=False, allow_future=True,
  81                 extra_context=extra_context)
  82         else:
  83             return date_based.archive_index(request, queryset, field.name,
  84                 template_name='databrowse/calendar_main.html', allow_empty=True, allow_future=True,
  85                 extra_context=extra_context)
  86         assert False, ('%s, %s, %s, %s' % (field, year, month, day))