1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
|
# -*- coding: utf-8 -*-
from django.contrib import admin
from django.conf.urls import url
from django.utils.html import format_html
from django.utils.translation import ugettext_lazy as _
from django.http import HttpResponse
from account.models import UserProfile, UserFile
from account.csv_unicode import UnicodeReader, UnicodeWriter
import os
import datetime
import csv
class UserProfileAdmin(admin.ModelAdmin):
"""
customize the admin interface for UserProfile
"""
actions = [
'approve_users',
'cancel_approve_users',
'reset_approve_users',
'sponsor_users',
'cancel_sponsor_users',
'reset_sponsor_users',
'users_checkin',
'users_not_checkin',
'users_checkin_reset',
]
list_display = (
'user',
'email',
'realname',
'gender',
'institute',
'identity',
'reason',
'transcript_url',
'supplement',
'attachments',
'is_approved',
'is_sponsored',
'is_checkin',
)
# search fields
search_fields = [
'user__username',
'user__email',
'realname',
'institute',
]
## custom admin actions
def approve_users(self, request, queryset):
"""
Approve the selected users.
"""
profiles_updated = queryset.update(is_approved='Y')
if profiles_updated == 1:
msg = _("1 user was successfully approved.")
else:
msg = _("%(num)s users were successfully approved." % {'num': profiles_updated})
self.message_user(request, msg)
approve_users.short_description = _("Approve users")
def sponsor_users(self, request, queryset):
"""
Sponsor the selected users.
"""
profiles_updated = queryset.update(is_sponsored='Y')
if profiles_updated == 1:
msg = _("1 user was successfully sponsored.")
else:
msg = _("%(num)s users were successfully sponsored." % {'num': profiles_updated})
self.message_user(request, msg)
sponsor_users.short_description = _("Sponsor users")
def cancel_approve_users(self, request, queryset):
"""
Cancel the approval of the selected users.
"""
profiles_updated = queryset.update(is_approved='N')
if profiles_updated == 1:
msg = _("1 user was successfully cancelled approval.")
else:
msg = _("%(num)s users were successfully cancelled approval." % {'num': profiles_updated})
self.message_user(request, msg)
cancel_approve_users.short_description = _("Cancel approve users")
def cancel_sponsor_users(self, request, queryset):
"""
Cancel the sponsor of the selected users.
"""
profiles_updated = queryset.update(is_sponsored='N')
if profiles_updated == 1:
msg = _("1 user was successfully cancelled sponsor.")
else:
msg = _("%(num)s users were successfully cancelled sponsor." % {'num': profiles_updated})
self.message_user(request, msg)
cancel_sponsor_users.short_description = _("Cancel sponsor users")
def reset_approve_users(self, request, queryset):
"""
Reset the approval of the selected users.
"""
profiles_updated = queryset.update(is_approved='C')
if profiles_updated == 1:
msg = _("1 user was successfully reset approval.")
else:
msg = _("%(num)s users were successfully reset approval." % {'num': profiles_updated})
self.message_user(request, msg)
reset_approve_users.short_description = _("Reset approve users")
def reset_sponsor_users(self, request, queryset):
"""
Reset the sponsor of the selected users.
"""
profiles_updated = queryset.update(is_sponsored='C')
if profiles_updated == 1:
msg = _("1 user was successfully reset sponsor.")
else:
msg = _("%(num)s users were successfully reset sponsor." % {'num': profiles_updated})
self.message_user(request, msg)
reset_sponsor_users.short_description = _("Reset sponsor users")
# control 'is_checkin' field
def users_checkin(self, request, queryset):
"""
Mark selected users as checked in.
"""
profiles_updated = queryset.update(is_checkin='Y')
if profiles_updated == 1:
msg = _("1 user was successfully marked as checkin.")
else:
msg = _("%(num)s users were successfully marked as checkin." % {'num': profiles_updated})
self.message_user(request, msg)
users_checkin.short_description = _("Mark users as checked in")
def users_not_checkin(self, request, queryset):
"""
Mark selected users as NOT checked in.
"""
profiles_updated = queryset.update(is_checkin='N')
if profiles_updated == 1:
msg = _("1 user was marked as not checkin.")
else:
msg = _("%(num)s users were marked as not checkin." % {'num': profiles_updated})
self.message_user(request, msg)
users_not_checkin.short_description = _("Mark users as not checked in")
def users_checkin_reset(self, request, queryset):
"""
Reset checkin status of selected users.
"""
profiles_updated = queryset.update(is_checkin='X')
if profiles_updated == 1:
msg = _("1 user was reset status of checkin.")
else:
msg = _("%(num)s users were reset status of checkin." % {'num': profiles_updated})
self.message_user(request, msg)
users_checkin_reset.short_description = _("Reset users checkin status")
## custom fields
def email(self, obj):
"""
return the email of the user profile
"""
user = obj.user
return user.email
email.short_description = _("E-mail")
def transcript_url(self, obj):
"""
return the html code of transcript with url link
"""
transcript = obj.transcript
if transcript:
html = '<a href="%(url)s">%(name)s</a>' % {
'url': transcript.url,
'name': os.path.basename(transcript.name),
}
else:
html = _("Null")
return format_html(html)
transcript_url.short_description = _("Transcript")
def attachments(self, obj):
"""
return the html code of attachments with url
"""
user = obj.user
files = user.userfile_set.all()
if files:
attachments = ['<a href="%(url)s">%(name)s</a>' % {
'url': userfile.file.url,
'name': os.path.basename(userfile.file.name),
}
for userfile in files
]
html = '<br>'.join(attachments)
else:
html = _("Null")
return format_html(html)
attachments.short_description = _("Attachments")
## added a view to display all userprofile info (csv format)
def get_urls(self):
urls = super(type(self), self).get_urls()
my_urls = [
url(r'^csv/$', self.userprofile_csv_view,
name='userprofile_csv_view'),
]
return my_urls + urls
def userprofile_csv_view(self, request):
"""
custom admin view to display all userprofile info in csv format
"""
userprofile_objs = UserProfile.objects.all()
fieldnames = userprofile_objs[0].dump_fieldnames()
fields = [k for k,v in fieldnames.items()]
# # header
# header = u''
# for k in fields:
# header += u'{0},'.format(fieldnames[k])
# header += '\n'
# # contents
# contents = u''
# for obj in userprofile_objs:
# data = obj.dump()
# cline = u''
# for k in fields:
# if k == 'attachments':
# cline += u'{0},'.format('||'.join(data[k]))
# else:
# cline += u'{0},'.format(data[k])
# cline += '\n'
# contents += cline
# return HttpResponse(header+contents, content_type='text/plain')
filename = '{0}-{1}.csv'.format('registration', datetime.date.today().strftime('%Y%m%d'))
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="{0}"'.format(filename)
# Python 2's csv module does not support unicode I/O
#writer = csv.writer(response)
writer = UnicodeWriter(response)
# header
header = []
for k in fields:
header.append(fieldnames[k])
writer.writerow(header)
# contents
for obj in userprofile_objs:
data = obj.dump()
row = []
for k in fields:
if k == 'attachments':
row.append(';'.join(data[k]))
else:
row.append(unicode(data[k]))
writer.writerow(row)
#
return response
###
admin.site.register(UserProfile, UserProfileAdmin)
admin.site.register(UserFile)
|
