Generate PDF with Python using PhantomJS and Jinga

python_phantom_jinga_pdf.py

#!/usr/bin/env python

'''
Created on 22 Sep 2016

@author: phegde
'''

from subprocess import call
#from string import Template
from jinja2 import Environment, FileSystemLoader
import codecs
import json
import os
        
def generate_pdf(dataJson, templateName):
    try:
        
        #jinja template engine
        env = Environment(loader=FileSystemLoader('templates'))
        template = env.get_template(templateName)
        substitutedTemplate = template.render(dataJson)
        
        #file names
        tempHTML = dataJson.get('fullName')+' - '+templateName+'-temp.html'
        pdfFileName = dataJson.get('fullName')+' - '+templateName
        
        FILE = open(os.path.join('./temp', tempHTML), "wb")
        FILE.write(substitutedTemplate)
        FILE.close()
        
        #generate pdf
        call(["phantomjs", "rasterize.js", './temp/'+tempHTML, pdfFileName+'.pdf'])
        
        #dispatch the file
        
        #delete the temp pdf
        os.remove('./temp/'+tempHTML)
        
    except Exception as e:
        print e
        
if __name__ == "__main__":
    
    #read dummy json    
    profileData =  open('profile.json')
    data = json.load(profileData)
    dataJson = data.get('userProfile')
    templateName = 'profile_snapshot.html'
    
    generate_pdf(dataJson,templateName)