From c08d21db8d1e5664fca98965e5767f692b89122b Mon Sep 17 00:00:00 2001
From: Joe Banks <joe@jb3.dev>
Date: Sun, 13 Oct 2024 00:26:12 +0100
Subject: [PATCH] Add new routes for print export for power tests

---
 RIGS/urls.py     | 1 +
 RIGS/views/hs.py | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/RIGS/urls.py b/RIGS/urls.py
index 8793b741..5c263c48 100644
--- a/RIGS/urls.py
+++ b/RIGS/urls.py
@@ -100,6 +100,7 @@ urlpatterns = [
          name='pt_edit'),
     path('event/power/<int:pk>/review/', permission_required_with_403('RIGS.review_power')(views.MarkReviewed.as_view()),
          name='pt_review', kwargs={'model': 'PowerTestRecord'}),
+    path('event/power/<int:pk>/print/', permission_required_with_403('RIGS.view_powertestrecord')(views.PowerPrint.as_view()), name='pt_print'),
 
     path('event/<int:pk>/checkin/', login_required(views.EventCheckIn.as_view()),
          name='event_checkin'),
diff --git a/RIGS/views/hs.py b/RIGS/views/hs.py
index fd97dfd7..cf67a826 100644
--- a/RIGS/views/hs.py
+++ b/RIGS/views/hs.py
@@ -231,6 +231,14 @@ class RAPrint(PrintView):
         context['filename'] = f"EventSpecificRiskAssessment_for_{context['object'].event.display_id}.pdf"
         return context
 
+class PowerPrint(PrintView):
+    model = models.PowerTestRecord
+    template_name = 'hs/power_print.xml'
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['filename'] = f"PowerTestRecord_for_{context['object'].event.display_id}.pdf"
+        return context
 
 class EventCheckIn(generic.CreateView, ModalURLMixin):
     model = models.EventCheckIn