Merge pull request #34 from mrunderline/feat/ExportCSV

feat: add ExportCsvMixin class to export CSV

course/admin.py

@@ -1,11 +1,12 @@
 from django.contrib import admin
 
 from .models import Course
+from easyschool.utils import ExportCsvMixin
 
 
 # Register your models here.
 @admin.register(Course)
-class CourseAdmin(admin.ModelAdmin):
+class CourseAdmin(admin.ModelAdmin, ExportCsvMixin):
     """Admin Display Options to display the Course data in admin panel"""
 
     list_display = (
@@ -16,3 +17,7 @@ class CourseAdmin(admin.ModelAdmin):
     search_fields = (
         'course_name',
     )
+
+    actions = [
+        'export_as_csv',
+    ]

easyschool/utils.py

@@ -1,4 +1,6 @@
 from datetime import date
+import csv
+from django.http import HttpResponse
 
 GENDER_CHOICES = (
     ('M', 'Male'),
@@ -25,4 +27,22 @@ def next_month():
     month = date.today().month + 1
     year = date.today().year
     day = 1
-    return date(year, month, day)
+    return date(year, month, day)
+
+
+class ExportCsvMixin:
+    def export_as_csv(self, request, queryset):
+        meta = self.model._meta
+        field_names = [field.name for field in meta.fields]
+
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = 'attachment; filename={}.csv'.format(meta)
+        writer = csv.writer(response)
+
+        writer.writerow(field_names)
+        for obj in queryset:
+            writer.writerow([getattr(obj, field) for field in field_names])
+
+        return response
+
+    export_as_csv.short_description = 'Export Selected'

students/admin.py

@@ -6,6 +6,7 @@
 from django.utils.html import format_html, mark_safe
 
 from .models import Guardian, Student, StudentFee
+from easyschool.utils import ExportCsvMixin
 
 # Register your models here.
 
@@ -20,7 +21,7 @@ class GuardianInline(admin.TabularInline):
 
 
 @admin.register(Student)
-class StudentAdmin(admin.ModelAdmin):
+class StudentAdmin(admin.ModelAdmin, ExportCsvMixin):
     """ Display Class of Student Model in Admin Panel"""
 
     # remove delete actions
@@ -130,6 +131,7 @@ def profile_image_display(self, obj):
     ]
     actions = [
         'expel_from_school',
+        'export_as_csv',
     ]
 
     def get_readonly_fields(self, request, obj=None):
@@ -147,7 +149,7 @@ def get_readonly_fields(self, request, obj=None):
 
 
 @admin.register(StudentFee)
-class StudentFeeAdmin(admin.ModelAdmin):
+class StudentFeeAdmin(admin.ModelAdmin, ExportCsvMixin):
     """ Admin display class for the model StudentFee """
 
     # Delete Delete Action from admin
@@ -175,14 +177,22 @@ def get_actions(self, request):
         'date_submitted',
     )
 
+    actions = [
+        'export_as_csv',
+    ]
+
 
 @admin.register(Guardian)
-class GuardianAdmin(admin.ModelAdmin):
+class GuardianAdmin(admin.ModelAdmin, ExportCsvMixin):
     list_display = (
         'name',
         'gender',
         'relation_to_student',
         'social_security_number',
         'phone_number',
         'profession',
-    )
+    )
+
+    actions = [
+        'export_as_csv',
+    ]

teachers/admin.py

@@ -6,6 +6,7 @@
 from django.utils.translation import ugettext_lazy as _
 
 from .models import Teacher, TeacherSalary
+from easyschool.utils import ExportCsvMixin
 
 
 # Change User Creation Form
@@ -35,7 +36,7 @@ def __init__(self, *args, **kwargs):
 # Register your models here.
 
 @admin.register(Teacher)
-class TeacherAdmin(admin.ModelAdmin):
+class TeacherAdmin(admin.ModelAdmin, ExportCsvMixin):
     '''Admin View for Teacher'''
 
     list_display = (
@@ -59,9 +60,13 @@ class TeacherAdmin(admin.ModelAdmin):
 
     ordering = ('-date_of_joining',)
 
+    actions = [
+        'export_as_csv',
+    ]
+
 
 @admin.register(TeacherSalary)
-class TeacherSalaryAdmin(admin.ModelAdmin):
+class TeacherSalaryAdmin(admin.ModelAdmin, ExportCsvMixin):
     """ Admin display class for the model StudentFee """
 
     # Delete Delete Action from admin
@@ -85,4 +90,8 @@ def get_actions(self, request):
         'valid_until',
         'total_amount',
         'paid_on',
-    )
+    )
+
+    actions = [
+        'export_as_csv',
+    ]