contrib/sobol/admin.py

   1 from django.conf import settings
   2 from django.conf.urls.defaults import patterns, url
   3 from django.contrib import admin
   4 from django.core.urlresolvers import reverse
   5 from django.db.models import Count
   6 from django.http import HttpResponseRedirect, Http404
   7 from django.shortcuts import render_to_response
   8 from django.template import RequestContext
   9 from django.utils.functional import update_wrapper
  10 from philo.admin import EntityAdmin
  11 from philo.contrib.sobol.models import Search, ResultURL, SearchView
  12
  13
  14 class ResultURLInline(admin.TabularInline):
  15         model = ResultURL
  16         readonly_fields = ('url',)
  17         can_delete = False
  18         extra = 0
  19         max_num = 0
  20
  21
  22 class SearchAdmin(admin.ModelAdmin):
  23         readonly_fields = ('string',)
  24         inlines = [ResultURLInline]
  25         list_display = ['string', 'unique_urls', 'total_clicks']
  26         search_fields = ['string', 'result_urls__url']
  27         actions = ['results_action']
  28         if 'grappelli' in settings.INSTALLED_APPS:
  29                 results_template = 'admin/sobol/search/grappelli_results.html'
  30         else:
  31                 results_template = 'admin/sobol/search/results.html'
  32
  33         def get_urls(self):
  34                 urlpatterns = super(SearchAdmin, self).get_urls()
  35
  36                 def wrap(view):
  37                         def wrapper(*args, **kwargs):
  38                                 return self.admin_site.admin_view(view)(*args, **kwargs)
  39                         return update_wrapper(wrapper, view)
  40
  41                 info = self.model._meta.app_label, self.model._meta.module_name
  42
  43                 urlpatterns = patterns('',
  44                         url(r'^results/$', wrap(self.results_view), name="%s_%s_selected_results" % info),
  45                         url(r'^(.+)/results/$', wrap(self.results_view), name="%s_%s_results" % info)
  46                 ) + urlpatterns
  47                 return urlpatterns
  48
  49         def unique_urls(self, obj):
  50                 return obj.unique_urls
  51         unique_urls.admin_order_field = 'unique_urls'
  52
  53         def total_clicks(self, obj):
  54                 return obj.total_clicks
  55         total_clicks.admin_order_field = 'total_clicks'
  56
  57         def queryset(self, request):
  58                 qs = super(SearchAdmin, self).queryset(request)
  59                 return qs.annotate(total_clicks=Count('result_urls__clicks', distinct=True), unique_urls=Count('result_urls', distinct=True))
  60
  61         def results_action(self, request, queryset):
  62                 info = self.model._meta.app_label, self.model._meta.module_name
  63                 if len(queryset) == 1:
  64                         return HttpResponseRedirect(reverse("admin:%s_%s_results" % info, args=(queryset[0].pk,)))
  65                 else:
  66                         url = reverse("admin:%s_%s_selected_results" % info)
  67                         return HttpResponseRedirect("%s?ids=%s" % (url, ','.join([str(item.pk) for item in queryset])))
  68         results_action.short_description = "View results for selected %(verbose_name_plural)s"
  69
  70         def results_view(self, request, object_id=None, extra_context=None):
  71                 if object_id is not None:
  72                         object_ids = [object_id]
  73                 else:
  74                         object_ids = request.GET.get('ids').split(',')
  75
  76                         if object_ids is None:
  77                                 raise Http404
  78
  79                 qs = self.queryset(request).filter(pk__in=object_ids)
  80                 opts = self.model._meta
  81
  82                 context = {
  83                         'queryset': qs,
  84                         'opts': opts,
  85                         'root_path': self.admin_site.root_path,
  86                         'app_label': opts.app_label
  87                 }
  88                 return render_to_response(self.results_template, context, context_instance=RequestContext(request))
  89
  90
  91 class SearchViewAdmin(EntityAdmin):
  92         raw_id_fields = ('results_page',)
  93         related_lookup_fields = {'fk': raw_id_fields}
  94
  95
  96 admin.site.register(Search, SearchAdmin)
  97 admin.site.register(SearchView, SearchViewAdmin)