wkhtmltopdf_bin = find_in_path('wkhtmltopdf')

if wkhtmltopdf_bin is None:

raise IOError

_inherit = 'report.paperformat'

_columns = {

'both_sides': fields.boolean('Both sides'),

_inherit = 'report'

def _merge_pdf(self, documents, both_sides = False):

print "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"

print "merge_pdf %s" % (both_sides)

print "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"

"""Merge PDF files into one.

:param documents: list of path of pdf files

:returns: path of the merged pdf

"""

blankpdfstr = '''JVBERi0xLjQKJcOkw7zDtsOfCjIgMCBvYmoKPDwvTGVuZ3RoIDMgMCBSL0ZpbHRlci9GbGF0ZURl

Y29kZT4+CnN0cmVhbQp4nDPQM1Qo5ypUMFAwALJMLU31jBQsTAz1LBSKUrnCtRTyuAIVAIcdB3IK

ZW5kc3RyZWFtCmVuZG9iagoKMyAwIG9iago0MgplbmRvYmoKCjUgMCBvYmoKPDwKPj4KZW5kb2Jq

Cgo2IDAgb2JqCjw8L0ZvbnQgNSAwIFIKL1Byb2NTZXRbL1BERi9UZXh0XQo+PgplbmRvYmoKCjEg

MCBvYmoKPDwvVHlwZS9QYWdlL1BhcmVudCA0IDAgUi9SZXNvdXJjZXMgNiAwIFIvTWVkaWFCb3hb

MCAwIDU5NSA4NDJdL0dyb3VwPDwvUy9UcmFuc3BhcmVuY3kvQ1MvRGV2aWNlUkdCL0kgdHJ1ZT4+

L0NvbnRlbnRzIDIgMCBSPj4KZW5kb2JqCgo0IDAgb2JqCjw8L1R5cGUvUGFnZXMKL1Jlc291cmNl

cyA2IDAgUgovTWVkaWFCb3hbIDAgMCA1OTUgODQyIF0KL0tpZHNbIDEgMCBSIF0KL0NvdW50IDE+

PgplbmRvYmoKCjcgMCBvYmoKPDwvVHlwZS9DYXRhbG9nL1BhZ2VzIDQgMCBSCi9PcGVuQWN0aW9u

WzEgMCBSIC9YWVogbnVsbCBudWxsIDBdCi9MYW5nKGZyLUZSKQo+PgplbmRvYmoKCjggMCBvYmoK

PDwvQ3JlYXRvcjxGRUZGMDA1NzAwNzIwMDY5MDA3NDAwNjUwMDcyPgovUHJvZHVjZXI8RkVGRjAw

NEMwMDY5MDA2MjAwNzIwMDY1MDA0RjAwNjYwMDY2MDA2OTAwNjMwMDY1MDAyMDAwMzMwMDJFMDAz

NT4KL0NyZWF0aW9uRGF0ZShEOjIwMTIxMTAzMTQ0NzEwKzAxJzAwJyk+PgplbmRvYmoKCnhyZWYK

MCA5CjAwMDAwMDAwMDAgNjU1MzUgZiAKMDAwMDAwMDIyNiAwMDAwMCBuIAowMDAwMDAwMDE5IDAw

MDAwIG4gCjAwMDAwMDAxMzIgMDAwMDAgbiAKMDAwMDAwMDM2OCAwMDAwMCBuIAowMDAwMDAwMTUx

IDAwMDAwIG4gCjAwMDAwMDAxNzMgMDAwMDAgbiAKMDAwMDAwMDQ2NiAwMDAwMCBuIAowMDAwMDAw

NTYyIDAwMDAwIG4gCnRyYWlsZXIKPDwvU2l6ZSA5L1Jvb3QgNyAwIFIKL0luZm8gOCAwIFIKL0lE

IFsgPEYyMjBCNDlBNjRDOEEzRDY3QUFBQzNCODAwNkI5RkRDPgo8RjIyMEI0OUE2NEM4QTNENjdB

QUFDM0I4MDA2QjlGREM+IF0KL0RvY0NoZWNrc3VtIC83NzUwQTAyMEVFNEUwQkU5NjVGMzBDNTND

MkRGNUFGNgo+PgpzdGFydHhyZWYKNzM2CiUlRU9GCg=='''

writer = PdfFileWriter()

blank_page = PdfFileReader(StringIO.StringIO(blankpdfstr.decode("base64"))).pages[0]

streams = [] # We have to close the streams *after* PdfFilWriter's call to write()

for document in documents:

pdfreport = file(document, 'rb')

streams.append(pdfreport)

reader = PdfFileReader(pdfreport)

for page in range(0, reader.getNumPages()):

writer.addPage(reader.getPage(page))

if reader.getNumPages() % 2 and both_sides:

writer.addPage(blank_page)

merged_file_fd, merged_file_path = tempfile.mkstemp(suffix='.html', prefix='report.merged.tmp.')

with closing(os.fdopen(merged_file_fd, 'w')) as merged_file:

writer.write(merged_file)

for stream in streams:

stream.close()

return merged_file_path

def _run_wkhtmltopdf(self, cr, uid, headers, footers, bodies, landscape, paperformat, spec_paperformat_args=None, save_in_attachment=None):

"""Execute wkhtmltopdf as a subprocess in order to convert html given in input into a pdf

document.

:param header: list of string containing the headers

:param footer: list of string containing the footers

:param bodies: list of string containing the reports

:param landscape: boolean to force the pdf to be rendered under a landscape format

:param paperformat: ir.actions.report.paperformat to generate the wkhtmltopf arguments

:param specific_paperformat_args: dict of prioritized paperformat arguments

:param save_in_attachment: dict of reports to save/load in/from the db

:returns: Content of the pdf as a string

"""

command_args = []

# Passing the cookie to wkhtmltopdf in order to resolve internal links.

try:

if request:

command_args.extend(['--cookie', 'session_id', request.session.sid])

except AttributeError:

pass

# Wkhtmltopdf arguments

command_args.extend(['--quiet']) # Less verbose error messages

if paperformat:

# Convert the paperformat record into arguments

command_args.extend(self._build_wkhtmltopdf_args(paperformat, spec_paperformat_args))

both_sides = paperformat.both_sides

else:

both_sides = False

# Force the landscape orientation if necessary

if landscape and '--orientation' in command_args:

command_args_copy = list(command_args)

for index, elem in enumerate(command_args_copy):

if elem == '--orientation':

del command_args[index]

del command_args[index]

command_args.extend(['--orientation', 'landscape'])

elif landscape and not '--orientation' in command_args:

command_args.extend(['--orientation', 'landscape'])

# Execute WKhtmltopdf

pdfdocuments = []

temporary_files = []

for index, reporthtml in enumerate(bodies):

local_command_args = []

pdfreport_fd, pdfreport_path = tempfile.mkstemp(suffix='.pdf', prefix='report.tmp.')

temporary_files.append(pdfreport_path)

# Directly load the document if we already have it

if save_in_attachment and save_in_attachment['loaded_documents'].get(reporthtml[0]):

with closing(os.fdopen(pdfreport_fd, 'w')) as pdfreport:

pdfreport.write(save_in_attachment['loaded_documents'][reporthtml[0]])

pdfdocuments.append(pdfreport_path)

continue

else:

os.close(pdfreport_fd)

# Wkhtmltopdf handles header/footer as separate pages. Create them if necessary.

if headers:

head_file_fd, head_file_path = tempfile.mkstemp(suffix='.html', prefix='report.header.tmp.')

temporary_files.append(head_file_path)

with closing(os.fdopen(head_file_fd, 'w')) as head_file:

head_file.write(headers[index])

local_command_args.extend(['--header-html', head_file_path])

if footers:

foot_file_fd, foot_file_path = tempfile.mkstemp(suffix='.html', prefix='report.footer.tmp.')

temporary_files.append(foot_file_path)

with closing(os.fdopen(foot_file_fd, 'w')) as foot_file:

foot_file.write(footers[index])

local_command_args.extend(['--footer-html', foot_file_path])

# Body stuff

content_file_fd, content_file_path = tempfile.mkstemp(suffix='.html', prefix='report.body.tmp.')

temporary_files.append(content_file_path)

with closing(os.fdopen(content_file_fd, 'w')) as content_file:

content_file.write(reporthtml[1])

try:

wkhtmltopdf = [_get_wkhtmltopdf_bin()] + command_args + local_command_args

wkhtmltopdf += [content_file_path] + [pdfreport_path]

process = subprocess.Popen(wkhtmltopdf, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

out, err = process.communicate()

if process.returncode not in [0, 1]:

raise osv.except_osv(_('Report (PDF)'),

_('Wkhtmltopdf failed (error code: %s). '

'Message: %s') % (str(process.returncode), err))

# Save the pdf in attachment if marked

if reporthtml[0] is not False and save_in_attachment.get(reporthtml[0]):

with open(pdfreport_path, 'rb') as pdfreport:

attachment = {

'name': save_in_attachment.get(reporthtml[0]),

'datas': base64.encodestring(pdfreport.read()),

'datas_fname': save_in_attachment.get(reporthtml[0]),

'res_model': save_in_attachment.get('model'),

'res_id': reporthtml[0],

}

try:

self.pool['ir.attachment'].create(cr, uid, attachment)

except AccessError:

_logger.warning("Cannot save PDF report %r as attachment",

attachment['name'])

else:

_logger.info('The PDF document %s is now saved in the database',

attachment['name'])

pdfdocuments.append(pdfreport_path)

except:

raise

# Return the entire document

if len(pdfdocuments) == 1:

entire_report_path = pdfdocuments[0]

else:

print "---------------------------------------"

print "---------------------------------------"

print "both sides : %s" % (both_sides)

print "---------------------------------------"

print "---------------------------------------"

entire_report_path = self._merge_pdf(pdfdocuments,both_sides)

temporary_files.append(entire_report_path)

with open(entire_report_path, 'rb') as pdfdocument:

content = pdfdocument.read()

# Manual cleanup of the temporary files

for temporary_file in temporary_files:

try:

os.unlink(temporary_file)

except (OSError, IOError):

_logger.error('Error when trying to remove file %s' % temporary_file)