DREF revisions via frontend calls

Author: szabozoltan69

dref/admin.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+from reversion_compare.admin import CompareVersionAdmin
 
 from lang.admin import TranslationAdmin
 from .models import (
@@ -15,7 +16,7 @@ class DrefFileAdmin(admin.ModelAdmin):
 
 
 @admin.register(Dref)
-class DrefAdmin(TranslationAdmin, admin.ModelAdmin):
+class DrefAdmin(CompareVersionAdmin, TranslationAdmin, admin.ModelAdmin):
     search_fields = ('title',)
     list_display = ('title', 'national_society', 'disaster_type',
                     'ns_request_date', 'submission_to_geneva', 'status',)
@@ -39,7 +40,7 @@ def get_queryset(self, request):
 
 
 @admin.register(DrefOperationalUpdate)
-class DrefOperationalUpdateAdmin(admin.ModelAdmin):
+class DrefOperationalUpdateAdmin(CompareVersionAdmin, admin.ModelAdmin):
     list_display = ('title', 'national_society', 'disaster_type')
     autocomplete_fields = (
         'national_society',
@@ -62,7 +63,7 @@ def get_queryset(self, request):
         )
 
 @admin.register(DrefFinalReport)
-class DrefFinalReportAdmin(admin.ModelAdmin):
+class DrefFinalReportAdmin(CompareVersionAdmin, admin.ModelAdmin):
     list_display = ('title', 'national_society', 'disaster_type')
     autocomplete_fields = (
         'national_society',

dref/models.py

@@ -658,6 +658,9 @@ def save(self, *args, **kwargs):
         self.__budget_file_id = self.budget_file_id
         super().save(*args, **kwargs)
 
+    def __str__(self):
+        return f'{self.title} – {self.status}'
+
 
 class DrefFile(models.Model):
     file = models.FileField(

dref/views.py

@@ -1,6 +1,7 @@
 from django.db import models
 from django.contrib.auth.models import User
 from django.utils.translation import gettext
+from reversion.views import RevisionMixin
 
 from rest_framework import (
     views,
@@ -34,7 +35,7 @@
 from dref.permissions import DrefOperationalUpdateCreatePermission
 
 
-class DrefViewSet(viewsets.ModelViewSet):
+class DrefViewSet(RevisionMixin, viewsets.ModelViewSet):
     serializer_class = DrefSerializer
     permission_classes = [permissions.IsAuthenticated]
     filterset_class = DrefFilter
@@ -68,7 +69,7 @@ def get_published(self, request, pk=None, version=None):
         return response.Response(serializer.data)
 
 
-class DrefOperationalUpdateViewSet(viewsets.ModelViewSet):
+class DrefOperationalUpdateViewSet(RevisionMixin, viewsets.ModelViewSet):
     serializer_class = DrefOperationalUpdateSerializer
     permission_classes = [permissions.IsAuthenticated, DrefOperationalUpdateCreatePermission]
     filterset_class = DrefOperationalUpdateFilter
@@ -113,7 +114,7 @@ def get_published(self, request, pk=None, version=None):
         return response.Response(serializer.data)
 
 
-class DrefFinalReportViewSet(viewsets.ModelViewSet):
+class DrefFinalReportViewSet(RevisionMixin, viewsets.ModelViewSet):
     serializer_class = DrefFinalReportSerializer
     permission_classes = [permissions.IsAuthenticated]
 
@@ -148,7 +149,7 @@ def get_published(self, request, pk=None, version=None):
 
 class DrefOptionsView(views.APIView):
     """
-    Options for various attrivute related to Dref
+    Options for various attribute related to Dref
     """
     permission_classes = [permissions.IsAuthenticated]