You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
145 lines
4.9 KiB
145 lines
4.9 KiB
from base64 import b64decode
|
|
from tempfile import NamedTemporaryFile as tempfile
|
|
|
|
from openerp import pooler
|
|
from openerp.report.report_sxw import *
|
|
from openerp.tools.translate import _
|
|
from openerp.osv.osv import except_osv
|
|
|
|
from py3o.template import Template
|
|
|
|
from oe_json_serializer import OESerializer
|
|
|
|
import json
|
|
import requests
|
|
import os
|
|
|
|
|
|
class py3o_report(report_sxw):
|
|
# def __init__(self, name, table):
|
|
# super(py3o_report, self).__init__(name, table)
|
|
|
|
def get_values(self, cr, uid, ids, data, context):
|
|
''' Override this function to customize the dictionary given to the
|
|
py3o.template renderer. '''
|
|
|
|
return {
|
|
'lang': self.get_lang(cr, uid, context),
|
|
'objects': self.getObjects(cr, uid, ids, context),
|
|
}
|
|
|
|
def get_lang(self, cr, uid, context):
|
|
pool = pooler.get_pool(cr.dbname)
|
|
lang_obj = pool.get('res.lang')
|
|
user_obj = pool.get('res.users')
|
|
|
|
lang_code = user_obj.browse(cr, uid, uid, context=context).lang
|
|
lang = lang_obj.search(cr, uid,
|
|
[('code', '=', lang_code)],
|
|
context=context)[0]
|
|
return lang_obj.browse(cr, uid, lang, context=context)
|
|
|
|
def format_date(self, date, values):
|
|
''' Return a date formatted according to the language extracted from
|
|
the "values" argument (which should be the result of get_values). '''
|
|
return date.strftime(values['lang'].date_format)
|
|
|
|
def create(self, cr, uid, ids, data, context=None):
|
|
# Find the report definition to get its settings.
|
|
pool = pooler.get_pool(cr.dbname)
|
|
report_xml_obj = pool.get('ir.actions.report.xml')
|
|
report_xml_ids = report_xml_obj.search(cr, uid,
|
|
[('report_name', '=', self.name[7:])], # Ignore "report."
|
|
context=context)
|
|
|
|
if not report_xml_ids:
|
|
return super(py3o_report, self).create(cr, uid, ids, data,
|
|
context=context)
|
|
report_xml = report_xml_obj.browse(cr, uid,
|
|
report_xml_ids[0],
|
|
context=context)
|
|
|
|
template = report_xml.py3o_template_id
|
|
filetype = report_xml.py3o_fusion_filetype
|
|
|
|
|
|
#Try to request fusion server:
|
|
|
|
fusion_server_obj = pool['py3o.server']
|
|
#TODO: Raise a message if no config found
|
|
fusion_server_id = fusion_server_obj.search(
|
|
cr, uid, [], context=context
|
|
)[0]
|
|
fusion_server = fusion_server_obj.browse(cr, uid, fusion_server_id)
|
|
|
|
# py3o.template operates on filenames so create temporary files.
|
|
in_temp = tempfile(suffix='.odt', prefix='py3o-template-')
|
|
|
|
in_temp.write(b64decode(template.py3o_template_data))
|
|
in_temp.seek(0)
|
|
out_temp = tempfile(suffix='.odt', prefix='py3o-report-')
|
|
|
|
# We need to get the variables used in the template
|
|
#TODO: Find a way to avoid calling Template
|
|
t = Template(in_temp.name, out_temp.name)
|
|
# Remove 'py3o.'
|
|
user_variable = [x[5:] for x in t.get_user_variable()]
|
|
print user_variable
|
|
|
|
values = self.get_values(cr, uid, ids, data, context)
|
|
print values
|
|
|
|
#WARNING: We rely on the fact that there is a for loop on the report
|
|
# on objects (for object in objects) due to lack of time
|
|
val_dict = {}
|
|
for val in values:
|
|
if val == 'objects':
|
|
o = []
|
|
for obj in values[val]:
|
|
x = OESerializer.serialize(
|
|
obj,
|
|
[
|
|
v[len('object') + 1:]
|
|
for v in user_variable
|
|
if v.startswith('object')
|
|
]
|
|
)
|
|
o.append(x)
|
|
val_dict.update({val: o})
|
|
continue
|
|
|
|
x = OESerializer.serialize(
|
|
values[val],
|
|
[
|
|
v[len(val) + 1:]
|
|
for v in user_variable
|
|
if v.startswith(val)
|
|
]
|
|
)
|
|
val_dict.update({val: x})
|
|
|
|
import pprint
|
|
pprint.pprint(val_dict)
|
|
val_json = json.dumps(val_dict)
|
|
|
|
fields = {
|
|
'targetformat': filetype.fusion_ext,
|
|
'datadict': val_json,
|
|
'image_mapping': '{}',
|
|
}
|
|
print fields
|
|
|
|
r = requests.post(
|
|
fusion_server.url, data=fields, files={'tmpl_file': in_temp}
|
|
)
|
|
in_temp.close()
|
|
if r.status_code == 400:
|
|
raise Exception("Problem with fusion server: %s" % r.json())
|
|
|
|
chunk_size = 1024
|
|
|
|
ext = filetype.human_ext
|
|
for chunk in r.iter_content(chunk_size):
|
|
out_temp.write(chunk)
|
|
out_temp.seek(0)
|
|
return out_temp.read(), ext
|