|
1 | 1 | from django.contrib import admin |
| 2 | +from django import template |
| 3 | +from django.utils.functional import update_wrapper |
| 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 | + |
2 | 14 | from models import * |
3 | 15 |
|
4 | 16 | class GroupAdmin(admin.ModelAdmin): |
5 | 17 | list_display = ["acronym", "name", "type"] |
| 18 | + list_display_links = ["acronym", "name"] |
| 19 | + list_filter = ["type"] |
6 | 20 | search_fields = ["name"] |
7 | 21 | ordering = ["name"] |
8 | | - raw_id_fields = ["charter"] |
| 22 | + raw_id_fields = ["charter", "parent", "ad"] |
| 23 | + |
| 24 | + def get_urls(self): |
| 25 | + from django.conf.urls.defaults import patterns, url |
| 26 | + |
| 27 | + def wrap(view): |
| 28 | + def wrapper(*args, **kwargs): |
| 29 | + return self.admin_site.admin_view(view)(*args, **kwargs) |
| 30 | + return update_wrapper(wrapper, view) |
| 31 | + |
| 32 | + info = self.model._meta.app_label, self.model._meta.module_name |
| 33 | + |
| 34 | + urls = patterns('', |
| 35 | + url(r'^reminder/$', |
| 36 | + wrap(self.send_reminder), |
| 37 | + name='%s_%s_reminder' % info), |
| 38 | + url(r'^(.+)/reminder/$', |
| 39 | + wrap(self.send_one_reminder), |
| 40 | + name='%s_%s_one_reminder' % info), |
| 41 | + ) |
| 42 | + urls += super(GroupAdmin, self).get_urls() |
| 43 | + return urls |
| 44 | + |
| 45 | + def send_reminder(self, request, sdo=None): |
| 46 | + opts = self.model._meta |
| 47 | + app_label = opts.app_label |
| 48 | + |
| 49 | + output = None |
| 50 | + sdo_pk = sdo and sdo.pk or None |
| 51 | + if request.method == 'POST' and request.POST.get('send', False): |
| 52 | + command = load_command_class('ietf.liaisons', 'remind_update_sdo_list') |
| 53 | + output=command.handle(return_output=True, sdo_pk=sdo_pk) |
| 54 | + output='\n'.join(output) |
| 55 | + |
| 56 | + context = { |
| 57 | + 'opts': opts, |
| 58 | + 'has_change_permission': self.has_change_permission(request), |
| 59 | + 'app_label': app_label, |
| 60 | + 'output': output, |
| 61 | + 'sdo': sdo, |
| 62 | + } |
| 63 | + return render_to_response('admin/group/group/send_sdo_reminder.html', |
| 64 | + context, |
| 65 | + context_instance = template.RequestContext(request, current_app=self.admin_site.name), |
| 66 | + ) |
| 67 | + |
| 68 | + def send_one_reminder(self, request, object_id): |
| 69 | + model = self.model |
| 70 | + opts = model._meta |
| 71 | + |
| 72 | + try: |
| 73 | + obj = self.queryset(request).get(pk=unquote(object_id)) |
| 74 | + except model.DoesNotExist: |
| 75 | + obj = None |
| 76 | + |
| 77 | + if not self.has_change_permission(request, obj): |
| 78 | + raise PermissionDenied |
| 79 | + |
| 80 | + if obj is None: |
| 81 | + raise Http404(_('%(name)s object with primary key %(key)r does not exist.') % {'name': force_unicode(opts.verbose_name), 'key': escape(object_id)}) |
| 82 | + |
| 83 | + return self.send_reminder(request, sdo=obj) |
| 84 | + |
9 | 85 |
|
10 | 86 | admin.site.register(Group, GroupAdmin) |
11 | 87 | admin.site.register(GroupHistory) |
12 | 88 |
|
13 | | -admin.site.register(Role) |
| 89 | +class RoleAdmin(admin.ModelAdmin): |
| 90 | + list_display = ["name", "email", "group"] |
| 91 | + list_display_links = ["name"] |
| 92 | + search_fields = ["name", "email"] |
| 93 | + list_filter = ["name"] |
| 94 | + ordering = ["id"] |
| 95 | + raw_id_fields = ["email", "group"] |
| 96 | + |
| 97 | +admin.site.register(Role, RoleAdmin) |
0 commit comments