From 0cbba9d4365ccc362bd46b38b57ff422e5f714c4 Mon Sep 17 00:00:00 2001
From: arona <aj@aronajones.com>
Date: Fri, 19 May 2023 18:20:49 +0000
Subject: [PATCH] Return user to current page when clicking 'mark reviewed'

---
 RIGS/views/hs.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/RIGS/views/hs.py b/RIGS/views/hs.py
index a2ac9786..9bb0e324 100644
--- a/RIGS/views/hs.py
+++ b/RIGS/views/hs.py
@@ -29,15 +29,15 @@ class HSCreateView(generic.CreateView):
         return context
 
 
-class MarkReviewed(generic.View):
-    def get(self, *args, **kwargs):
+class MarkReviewed(generic.RedirectView):
+    def get_redirect_url(self, *args, **kwargs):
         obj = apps.get_model('RIGS', kwargs.get('model')).objects.get(pk=kwargs.get('pk'))
         with reversion.create_revision():
             reversion.set_user(self.request.user)
             obj.reviewed_by = self.request.user
             obj.reviewed_at = timezone.now()
             obj.save()
-        return HttpResponseRedirect(reverse('hs_list'))
+        return self.request.META.get('HTTP_REFERER', reverse('hs_list'))
 
 
 class EventRiskAssessmentCreate(HSCreateView):