FEAT: Add ability to generate RA printouts

RIGS/views/hs.py

@@ -12,7 +12,7 @@ from PyRIGS.views import PrintView
 
 class EventRiskAssessmentCreate(generic.CreateView):
     model = models.RiskAssessment
-    template_name = 'risk_assessment_form.html'
+    template_name = 'hs/risk_assessment_form.html'
     form_class = forms.EventRiskAssessmentForm
 
     def get(self, *args, **kwargs):
@@ -49,7 +49,7 @@ class EventRiskAssessmentCreate(generic.CreateView):
 
 class EventRiskAssessmentEdit(generic.UpdateView):
     model = models.RiskAssessment
-    template_name = 'risk_assessment_form.html'
+    template_name = 'hs/risk_assessment_form.html'
     form_class = forms.EventRiskAssessmentForm
 
     def get_success_url(self):
@@ -72,7 +72,7 @@ class EventRiskAssessmentEdit(generic.UpdateView):
 
 class EventRiskAssessmentDetail(generic.DetailView):
     model = models.RiskAssessment
-    template_name = 'risk_assessment_detail.html'
+    template_name = 'hs/risk_assessment_detail.html'
 
     def get_context_data(self, **kwargs):
         context = super(EventRiskAssessmentDetail, self).get_context_data(**kwargs)
@@ -83,7 +83,7 @@ class EventRiskAssessmentDetail(generic.DetailView):
 class EventRiskAssessmentList(generic.ListView):
     paginate_by = 20
     model = models.RiskAssessment
-    template_name = 'hs_object_list.html'
+    template_name = 'hs/hs_object_list.html'
 
     def get_queryset(self):
         return self.model.objects.exclude(event__status=models.Event.CANCELLED).order_by('reviewed_at').select_related('event')
@@ -112,7 +112,7 @@ class EventRiskAssessmentReview(generic.View):
 
 class EventChecklistDetail(generic.DetailView):
     model = models.EventChecklist
-    template_name = 'event_checklist_detail.html'
+    template_name = 'hs/event_checklist_detail.html'
 
     def get_context_data(self, **kwargs):
         context = super(EventChecklistDetail, self).get_context_data(**kwargs)
@@ -122,7 +122,7 @@ class EventChecklistDetail(generic.DetailView):
 
 class EventChecklistEdit(generic.UpdateView):
     model = models.EventChecklist
-    template_name = 'event_checklist_form.html'
+    template_name = 'hs/event_checklist_form.html'
     form_class = forms.EventChecklistForm
 
     def get_success_url(self):
@@ -145,7 +145,7 @@ class EventChecklistEdit(generic.UpdateView):
 
 class EventChecklistCreate(generic.CreateView):
     model = models.EventChecklist
-    template_name = 'event_checklist_form.html'
+    template_name = 'hs/event_checklist_form.html'
     form_class = forms.EventChecklistForm
 
     # From both business logic and programming POVs, RAs must exist before ECs!
@@ -184,7 +184,7 @@ class EventChecklistCreate(generic.CreateView):
 class EventChecklistList(generic.ListView):
     paginate_by = 20
     model = models.EventChecklist
-    template_name = 'hs_object_list.html'
+    template_name = 'hs/hs_object_list.html'
 
     def get_queryset(self):
         return self.model.objects.exclude(event__status=models.Event.CANCELLED).order_by('reviewed_at').select_related('event')
@@ -214,7 +214,7 @@ class EventChecklistReview(generic.View):
 class HSList(generic.ListView):
     paginate_by = 20
     model = models.Event
-    template_name = 'hs_list.html'
+    template_name = 'hs/hs_list.html'
 
     def get_queryset(self):
         return models.Event.objects.all().exclude(status=models.Event.CANCELLED).order_by('-start_date').select_related('riskassessment').prefetch_related('checklists')
@@ -227,3 +227,9 @@ class HSList(generic.ListView):
 
 class RAPrint(PrintView):
     model = models.RiskAssessment
+    template_name = 'hs/ra_print.xml'
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['filename'] = f"EventSpecificRiskAssessment_for_{context['object'].event.display_id}.pdf"
+        return context

RIGS/views/rigboard.py

@@ -1,14 +1,9 @@
 import copy
 import datetime
 import re
-import urllib.error
-import urllib.parse
-import urllib.request
-from io import BytesIO
-
 import premailer
 import simplejson
-from PyPDF2 import PdfFileMerger, PdfFileReader
+
 from django.conf import settings
 from django.contrib import messages
 from django.contrib.staticfiles import finders
@@ -24,10 +19,9 @@ from django.urls import reverse_lazy
 from django.utils import timezone
 from django.utils.decorators import method_decorator
 from django.views import generic
-from z3c.rml import rml2pdf
 
 from PyRIGS import decorators
-from PyRIGS.views import OEmbedView, is_ajax, ModalURLMixin
+from PyRIGS.views import OEmbedView, is_ajax, ModalURLMixin, PrintView
 from RIGS import models, forms
 
 __author__ = 'ghost'
@@ -178,42 +172,16 @@ class EventDuplicate(EventUpdate):
         return context
 
 
-class EventPrint(generic.View):
-    def get(self, request, pk):
-        object = get_object_or_404(models.Event, pk=pk)
-        template = get_template('event_print.xml')
+class EventPrint(PrintView):
+    model = models.Event
+    template_name = 'event_print.xml'
+    append_terms = True
 
-        merger = PdfFileMerger()
-
-        user_str = f"by {request.user.name} " if request.user is not None else ""
-        time = timezone.now().strftime('%d/%m/%Y %H:%I')
-
-        name = re.sub(r'[^a-zA-Z0-9 \n\.]', '', object.name)
-        filename = f"Event_{object.display_id}_{name}_{object.start_date}.pdf"
-
-        context = {
-            'object': object,
-            'quote': True,
-            'current_user': request.user,
-            'filename': filename,
-            'info_string': f"[Paperwork generated {user_str}on {time} - {object.current_version_id}]",
-        }
-
-        rml = template.render(context)
-        buffer = rml2pdf.parseString(rml)
-        merger.append(PdfFileReader(buffer))
-        buffer.close()
-
-        terms = urllib.request.urlopen(settings.TERMS_OF_HIRE_URL)
-        merger.append(BytesIO(terms.read()))
-
-        merged = BytesIO()
-        merger.write(merged)
-
-        response = HttpResponse(content_type='application/pdf')
-        response['Content-Disposition'] = f'filename="{filename}"'
-        response.write(merged.getvalue())
-        return response
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['quote'] = True
+        context['filename'] = f"Event_{context['object'].display_id}_{context['object_name']}_{context['object'].start_date}.pdf"
+        return context
 
 
 class EventArchive(generic.ListView):
@@ -223,7 +191,6 @@ class EventArchive(generic.ListView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-
         context['start'] = self.request.GET.get('start', None)
         context['end'] = self.request.GET.get('end', datetime.date.today().strftime('%Y-%m-%d'))
         context['statuses'] = models.Event.EVENT_STATUS_CHOICES