generate.py

# -*- encoding: utf-8 -*-

'''
HACKWAW badge generator

It's just a very simple pdf generator. It takes csv file column and put each row text, formats it and inserts into the pdf.
Obviously it will generate a lot of pdfs if you have a lot of rows in a csv file(it takes seconds to generate hundreds, so..)
You can use to generate any type of pdf files which requires embedding different text in each file.

Just put your template pdf file, font file and csv file in the folder with the script and configure settings below

When you're done, just do:
python generate.py

'''

from conf import *

import StringIO
import csv

from codegen import fetch_code
from reportlab.platypus import Paragraph
from reportlab.pdfgen import canvas
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.lib.styles import getSampleStyleSheet, TA_CENTER
from pyPdf import PdfFileWriter, PdfFileReader

# fonts import
pdfmetrics.registerFont(TTFont('ThinFont', FONT_NAME[0]))
pdfmetrics.registerFont(TTFont('RegularFont', FONT_NAME[1]))

# data import
data = csv.reader(open(CSV_FILE))

# read the column names from the first line of the file
fields = data.next()

for row in data:
    item = dict(zip(fields, row))

    # importing 'base' file on which we write stuff
    input = PdfFileReader(file(PDF_TEMPLATE, "rb"))

    packet = StringIO.StringIO()
    c = canvas.Canvas(packet, pagesize=PDF_SIZE)
    width, height = PDF_SIZE

    # word number 1
    stylesheet=getSampleStyleSheet()
    styleN = stylesheet['Normal']
    styleN.alignment = TA_CENTER
    styleN.fontSize = FONT_SIZE
    styleN.fontName = 'RegularFont'
    # changing the size of the font depending on the lenght of the word

    p = Paragraph(u'<font color="black">' + item['Ticket First Name'].decode('utf-8')+u'</font>', styleN)
    # width/height of the paragraph
    w,h = p.wrap(width, height)
    # x/y axis when the paragraph should be added
    p.drawOn(c, 0, height-HPOSITION)

    # word number 2
    stylesheet=getSampleStyleSheet()
    styleN = stylesheet['Normal']
    styleN.alignment = TA_CENTER
    styleN.fontSize = FONT_SIZE
    styleN.fontName = 'ThinFont'
    # changing the size of the font depending on the lenght of the word

    p = Paragraph(u'<font color="black">' + item['Ticket Last Name'].decode('utf-8')+u'</font>', styleN)
    # width/height of the paragraph
    w,h = p.wrap(width, height)
    # x/y axis when the paragraph should be added
    p.drawOn(c, 0, height-HPOSITION-30)


    # saving canvas
    c.save()
    packet.seek(0)
    text = PdfFileReader(packet)
    output = PdfFileWriter()
    # merging text with base
    page = input.getPage(0)
    page.mergePage(text.getPage(0))
    output.addPage(page)

    # saving file
    if str(item[COLUMN]) != "" or GENERATE_EMPTY:
        filename = item[COLUMN].replace(" ", "")[0:FILENAME_LENGTH] + fetch_code() + ".pdf"
        outputStream = file("output/" + filename, "wb")
        output.write(outputStream)
        outputStream.close()