|
1 |
|
2 from django.contrib.auth.models import User, Group |
|
3 from django.core.exceptions import PermissionDenied |
|
4 from django import template |
|
5 from django.shortcuts import render_to_response, get_object_or_404 |
|
6 from django.template import RequestContext |
|
7 from django.utils.html import escape |
|
8 from django.http import HttpResponseRedirect |
|
9 from django.utils.translation import ugettext, ugettext_lazy as _ |
|
10 from django.contrib.auth.forms import UserCreationForm, UserChangeForm, AdminPasswordChangeForm |
|
11 from django.contrib import admin |
|
12 |
|
13 class GroupAdmin(admin.ModelAdmin): |
|
14 search_fields = ('name',) |
|
15 ordering = ('name',) |
|
16 filter_horizontal = ('permissions',) |
|
17 |
|
18 class UserAdmin(admin.ModelAdmin): |
|
19 fieldsets = ( |
|
20 (None, {'fields': ('username', 'password')}), |
|
21 (_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}), |
|
22 (_('Permissions'), {'fields': ('is_staff', 'is_active', 'is_superuser', 'user_permissions')}), |
|
23 (_('Important dates'), {'fields': ('last_login', 'date_joined')}), |
|
24 (_('Groups'), {'fields': ('groups',)}), |
|
25 ) |
|
26 form = UserChangeForm |
|
27 add_form = UserCreationForm |
|
28 change_password_form = AdminPasswordChangeForm |
|
29 list_display = ('username', 'email', 'first_name', 'last_name', 'is_staff') |
|
30 list_filter = ('is_staff', 'is_superuser') |
|
31 search_fields = ('username', 'first_name', 'last_name', 'email') |
|
32 ordering = ('username',) |
|
33 filter_horizontal = ('user_permissions',) |
|
34 |
|
35 def __call__(self, request, url): |
|
36 # this should not be here, but must be due to the way __call__ routes |
|
37 # in ModelAdmin. |
|
38 if url is None: |
|
39 return self.changelist_view(request) |
|
40 if url.endswith('password'): |
|
41 return self.user_change_password(request, url.split('/')[0]) |
|
42 return super(UserAdmin, self).__call__(request, url) |
|
43 |
|
44 def add_view(self, request): |
|
45 if not self.has_change_permission(request): |
|
46 raise PermissionDenied |
|
47 if request.method == 'POST': |
|
48 form = self.add_form(request.POST) |
|
49 if form.is_valid(): |
|
50 new_user = form.save() |
|
51 msg = _('The %(name)s "%(obj)s" was added successfully.') % {'name': 'user', 'obj': new_user} |
|
52 self.log_addition(request, new_user) |
|
53 if "_addanother" in request.POST: |
|
54 request.user.message_set.create(message=msg) |
|
55 return HttpResponseRedirect(request.path) |
|
56 elif '_popup' in request.REQUEST: |
|
57 return self.response_add(request, new_user) |
|
58 else: |
|
59 request.user.message_set.create(message=msg + ' ' + ugettext("You may edit it again below.")) |
|
60 return HttpResponseRedirect('../%s/' % new_user.id) |
|
61 else: |
|
62 form = self.add_form() |
|
63 return render_to_response('admin/auth/user/add_form.html', { |
|
64 'title': _('Add user'), |
|
65 'form': form, |
|
66 'is_popup': '_popup' in request.REQUEST, |
|
67 'add': True, |
|
68 'change': False, |
|
69 'has_add_permission': True, |
|
70 'has_delete_permission': False, |
|
71 'has_change_permission': True, |
|
72 'has_file_field': False, |
|
73 'has_absolute_url': False, |
|
74 'auto_populated_fields': (), |
|
75 'opts': self.model._meta, |
|
76 'save_as': False, |
|
77 'username_help_text': self.model._meta.get_field('username').help_text, |
|
78 'root_path': self.admin_site.root_path, |
|
79 'app_label': self.model._meta.app_label, |
|
80 }, context_instance=template.RequestContext(request)) |
|
81 |
|
82 def user_change_password(self, request, id): |
|
83 if not request.user.has_perm('auth.change_user'): |
|
84 raise PermissionDenied |
|
85 user = get_object_or_404(self.model, pk=id) |
|
86 if request.method == 'POST': |
|
87 form = self.change_password_form(user, request.POST) |
|
88 if form.is_valid(): |
|
89 new_user = form.save() |
|
90 msg = ugettext('Password changed successfully.') |
|
91 request.user.message_set.create(message=msg) |
|
92 return HttpResponseRedirect('..') |
|
93 else: |
|
94 form = self.change_password_form(user) |
|
95 return render_to_response('admin/auth/user/change_password.html', { |
|
96 'title': _('Change password: %s') % escape(user.username), |
|
97 'form': form, |
|
98 'is_popup': '_popup' in request.REQUEST, |
|
99 'add': True, |
|
100 'change': False, |
|
101 'has_delete_permission': False, |
|
102 'has_change_permission': True, |
|
103 'has_absolute_url': False, |
|
104 'opts': self.model._meta, |
|
105 'original': user, |
|
106 'save_as': False, |
|
107 'show_save': True, |
|
108 'root_path': self.admin_site.root_path, |
|
109 }, context_instance=RequestContext(request)) |
|
110 |
|
111 |
|
112 admin.site.register(Group, GroupAdmin) |
|
113 admin.site.register(User, UserAdmin) |
|
114 |