OCA reporting engine fork for dev and update.
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

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