partner-contact/partner_financial_risk/models/sale.py

# -*- coding: utf-8 -*-
# © 2016 Carlos Dauden <carlos.dauden@tecnativa.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from openerp import api, fields, models, _


class SaleOrder(models.Model):
    _inherit = 'sale.order'

    invoice_amount = fields.Monetary(
        compute='_compute_invoice_amount', store=True)
    invoice_pending_amount = fields.Monetary(
        compute='_compute_invoice_amount', store=True)

    @api.multi
    @api.depends('state', 'order_line.invoice_lines.invoice_id.amount_total')
    def _compute_invoice_amount(self):
        for order in self.filtered(lambda x: x.state == 'sale'):
            order.invoice_amount = sum(
                order.invoice_ids.mapped('amount_total'))
            order.invoice_pending_amount = (
                order.amount_total - order.invoice_amount)

    @api.multi
    def action_confirm(self):
        if not self.env.context.get('bypass_risk', False):
            partner = self.partner_id
            risk_exception = False
            if partner.risk_exception:
                risk_exception = True
                exception_msg = _("Financial risk exceeded.\n")
            elif partner.risk_sale_order_limit and (
                    (partner.risk_sale_order + self.amount_total) >
                    partner.risk_sale_order_limit):
                risk_exception = True
                exception_msg = _(
                    "This sale order exceeds the sales orders risk.\n")
            elif partner.risk_sale_order_include and (
                    (partner.risk_total + self.amount_total) >
                    partner.credit_limit):
                risk_exception = True
                exception_msg = _(
                    "This sale order exceeds the financial risk.\n")
            if risk_exception:
                wizard = self.env['sale.order.risk.exceeded'].create({
                    'exception_msg': exception_msg,
                    'partner_id': partner.id,
                    'amount': self.amount_total})
                return {
                    'type': 'ir.actions.act_window',
                    'name': _('Partner risk exceeded'),
                    'res_model': wizard._name,
                    'res_id': wizard.id,
                    'view_type': 'form',
                    'view_mode': 'form',
                    'target': 'new',
                }
        return super(SaleOrder, self).action_confirm()