Make pdf ticket files downloadable

Author: hhamalai

backend/views/routes.py

@@ -17,6 +17,7 @@
     'expiration_date': fields.DateTime(dt_format='iso8601'),
     'price': fields.Float,
     'qr': fields.String,
+    'pdf': fields.String,
     'order_id': fields.String,
     'ticket_type': fields.String,
     'reserved': fields.DateTime(dt_format='iso8601'),

backend/views/upload.py

@@ -22,7 +22,7 @@
 
 
 class UploadView(Resource):
-    def base64_encode_qrcode(self, qrcode):
+    def base64_encode(self, qrcode):
         try:
             with open(qrcode, "rb") as image_file:
                 return base64.b64encode(image_file.read())
@@ -58,7 +58,7 @@ def get_ticket_id(self, soup):
         ticket_id = re.search("- (.+) -", soup.text)
         if not ticket_id:
             abort(422)
-        return ticket_id.groups()[0]
+        return ticket_id.groups()[0].strip()
 
     def get_route(self, soup):
         route = re.search("\d+ / \d+(.+) - (.*)Aikuinen", soup.text)
@@ -94,43 +94,49 @@ def is_square(fn):
         if 'file' not in args:
             abort(422)
 
-        # Danger Zone, parsing PDF files is dirty and this one definately ain't
-        # the prettiest creature alive. Cleanse yourself with strong liqueur
-        # afterwards.
-
         tmp_dir = tempfile.mkdtemp()
         tmp_src_file = tmp_dir + '/in.pdf'
         with open(tmp_src_file, 'wb') as fd:
             fd.write(args['file'].read())
-
-        pages = self.get_pages(tmp_src_file)
-        if not pages:
-            return {"message": "Could not parse the uploaded file, is this actually the PDF file with tickets?"}, 422
 
-        first_page = pages[0]
-        ticket_count = mongo.db.tickets.find({"order_id": self.get_order_id(first_page)}).count()
+        qr_codes = []
+        pdf_files = []
 
-        qr_codes = iter([])
-        if ticket_count:
-            return {"message": "Ticket already uploaded"}, 400
         try:
+            # Danger Zone, parsing PDF files is dirty and this one definately ain't
+            # the prettiest creature alive. Cleanse yourself with strong liqueur
+            # afterwards.
             os.popen('pdfimages -png ' + tmp_src_file + ' ' + tmp_dir + '/out')
-            ims = os.popen("ls " + tmp_dir + "/out*").read().split()
-            qr_codes = iter([fn for fn in ims if is_square(fn)])
-            # qr_codes = iter(ims[1:len(ims):2])
+            ims = os.popen("ls " + tmp_dir + "/out*.png").read().split()
+            qr_codes = [fn for fn in ims if is_square(fn)]
+            os.popen("pdfseparate " + tmp_src_file + ' ' + tmp_dir + "/pdfout-%d.pdf")
+            pdfs = os.popen("ls " + tmp_dir + "/pdfout-*.pdf").read().split()
+            pdf_files = [fn for fn in pdfs]
         except:
             print("XXX: No poppler installed, unable to produce 2D barcodes")
 
+        if len(qr_codes) != len(pdf_files) != len(pages):
+            return {"message": "Invalid ticket file"}, 400
+
         tickets = []
-        for page in pages:
+        for qr_code, pdf_file in iter(zip(qr_codes, pdf_files)):
+            pages = self.get_pages(pdf_file)
+            if not pages:
+                return {"message": "Could not parse the uploaded file, is this actually the PDF file with tickets?"}, 422
+            page = pages[0]
+            ticket_count = mongo.db.tickets.find({"order_id": self.get_order_id(page)}).count()
+            if ticket_count:
+                return {"message": "Ticket already uploaded"}, 400
             route = self.get_route(page)
             ticket_type, expires = self.get_ticket_type_and_expiration(page)
-            qr_base64 = self.base64_encode_qrcode(qr_codes.next())
+            qr_base64 = self.base64_encode(qr_code)
+            pdf_base64 = self.base64_encode(pdf_file)
             tickets.append({
                 'ticket_id': uuid.uuid4().hex,
                 'src': route[0].upper(),
                 'dest': route[1].upper(),
                 'qr': 'data:image/png;base64,' + qr_base64,
+                'pdf': 'data:application/pdf;base64,' + pdf_base64,
                 'order_id': self.get_order_id(page),
                 'price': self.get_price(page) / len(pages),
                 'ticket_type': ticket_type,

docker-compose.yml

@@ -8,11 +8,15 @@ services:
         - "27017"
 
     web:
-      build: .
+      build:
+        context: .
+        args:
+          CONFIG_FILE: config/app-docker.cfg
       restart: always
       volumes:
         - ./config:/webapps/sarjalipputasku/config:ro
-        - ./backend:/webapps/sarjalipputasku/backend:ro
+        - ./backend:/webapps/sarjalipputasku/source:ro
+        - ./frontend:/webapps/sarjalipputasku/static:ro
       expose:
         - "8000"
       depends_on:

docker-entrypoint.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
 export OAUTHLIB_INSECURE_TRANSPORT=1
 export OAUTHLIB_RELAX_TOKEN_SCOPE=1
-/usr/local/bin/gunicorn -w 2 -b 0.0.0.0:8000 server:app
+export ENV SARJALIPPUTASKU_DIR=/webapps/sarjalipputasku/config/app.cfg
+/usr/local/bin/gunicorn -w 2 -b 0.0.0.0:8000 --reload server:app

frontend/src/components/MyTickets.vue

@@ -8,7 +8,9 @@
         <span>{{ ticket.src }} - {{ ticket.dest }}</span><br />
         Varattu: <span>{{ formatDate(ticket.reserved) }}</span><br />
         Hinta: <span>{{ ticket.price }} &euro; (alv 0%)</span><br />
-        <span>{{ ticket.order_id }} - {{ ticket.vr_id }}</span><br />
+        <span>{{ ticket.order_id }} - {{ ticket.vr_id }} - 
+          <a download="biljet.pdf" v-bind:href="ticket.pdf" title="Download pdf">pdf</a>
+        </span><br />
       </div>
       <div class="col-md-2">
         <button class="btn btn-success" v-on:click="useTicket(ticket.id)" v-if="!ticket.used">Käytä</button>

frontend/src/components/ticket-item.vue

@@ -4,7 +4,9 @@
     <span>{{ ticket.src }} - {{ ticket.dest }} {{ ticket.ticket_type }}</span><br />
     Voimassa: <span>{{ formatDate(ticket.expiration_date) }}</span><br />
     Hinta: <span>{{ ticket.price }} &euro; (alv 0%)</span><br />
-    <span>{{ ticket.vr_id }}</span><br />
+    <span>{{ ticket.vr_id }}</span>
+    <a download="biljet.pdf" v-bind:href="ticket.pdf" title="Download pdf">pdf</a>
+    <br />
     <p>
       <img v-bind:src="ticket.qr" />
     </p>