karmaworld/apps/users/admin.py

   1 #!/usr/bin/env python
   2 # -*- coding:utf8 -*-
   3 # Copyright (C) 2013  FinalsClub Foundation
   4 from django.contrib import admin
   5 import csv
   6 import StringIO
   7 from django.contrib.admin import ModelAdmin
   8 from django.http import HttpResponse
   9 from karmaworld.apps.users.models import UserProfile, NoteKarmaEvent, CourseKarmaEvent, GenericKarmaEvent
  10
  11
  12 def user_export(modeladmin, request, queryset):
  13     f = StringIO.StringIO()
  14     writer = csv.writer(f, dialect='excel')
  15     writer.writerow(['First Name', 'Last Name', 'Email'])
  16     for user_profile in queryset:
  17         user = user_profile.user
  18         writer.writerow([user.first_name, user.last_name, user.email])
  19     return HttpResponse(f.getvalue())
  20 user_export.short_description = "Download a CSV file with users' information"
  21
  22
  23 class UserProfileAdmin(ModelAdmin):
  24     actions = (user_export,)
  25
  26 admin.site.register(UserProfile, UserProfileAdmin)
  27 admin.site.register(GenericKarmaEvent)
  28 admin.site.register(NoteKarmaEvent)
  29 admin.site.register(CourseKarmaEvent)