partner_credito_limite/models/sale.py

#-*- coding:utf-8 -*-
from openerp import models, api, _
from openerp.exceptions import Warning
from datetime import datetime, date


class sale_order(models.Model):
    _inherit = "sale.order"

    @api.one
    def action_wait(self):
        self.check_limit()
        return super(sale_order, self).action_wait()

    @api.one
    def check_limit(self):

        # if self.order_policy == 'prepaid':
        if self.contado == True:
            return True

        # We sum from all the sale orders that are aproved, the sale order
        # lines that are not yet invoiced
        domain = [('order_id.partner_id', '=', self.partner_id.id),
                  ('invoiced', '=', False),
                  ('order_id.state', 'not in', ['draft', 'cancel', 'sent'])]
        order_lines = self.env['sale.order.line'].search(domain)
        none_invoiced_amount = sum([x.price_subtotal for x in order_lines])

        # We sum from all the invoices that are in draft the total amount
        domain = [
            ('partner_id', '=', self.partner_id.id), ('state', '=', 'draft')]
        draft_invoices = self.env['account.invoice'].search(domain)
        draft_invoices_amount = sum([x.amount_total for x in draft_invoices])

        available_credit = self.partner_id.credit_limit - \
            self.partner_id.credit - \
            none_invoiced_amount - draft_invoices_amount

        if self.amount_total > available_credit:
            msg = _('No se puede confirmar el Pedido ya que el cliente no tiene credito suficiente.\
                    Cambia el límite de cliente en su ficha e intente de nuevo"')
            raise Warning(msg)
            return False
        return True


# class sale_order(models.Model):
#     _inherit = "sale.order"
#
#     @api.one
#     def action_wait(self):
#         self.check_limit()
#         # self.check_morosidad()
#         return super(sale_order, self).action_wait()
#
#     @api.one
#     def check_limit(self):
#
#             if self.contado == True:
#                 return True
#
#             available_credit = self.partner_id.credit_limit - self.partner_id.credit
#
#             if self.amount_total > available_credit:
#                 if not self.user_has_groups('partner_credito_limite.credit_config'):
#                     msg = 'No se puede confirmar el Pedido ya que el cliente no tiene crédito suficiente.\
#                         Pruebe marcar la opción "Contado"'
#                     raise Warning(_(msg))
#                     return False
#                 return True
#
#