docx_report_generation/models/contract_wizard.py

# -*- coding: utf-8 -*-
import base64
import math
from datetime import datetime

from odoo import api, fields, models
from odoo.exceptions import UserError
from pytils import numeral

from ..utils.docxtpl import get_document_from_values_stream

class ContractWizard(models.TransientModel):
    _name = 'res.partner.contract.wizard'

    def _get_default_template(self):
        _template = self.env['res.partner.contract.template'].search([
            ('is_default', '=', True)
        ])
        if _template:
            return _template[0].id
        return False

    def _get_default_partner(self):
        current_id = self.env.context.get('active_ids')
        return self.env['res.partner.contract'].browse([current_id[0]]).partner_id.id

    def _get_default_contract(self):
        current_id = self.env.context.get('active_ids')
        return self.env['res.partner.contract'].browse(current_id[0])

    def _get_partner_representer(self):
        return self.partner_id.representative_id

    _context_name = fields.Char(
        string='Contract number',
        compute='_compute_context_name',
        readonly=True
    )
    _context_date = fields.Char(
        string='Contract date',
        compute='_compute_context_date',
        readonly=True
    )
    _context_partner_contract_name = fields.Char(
        string='Partner contract name',
        compute='_compute_context_partner_contract_name',
        readonly=True
    )
    _context_partner_adress = fields.Char(
        compute='_compute_context_partner_adress',
        readonly=True
    )
    _context_partner_representer_contract_name = fields.Char(
        string='partner representer contract name',
        compute='_compute_context_partner_representer_contract_name',
        readonly=True
    )
    _context_partner_inn = fields.Char(
        string='Partner inn',
        compute='_compute_context_partner_inn',
        readonly=True
    )
    _context_partner_kpp = fields.Char(
        string='Partner kpp',
        compute='_compute_context_partner_kpp',
        readonly=True
    )
    _context_partner_rs = fields.Char(
        string='Partner corresponding account',
        compute='_compute_context_partner_rs',
        readonly=True
    )
    _context_partner_bik = fields.Char(
        string='Partner bank bik',
        compute='_compute_context_partner_bik',
        readonly=True
    )
    _context_partner_bank = fields.Char(
        string='Partner bank name',
        compute='_compute_context_partner_bank',
        readonly=True
    )
    _context_partner_phone = fields.Char(
        string='Partner phone',
        compute='_compute_context_partner_phone',
        readonly=True
    )
    _context_partner_representer_name = fields.Char(
        string='Partner representer name',
        compute='_compute_context_partner_representer_name',
        readonly=True
    )
    _context_seller_contract_name = fields.Char(
        string='Seller contract name',
        compute='_compute_context_seller_contract_name',
        readonly=True
    )
    _context_seller_adress = fields.Char(
        string='Seller full adress',
        compute='_compute_context_seller_adress',
        readonly=True
    )
    _context_seller_representer_contract_job_name = fields.Char(
        string='Seller representer contract job name',
        compute='_compute_context_seller_representer_contract_job_name',
        readonly=True
    )
    _context_seller_representer_contract_name = fields.Char(
        string='Seller representer contract name',
        compute='_compute_context_seller_representer_contract_name',
        readonly=True
    )
    _context_seller_inn = fields.Char(
        string='Seller inn',
        compute='_compute_context_seller_inn',
        readonly=True
    )
    _context_seller_kpp = fields.Char(
        string='Seller kpp',
        compute='_compute_context_seller_kpp',
        readonly=True
    )
    _context_seller_rs = fields.Char(
        string='Seller corresponding account',
        compute='_compute_context_seller_rs',
        readonly=True
    )
    _context_seller_bik = fields.Char(
        string='Seller bank bik',
        compute='_compute_context_seller_bik',
        readonly=True
    )
    _context_seller_bank = fields.Char(
        string='Seller bank name',
        compute='_compute_context_seller_bank',
        readonly=True
    )
    _context_seller_phone = fields.Char(
        string='Seller phone',
        compute='_compute_context_seller_phone',
        readonly=True
    )
    _context_seller_representer_job_name = fields.Char(
        string='Seller representer job name',
        compute='_compute_context_seller_representer_job_name',
        readonly=True
    )
    _context_seller_representer_name = fields.Char(
        string='Seller representer name',
        compute='_compute_context_seller_representer_name',
        readonly=True
    )
    _context_summ_rub = fields.Char(
        string='Contract summ(rub)',
        compute='_compute_context_summ_rub',
        readonly=True
    )
    _context_summ_rub_word = fields.Char(
        string='Contract summ(rub), word',
        compute='_compute_context_summ_rub_word',
        readonly=True
    )
    _context_summ_kop = fields.Char(
        string='Contract summ(kop)',
        compute='_compute_context_summ_kop',
        readonly=True
    )
    _context_summ_word = fields.Char(
        string='Contract summ word',
        compute='_compute_context_summ_word',
        readonly=True
    )
    _context_delivery_term = fields.Char(
        string='Contract delivery term',
        compute='_compute_context_delivery_term',
        readonly=True
    )
    _context_delivery_term_word = fields.Char(
        string='Contract delivery term word',
        compute='_compute_context_delivery_term_word',
        readonly=True
    )
    _context_payment_term = fields.Char(
        string='Contract payment term',
        compute='_compute_context_payment_term', readonly=True
    )
    _context_payment_term_word = fields.Char(
        string='Contract payment term word',
        compute='_compute_context_payment_term_word', readonly=True
    )
    _context_partner_passport_data = fields.Char(
        string='Partner passport data',
        compute='_compute_partner_passport_data',
        readonly=True
    )

    annex_lines = fields.One2many(
        'res.partner.contract.annex.line',
        'id',
        auto_join=True,
        copy=True
    )
    company_id = fields.Many2one(
        'res.partner',
        string='Company',
        help='Seller company',
        default=lambda self: self.env.user.company_id.partner_id
    )
    contract_id = fields.Many2one(
        'res.partner.contract',
        string='Contract',
        default=_get_default_contract
    )
    delivery_terms = fields.Integer(
        string='Delivery terms',
        help='When product must be delivered',
        default=10
    )
    order_id = fields.Many2one(
        'sale.order',
        string='Appex order',
        help='Appex',
    )
    partner_id = fields.Many2one(
        'res.partner',
        string='Partner',
        help='Partner to render contract',
        default=_get_default_partner
    )
    payment_terms = fields.Integer(
        string='Payment term',
        help='When customer must pay',
        default=45
    )
    template = fields.Many2one(
        'res.partner.contract.template',
        string='Template',
        help='Template for contract',
        default=_get_default_template
    )
    type = fields.Selection(
        selection=[
            ('person', 'With person'),
            ('company', 'With company')
        ],
        string='Type of contract',
        default='company'
    )

    transient_field_ids = fields.One2many(
        'res.partner.contract.field.transient',
        '_contract_wizard_id',
        string="Contract Fields",
    )

    @api.onchange('partner_id')
    def _set_order_domain(self):
        current_id = self.env.context.get('active_ids')
        domain = [('contract_id', '=', current_id)]
        return {'domain': {'order_id': domain}}

    @api.onchange('partner_id')
    def _onchange_partner_id(self):
        """Creates transient fields for generate contract template
        Looks as a tree view of *_contract_field_transient model in xml
        """
        def get_contract_field(technical_name):
            return self.env['res.partner.contract.field'].search([
                ('technical_name', '=', technical_name),
            ])

        contract_context_values = self.env.ref(
            'client_contracts.action_get_context').with_context({
                "onchange_self": self,
            }).run()

        self.transient_field_ids = [  # one2many
            (
                4,
                self.env['res.partner.contract.field.transient'].create({
                    "contract_field_id": get_contract_field(field).id,
                    "value": value,
                }).id,
                0,
            ) for field, value in contract_context_values.items()
        ]

    @api.multi
    def get_docx_contract(self):
        template = self.template.attachment_id
        if not template:
            raise UserError("Template must be set up")

        path_to_template = template._full_path(template.store_fname)

        fields = {
            transient_field.technical_name: transient_field.value
            for transient_field
            in self.transient_field_ids
            if transient_field.technical_name and transient_field.value
        }

        binary_data = get_document_from_values_stream(
            path_to_template, fields).read()
        encoded_data = base64.b64encode(binary_data)

        attachment_name = "Contract-{number}.{ext}".format(
            number=self.contract_id.name,
            ext="docx"
        )
        document_as_attachment = self.env['ir.attachment'].create({
            "name": attachment_name,
            "datas_fname": attachment_name,
            "type": "binary",
            "datas": encoded_data,
        })

        # Send message with attachment to a mail.thread of the company
        self.env['mail.message'].create({
            "model": "res.partner.contract",
            "res_id": self.contract_id.id,
            "message_type": "comment",
            "attachment_ids": [(4, document_as_attachment.id, False)]
        })

        return document_as_attachment

    def modf(self, arg):
        """Math.modf function for using in XML ir.action.server code
        Uses in data/fields_default.xml
        """
        return math.modf(arg)


class AnnexLine(models.TransientModel):
    _name = 'res.partner.contract.annex.line'

    @api.onchange('annex_type')
    def _get_default_description(self):
        self.description = self.annex_type.description

    annex_type = fields.Many2one(
        'res.partner.contract.annex.type'
    )
    description = fields.Text()