views.py 文件源码

def merge_pdf(request, technology_id):

    # generate_quality_pdf(request, 1)
    technology = Technology.objects.get(id=technology_id)
    operation_groups = technology.operation_groups.all()

    data = technology_pdf_data(technology_id)
    merger = PdfFileMerger()
    file_name = 'technology.pdf'
    with open(file_name, 'w') as mypdf:
        mypdf.write(data)
    merger.append(PdfFileReader(file(file_name, 'rb')))
    os.remove(file_name)

    for operation_group in operation_groups:
        file_name = 'operation_group_%d.pdf' %(operation_group.id)
        with open(file_name, 'w') as mypdf:
            data2= technology_subpicture_pdf_data(operation_group.id)
            mypdf.write(data2)
        merger.append(PdfFileReader(file(file_name, 'rb')))
        os.remove(file_name)

    output_name = "%s.pdf" %(technology.code)
    merger.write(str(output_name))

    response = HttpResponse(mimetype='application/pdf')
    with open(output_name, 'rb') as mypdf:
        response.write(mypdf.read())

    os.remove(output_name)

    response['Content-Disposition'] = 'attachment; filename=%s' %(output_name)
    return response