46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2015 ACSONE SA/NV (<http://acsone.eu>)
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
|
|
from cStringIO import StringIO
|
|
|
|
from openerp.report.report_sxw import report_sxw
|
|
from openerp.api import Environment
|
|
|
|
import logging
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
import xlsxwriter
|
|
except ImportError:
|
|
_logger.debug('Can not import xlsxwriter`.')
|
|
|
|
|
|
class ReportXlsx(report_sxw):
|
|
|
|
def create(self, cr, uid, ids, data, context=None):
|
|
self.env = Environment(cr, uid, context)
|
|
report_obj = self.env['ir.actions.report.xml']
|
|
report = report_obj.search([('report_name', '=', self.name[7:])])
|
|
if report.ids:
|
|
self.title = report.name
|
|
if report.report_type == 'xlsx':
|
|
return self.create_xlsx_report(ids, data, report)
|
|
return super(ReportXlsx, self).create(cr, uid, ids, data, context)
|
|
|
|
def create_xlsx_report(self, ids, data, report):
|
|
self.parser_instance = self.parser(
|
|
self.env.cr, self.env.uid, self.name2, self.env.context)
|
|
objs = self.getObjects(
|
|
self.env.cr, self.env.uid, ids, self.env.context)
|
|
self.parser_instance.set_context(objs, data, ids, 'xlsx')
|
|
file_data = StringIO()
|
|
workbook = xlsxwriter.Workbook(file_data)
|
|
self.generate_xlsx_report(workbook, data, objs)
|
|
workbook.close()
|
|
file_data.seek(0)
|
|
return (file_data.read(), 'xlsx')
|
|
|
|
def generate_xlsx_report(self, workbook, data, objs):
|
|
raise NotImplementedError()
|