Rig paperwork printing enabled

RIGS/rigboard.py

@@ -1,9 +1,20 @@
+import os
+import cStringIO as StringIO
+from io import BytesIO
+import urllib2
+
 from django.views import generic
 from django.core.urlresolvers import reverse_lazy
+from django.shortcuts import get_object_or_404
+from django.template import Context, RequestContext
+from django.template.loader import get_template
+from django.conf import settings
+from django.http import HttpResponse
+from z3c.rml import rml2pdf
+from PyPDF2 import PdfFileMerger, PdfFileReader
 
 from RIGS import models, forms
 
-
 __author__ = 'ghost'
 
 
@@ -35,6 +46,7 @@ class EventCreate(generic.CreateView):
     def get_success_url(self):
         return reverse_lazy('event_detail', kwargs={'pk': self.object.pk})
 
+
 class EventUpdate(generic.UpdateView):
     model = models.Event
     form_class = forms.EventForm
@@ -46,3 +58,43 @@ class EventUpdate(generic.UpdateView):
 
     def get_success_url(self):
         return reverse_lazy('event_detail', kwargs={'pk': self.object.pk})
+
+
+class EventPrint(generic.View):
+    def get(self, request, pk):
+        object = get_object_or_404(models.Event, pk=pk)
+        template = get_template('RIGS/event_print.xml')
+        copies = ('TEC', 'Client')
+        context = RequestContext(request, {
+            'object': object,
+            'fonts': {
+                'opensans': {
+                    'regular': 'RIGS/static/fonts/OPENSANS-REGULAR.TTF',
+                    'bold': 'RIGS/static/fonts/OPENSANS-BOLD.TTF',
+                }
+            },
+        })
+
+        merger = PdfFileMerger()
+
+        for copy in copies:
+            context['copy'] = copy
+            rml = template.render(context)
+            buffer = StringIO.StringIO()
+
+            buffer = rml2pdf.parseString(rml)
+
+            merger.append(PdfFileReader(buffer))
+
+            buffer.close()
+
+        terms = urllib2.urlopen(settings.TERMS_OF_HIRE_URL)
+        merger.append(StringIO.StringIO(terms.read()))
+
+        merged = BytesIO()
+        merger.write(merged)
+
+        response = HttpResponse(content_type='application/pdf')
+        response['Content-Disposition'] = "filename=N%05d | %s.pdf" % (object.pk, object.name)
+        response.write(merged.getvalue())
+        return response