account_invoice_xrechnung/wizard_runreport.py

# -*- 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.

from trytond.model import ModelView, fields
from trytond.wizard import Wizard, StateView, StateReport, Button
from trytond.pool import Pool
from trytond.exceptions import UserError
from trytond.i18n import gettext
from trytond.transaction import Transaction


sel_edocument = [
    ('edocument.xrechnung.invoice-2.2', 'XRechnung UBL Invoice 2.2'),
    ('edocument.xrechnung.invoice-2.3', 'XRechnung UBL Invoice 2.3'),
    ('edocument.xrechnung.invoice-3.0', 'XRechnung UBL Invoice 3.0'),
    ('edocument.uncefact.invoice', 'CII CrossIndustryInvoice D16B'),
    ]

edoc_versions = {
    'edocument.xrechnung.invoice-2.2': 'XRechnung-2.2',
    'edocument.xrechnung.invoice-2.3': 'XRechnung-2.3',
    'edocument.xrechnung.invoice-3.0': 'XRechnung-3.0',
    'edocument.uncefact.invoice': '16B-CII'}


class RunXRechnungReportStart(ModelView):
    'eDocument Export'
    __name__ = 'account_invoice_xrechnung.runrep.start'

    invoice = fields.Many2One(
        string='Invoice', readonly=True,
        model_name='account.invoice', required=True)
    state = fields.Char(
        string='State', readonly=True, states={'invisible': True})
    edocument = fields.Selection(
        string='Type', required=True, selection=sel_edocument)
    as_zip = fields.Boolean(string='ZIP-File')

    @classmethod
    def default_as_zip(cls):
        """ default: True
        """
        return True

    @classmethod
    def default_edocument(cls):
        """ default xrechnung
        """
        return 'edocument.xrechnung.invoice-3.0'

# end RunXRechnungReportStart


class RunXRechnungReport(Wizard):
    'eDocument Export'
    __name__ = 'account_invoice_xrechnung.runrep'

    start_state = 'start'
    export = StateReport('account_invoice_xrechnung.export')
    start = StateView(
        model_name='account_invoice_xrechnung.runrep.start',
        view='account_invoice_xrechnung.wiz_start_form',
        buttons=[
            Button(string='Cancel', state='end', icon='tryton-cancel'),
            Button(string='Export', state='export', icon='tryton-export'),
            ])

    def default_start(self, fields):
        """ set defaults
        """
        context = Transaction().context
        pool = Pool()
        Invoice = pool.get('account.invoice')
        WizRepStart = pool.get('account_invoice_xrechnung.runrep.start')

        invoice = Invoice.browse([context.get('active_id', -1)])
        result = {
            'edocument': WizRepStart.default_edocument(),
            'invoice': context.get('active_id', -1),
            'state': invoice[0].state if invoice else ''}
        return result

    def do_export(self, action):
        """ run export
        """
        if self.start.state != 'posted':
            raise UserError(gettext(
                'account_invoice_xrechnung.msg_invoice_must_posted',
                invname=self.start.invoice.rec_name))
        return action, {
            'invoice': self.start.invoice.id,
            'edocument': self.start.edocument,
            'as_zip': self.start.as_zip}

# end RunXRechnungReport