account_invoice_xrechnung/xreport.py
Frederik Jaeckel 7667a9b9b9 formatting
2023-06-30 11:21:48 +02:00

58 lines
1.8 KiB
Python

# -*- coding: utf-8 -*-
# This file is part of the account-invoice-xrechnung-module
# from m-ds for Tryton. The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
import zipfile
from io import BytesIO
from trytond.report import Report
from trytond.pool import Pool
from slugify import slugify
from .wizard_runreport import edoc_versions
class XReport(Report):
'eDocument Export'
__name__ = 'account_invoice_xrechnung.export'
@classmethod
def compress_as_zip(cls, fname, data):
""" compress
"""
content = BytesIO()
with zipfile.ZipFile(content, 'w') as content_zip:
content_zip.writestr(fname, data)
return content.getvalue()
@classmethod
def execute(cls, ids, data):
""" skip export-engine, run edocument-xml-convert
"""
pool = Pool()
IrDate = pool.get('ir.date')
Invoice = pool.get('account.invoice')
EDocument = pool.get(data['edocument'])
invoice, = Invoice.browse([data['invoice']])
template = EDocument(invoice)
invoice_string = template.render(edoc_versions[data['edocument']])
file_name = slugify('%(date)s-%(descr)s' % {
'date': IrDate.today().isoformat().replace('-', ''),
'descr': invoice.rec_name,
},
max_length=100, word_boundary=True, save_order=True)
if data['as_zip'] is True:
return (
'zip',
cls.compress_as_zip('%(fname)s.%(ext)s' % {
'fname': file_name,
'ext': 'xml',
}, invoice_string),
False,
file_name)
else:
return ('xml', invoice_string, False, file_name)
# end XReport