|
1 | 1 | #coding: utf-8 |
| 2 | +from django import template |
2 | 3 | from django.contrib import admin |
3 | | -from ietf.liaisons.models import * |
4 | | - |
| 4 | +from django.contrib.admin.util import unquote |
| 5 | +from django.core.exceptions import PermissionDenied |
| 6 | +from django.core.management import load_command_class |
| 7 | +from django.http import Http404 |
| 8 | +from django.shortcuts import render_to_response |
| 9 | +from django.utils.encoding import force_unicode |
| 10 | +from django.utils.functional import update_wrapper |
| 11 | +from django.utils.html import escape |
| 12 | +from django.utils.translation import ugettext as _ |
| 13 | + |
| 14 | +from ietf.liaisons.models import (FromBodies, LiaisonDetail, LiaisonPurpose, |
| 15 | + SDOs, LiaisonManagers, SDOAuthorizedIndividual) |
| 16 | + |
| 17 | + |
5 | 18 | class FromBodiesAdmin(admin.ModelAdmin): |
6 | 19 | pass |
7 | | -admin.site.register(FromBodies, FromBodiesAdmin) |
| 20 | + |
8 | 21 |
|
9 | 22 | class LiaisonDetailAdmin(admin.ModelAdmin): |
10 | | - pass |
11 | | -admin.site.register(LiaisonDetail, LiaisonDetailAdmin) |
| 23 | + ordering = ('title', ) |
| 24 | + fields = ('title', 'body','submitted_date', 'last_modified_date', 'to_email', 'cc1', 'cc2', 'to_poc', |
| 25 | + 'response_contact', 'technical_contact', 'purpose', 'purpose_text', 'deadline_date', 'taken_care', |
| 26 | + 'related_to') |
| 27 | + raw_id_fields=['related_to'] |
12 | 28 |
|
13 | 29 | class LiaisonPurposeAdmin(admin.ModelAdmin): |
14 | | - pass |
15 | | -admin.site.register(LiaisonPurpose, LiaisonPurposeAdmin) |
| 30 | + ordering = ('purpose_text', ) |
| 31 | + |
| 32 | + |
| 33 | +class LiaisonManagersInline(admin.TabularInline): |
| 34 | + model = LiaisonManagers |
| 35 | + raw_id_fields=['person'] |
| 36 | + |
| 37 | + |
| 38 | +class SDOAuthorizedIndividualInline(admin.TabularInline): |
| 39 | + model = SDOAuthorizedIndividual |
| 40 | + raw_id_fields=['person'] |
| 41 | + |
| 42 | + |
| 43 | +class LiaisonManagersAdmin(admin.ModelAdmin): |
| 44 | + ordering = ('person__first_name', 'person__last_name' ) |
| 45 | + fields = ('person', 'sdo') |
| 46 | + raw_id_fields=['person'] |
| 47 | + |
| 48 | + |
| 49 | +class SDOAuthorizedIndividualAdmin(admin.ModelAdmin): |
| 50 | + raw_id_fields=['person'] |
| 51 | + |
16 | 52 |
|
| 53 | +class SDOsAdmin(admin.ModelAdmin): |
| 54 | + inlines = [LiaisonManagersInline, SDOAuthorizedIndividualInline] |
| 55 | + |
| 56 | + def get_urls(self): |
| 57 | + from django.conf.urls.defaults import patterns, url |
| 58 | + |
| 59 | + def wrap(view): |
| 60 | + def wrapper(*args, **kwargs): |
| 61 | + return self.admin_site.admin_view(view)(*args, **kwargs) |
| 62 | + return update_wrapper(wrapper, view) |
| 63 | + |
| 64 | + info = self.model._meta.app_label, self.model._meta.module_name |
| 65 | + |
| 66 | + urls = patterns('', |
| 67 | + url(r'^reminder/$', |
| 68 | + wrap(self.send_reminder), |
| 69 | + name='%s_%s_reminder' % info), |
| 70 | + url(r'^(.+)/reminder/$', |
| 71 | + wrap(self.send_one_reminder), |
| 72 | + name='%s_%s_one_reminder' % info), |
| 73 | + ) |
| 74 | + urls += super(SDOsAdmin, self).get_urls() |
| 75 | + return urls |
| 76 | + |
| 77 | + def send_reminder(self, request, sdo=None): |
| 78 | + opts = self.model._meta |
| 79 | + app_label = opts.app_label |
| 80 | + |
| 81 | + output = None |
| 82 | + sdo_pk = sdo and sdo.pk or None |
| 83 | + if request.method == 'POST' and request.POST.get('send', False): |
| 84 | + command = load_command_class('ietf.liaisons', 'remind_update_sdo_list') |
| 85 | + output=command.handle(return_output=True, sdo_pk=sdo_pk) |
| 86 | + output='\n'.join(output) |
| 87 | + |
| 88 | + context = { |
| 89 | + 'opts': opts, |
| 90 | + 'has_change_permission': self.has_change_permission(request), |
| 91 | + 'app_label': app_label, |
| 92 | + 'output': output, |
| 93 | + 'sdo': sdo, |
| 94 | + } |
| 95 | + return render_to_response('admin/liaisons/sdos/send_reminder.html', |
| 96 | + context, |
| 97 | + context_instance = template.RequestContext(request, current_app=self.admin_site.name), |
| 98 | + ) |
| 99 | + |
| 100 | + def send_one_reminder(self, request, object_id): |
| 101 | + model = self.model |
| 102 | + opts = model._meta |
| 103 | + |
| 104 | + try: |
| 105 | + obj = self.queryset(request).get(pk=unquote(object_id)) |
| 106 | + except model.DoesNotExist: |
| 107 | + obj = None |
| 108 | + |
| 109 | + if not self.has_change_permission(request, obj): |
| 110 | + raise PermissionDenied |
| 111 | + |
| 112 | + if obj is None: |
| 113 | + raise Http404(_('%(name)s object with primary key %(key)r does not exist.') % {'name': force_unicode(opts.verbose_name), 'key': escape(object_id)}) |
| 114 | + |
| 115 | + return self.send_reminder(request, sdo=obj) |
| 116 | + |
| 117 | + |
| 118 | +#admin.site.register(FromBodies, FromBodiesAdmin) |
| 119 | +admin.site.register(LiaisonDetail, LiaisonDetailAdmin) |
| 120 | +admin.site.register(LiaisonPurpose, LiaisonPurposeAdmin) |
| 121 | +admin.site.register(SDOs, SDOsAdmin) |
| 122 | +admin.site.register(LiaisonManagers, LiaisonManagersAdmin) |
| 123 | +admin.site.register(SDOAuthorizedIndividual, SDOAuthorizedIndividualAdmin) |
0 commit comments