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
|
from django.contrib import admin
import csv
from django.http import HttpResponse
from .models import (Category, Profile, User, Animation, Comment,
AnimationStats)
# Register your models here.
try:
from StringIO import StringIO as string_io
except ImportError:
from io import BytesIO as string_io
# Custom Classes
class ProfileAdmin(admin.ModelAdmin):
list_display = ['title','user', 'institute','location','department',
'phone_number','position']
list_filter = ['position', 'department']
actions = ['download_csv']
def download_csv(self, request, queryset):
openfile = string_io()
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment;\
filename=profile_data.csv'
writer = csv.writer(response)
writer.writerow(['email_id', 'title','username', 'first_name', 'last_name',
'institute', 'location', 'department',
'phone_number', 'position'])
for q in queryset:
writer.writerow([q.user.email, q.title, q.user, q.user.first_name,
q.user.last_name, q.institute,
q.location, q.department, q.phone_number,
q.position])
openfile.seek(0)
response.write(openfile.read())
return response
download_csv.short_description = "Download CSV file for selected stats."
class CategoryAdmin(admin.ModelAdmin):
list_display = ['name', 'created', 'description']
list_filter = ['name']
class AnimationAdmin(admin.ModelAdmin):
list_display = ['title', 'status', 'contributor']
list_filter = ['category']
actions = ['download_data']
def download_data(self, request, queryset):
openfile = string_io()
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment;\
filename=animations_data.csv'
writer = csv.writer(response)
writer.writerow(['title', 'contributor', 'reviewer', 'description',
'status', 'github link', 'category'])
for q in queryset:
writer.writerow([q.title, q.contributor.get_full_name(),
q.reviewer.get_full_name(), q.description, q.status, q.github,
q.category])
openfile.seek(0)
response.write(openfile.read())
return response
download_data.short_description = "Download data CSV file."
admin.site.register(Category, CategoryAdmin)
admin.site.register(Profile, ProfileAdmin)
admin.site.register(Animation, AnimationAdmin)
|
