Merge pull request #144 from swiftss-org/csv-download-admin

CSV export in admin and remove permissions for only admins

config/settings/base.py

@@ -68,6 +68,7 @@
     "rest_framework",
     "rest_framework.authtoken",
     "corsheaders",
+    "import_export"
 ]
 
 LOCAL_APPS = [
@@ -266,8 +267,7 @@
         "rest_framework.authentication.BasicAuthentication",
     ),
     "DEFAULT_PERMISSION_CLASSES": (
-        "rest_framework.permissions.IsAdminUser",
-        "tmh_registry.users.api.permissions.IsMedicalPersonnel"
+        "tmh_registry.users.api.permissions.IsMedicalPersonnel",
     ),
     'EXCEPTION_HANDLER': 'tmh_registry.common.error_handling.error_handler',
     'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.PageNumberPagination',

config/settings/local.py

@@ -13,16 +13,6 @@
 # https://docs.djangoproject.com/en/dev/ref/settings/#allowed-hosts
 ALLOWED_HOSTS = ["localhost", "0.0.0.0", "127.0.0.1"]  # nosec
 
-# CACHES
-# ------------------------------------------------------------------------------
-# https://docs.djangoproject.com/en/dev/ref/settings/#caches
-CACHES = {
-    "default": {
-        "BACKEND": "django.core.cache.backends.locmem.LocMemCache",
-        "LOCATION": "",
-    }
-}
-
 # EMAIL
 # ------------------------------------------------------------------------------
 # https://docs.djangoproject.com/en/dev/ref/settings/#email-backend

config/settings/production.py

@@ -149,8 +149,7 @@
         "rest_framework.authentication.TokenAuthentication",
     ),
     "DEFAULT_PERMISSION_CLASSES": (
-        "rest_framework.permissions.IsAdminUser",
-        "tmh_registry.users.api.permissions.IsMedicalPersonnel"
+        "tmh_registry.users.api.permissions.IsMedicalPersonnel",
     ),
     'EXCEPTION_HANDLER': 'tmh_registry.common.error_handling.error_handler',
     'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.LimitOffsetPagination',

requirements/base.txt

@@ -19,6 +19,7 @@ django-cors-headers==3.7.0 # https://github.com/adamchainz/django-cors-headers
 django-csp==3.7  # https://github.com/mozilla/django-csp
 whitenoise==5.2.0 # https://github.com/evansd/whitenoise
 django-filter==2.4.0
+django-import-export
 
 # Django REST Framework
 djangorestframework==3.12.2  # https://github.com/encode/django-rest-framework

tmh_registry/registry/admin.py

@@ -1,4 +1,9 @@
 from django.contrib import admin
+from import_export.admin import ExportMixin
+from import_export import resources
+from django.contrib.auth.admin import UserAdmin
+from django.contrib.auth.models import User
+
 
 from tmh_registry.registry.models import (
     Episode,
@@ -7,29 +12,46 @@
     PatientHospitalMapping, Discharge, FollowUp,
 )
 
+
 @admin.register(Discharge)
-class DischargeAdmin(admin.ModelAdmin):
+class DischargeAdmin(ExportMixin, admin.ModelAdmin):
     model = Discharge
 
 @admin.register(FollowUp)
-class FollowUpAdmin(admin.ModelAdmin):
+class FollowUpAdmin(ExportMixin, admin.ModelAdmin):
     model = FollowUp
 
 @admin.register(Hospital)
-class HospitalAdmin(admin.ModelAdmin):
+class HospitalAdmin(ExportMixin, admin.ModelAdmin):
     model = Hospital
 
 
 @admin.register(Patient)
-class PatientAdmin(admin.ModelAdmin):
+class PatientAdmin(ExportMixin, admin.ModelAdmin):
     model = Patient
 
 
 @admin.register(PatientHospitalMapping)
-class PatientHospitalMappingAdmin(admin.ModelAdmin):
+class PatientHospitalMappingAdmin(ExportMixin, admin.ModelAdmin):
     model = PatientHospitalMapping
 
 
 @admin.register(Episode)
-class EpisodeAdmin(admin.ModelAdmin):
+class EpisodeAdmin(ExportMixin, admin.ModelAdmin):
     model = Episode
+
+
+"""
+To allow CSV export on User model
+"""
+class UserResource(resources.ModelResource):
+    class Meta:
+        model = User
+        fields = ('first_name', 'last_name', 'email')
+
+class UserAdmin(ExportMixin, UserAdmin):
+    resource_class = UserResource
+    pass
+
+admin.site.unregister(User)
+admin.site.register(User, UserAdmin)

tmh_registry/users/admin.py

@@ -1,8 +1,10 @@
 from django.contrib import admin
+from import_export.admin import ExportMixin
+
 
 from tmh_registry.users.models import MedicalPersonnel
 
 
 @admin.register(MedicalPersonnel)
-class MedicalPersonnelAdmin(admin.ModelAdmin):
+class MedicalPersonnelAdmin(ExportMixin ,admin.ModelAdmin):
     model = MedicalPersonnel