Browse Source

[PEP8]

pull/3/head
Yannick Vaucher 11 years ago
parent
commit
eae4fbbb75
  1. 4
      base_report_assembler/assembled_report.py
  2. 14
      base_report_assembler/ir_report.py
  3. 8
      base_report_assembler/report_assembler.py

4
base_report_assembler/assembled_report.py

@ -20,13 +20,15 @@
############################################################################## ##############################################################################
from openerp.osv import orm, fields from openerp.osv import orm, fields
class AssembledReport(orm.Model): class AssembledReport(orm.Model):
_name = 'assembled.report' _name = 'assembled.report'
_order = 'sequence' _order = 'sequence'
_columns = { _columns = {
'report_id': fields.many2one('ir.actions.report.xml', 'Report',
'report_id': fields.many2one(
'ir.actions.report.xml', 'Report',
domain="[('model', '=', model)," domain="[('model', '=', model),"
"('report_type', '!=', 'assemblage')]", required=True), "('report_type', '!=', 'assemblage')]", required=True),
'model': fields.char('Object model'), 'model': fields.char('Object model'),

14
base_report_assembler/ir_report.py

@ -18,12 +18,12 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
# #
############################################################################## ##############################################################################
from openerp.osv import orm from openerp.osv import orm
from openerp import netsvc from openerp import netsvc
from openerp.report.report_sxw import rml_parse from openerp.report.report_sxw import rml_parse
from report_assembler import PDFReportAssembler from report_assembler import PDFReportAssembler
def register_report(name, model, parser=rml_parse): def register_report(name, model, parser=rml_parse):
"""Register the report into the services""" """Register the report into the services"""
name = 'report.%s' % name name = 'report.%s' % name
@ -72,8 +72,7 @@ class ReportAssembleXML(orm.Model):
cursor, cursor,
user, user,
ids, ids,
context
)
context)
return res return res
def create(self, cursor, user, vals, context=None): def create(self, cursor, user, vals, context=None):
@ -83,19 +82,18 @@ class ReportAssembleXML(orm.Model):
# I really look forward to virtual functions :S # I really look forward to virtual functions :S
register_report( register_report(
vals['report_name'], vals['report_name'],
vals['model'],
)
vals['model'])
return res return res
def write(self, cr, uid, ids, vals, context=None): def write(self, cr, uid, ids, vals, context=None):
"Edit report and manage its registration" "Edit report and manage its registration"
if isinstance(ids, (int, long)): if isinstance(ids, (int, long)):
ids = [ids,]
ids = [ids]
for rep in self.browse(cr, uid, ids, context=context): for rep in self.browse(cr, uid, ids, context=context):
if rep.report_type != 'assemblage': if rep.report_type != 'assemblage':
continue continue
if vals.get('report_name', False) and \
vals['report_name'] != rep.report_name:
if (vals.get('report_name', False)
and vals['report_name'] != rep.report_name):
report_name = vals['report_name'] report_name = vals['report_name']
else: else:
report_name = rep.report_name report_name = rep.report_name

8
base_report_assembler/report_assembler.py

@ -25,10 +25,11 @@ from StringIO import StringIO
from openerp.netsvc import ExportService from openerp.netsvc import ExportService
from openerp.report import report_sxw from openerp.report import report_sxw
from openerp import pooler, sql_db
from openerp import pooler
_POLLING_DELAY = 0.25 _POLLING_DELAY = 0.25
def assemble_pdf(pdf_list): def assemble_pdf(pdf_list):
""" """
Assemble a list of pdf Assemble a list of pdf
@ -53,6 +54,7 @@ def assemble_pdf(pdf_list):
output.write(s) output.write(s)
return s.getvalue() return s.getvalue()
class PDFReportAssembler(report_sxw.report_sxw): class PDFReportAssembler(report_sxw.report_sxw):
""" PDFReportAssembler allows to put 2 invoice reports in one single pdf""" """ PDFReportAssembler allows to put 2 invoice reports in one single pdf"""
@ -61,7 +63,6 @@ class PDFReportAssembler(report_sxw.report_sxw):
Return a list of pdf encoded in base64 Return a list of pdf encoded in base64
""" """
pool = pooler.get_pool(cr.dbname) pool = pooler.get_pool(cr.dbname)
db = sql_db.db_connect(cr.dbname)
report_obj = pool.get('ir.actions.report.xml') report_obj = pool.get('ir.actions.report.xml')
spool = ExportService._services['report'] spool = ExportService._services['report']
@ -70,7 +71,8 @@ class PDFReportAssembler(report_sxw.report_sxw):
report_list = report_obj.browse(cr, uid, report_ids, context=context) report_list = report_obj.browse(cr, uid, report_ids, context=context)
for report in report_list: for report in report_list:
report_key = spool.exp_report(cr.dbname, uid, report.report_name, ids, datas=data, context=context)
report_key = spool.exp_report(cr.dbname, uid, report.report_name,
ids, datas=data, context=context)
while 1: while 1:
res = spool.exp_report_get(cr.dbname, uid, report_key) res = spool.exp_report_get(cr.dbname, uid, report_key)
if res.get('state'): if res.get('state'):

Loading…
Cancel
Save