gistfile1.py

from os import unlink
from uuid import uuid3, NAMESPACE_DNS
from task_queue.app import app
from utils import upload, pdf, html
from model import Session, AgencyPdf


def upload_pdf(pdf_file):
    file_name = str(uuid3(NAMESPACE_DNS, 'takeoffgo.com')) + '.pdf'
    path = 'pdf/' + file_name

    return upload(pdf_file, path)


@app.task
def generate_pdf(id):
    db = Session()
    req = db.query(AgencyPdf).get(id)
    html_file = html.from_template(req.agencytemplate.content, req.data)
    pdf_file = pdf.from_html(html_file)

    url = upload_pdf(pdf_file)

    unlink(html_file)
    unlink(pdf_file)

    req.url = url

    db.commit()

    return url