arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-28 02:42:17 +00:00
Code Issues Projects Releases Wiki Activity

Return user to current page when clicking 'mark reviewed'

Browse Source
This commit is contained in:
FreneticScribbler
2023-05-19 18:20:49 +00:00
parent d47322a959
commit 0cbba9d436
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
RIGS/views/hs.py
View File

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