account-financial-tools/account_partner_required/models/account.py

# © 2014-2016 Acsone (http://acsone.eu)
# © 2016 Akretion (http://www.akretion.com/)
# @author Stéphane Bidoul <stephane.bidoul@acsone.eu>
# @author Alexis de Lattre <alexis.delattre@akretion.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)

from odoo import models, fields, api, _
from odoo.tools import float_is_zero
from odoo.exceptions import ValidationError


class AccountAccountType(models.Model):
    _inherit = "account.account.type"

    partner_policy = fields.Selection(
        selection=[
            ('optional', 'Optional'),
            ('always', 'Always'),
            ('never', 'Never')],
        string='Policy for Partner Field',
        required=True,
        default='optional',
        help="Set the policy for the partner field : if you select "
             "'Optional', the accountant is free to put a partner "
             "on an account move line with this type of account ; "
             "if you select 'Always', the accountant will get an error "
             "message if there is no partner ; if you select 'Never', "
             "the accountant will get an error message if a partner "
             "is present.")


class AccountAccount(models.Model):
    _inherit = 'account.account'

    @api.multi
    def get_partner_policy(self):
        """ Extension point to obtain partner policy for an account """
        self.ensure_one()
        return self.user_type_id.partner_policy


class AccountMoveLine(models.Model):
    _inherit = "account.move.line"

    @api.multi
    def _check_partner_required_msg(self):
        prec = self.env.user.company_id.currency_id.rounding
        for line in self:
            if (
                    float_is_zero(line.debit, precision_rounding=prec) and
                    float_is_zero(line.credit, precision_rounding=prec)):
                continue
            policy = line.account_id.get_partner_policy()
            if policy == 'always' and not line.partner_id:
                return _(
                    "Partner policy is set to 'Always' with account '%s' but "
                    "the partner is missing in the account move line with "
                    "label '%s'.") % (line.account_id.name_get()[0][1],
                                      line.name)
            elif policy == 'never' and line.partner_id:
                return _(
                    "Partner policy is set to 'Never' with account '%s' but "
                    "the account move line with label '%s' has a partner "
                    "'%s'.") % (line.account_id.name_get()[0][1],
                                line.name,
                                line.partner_id.name)

    @api.multi
    @api.constrains('partner_id', 'account_id', 'debit', 'credit')
    def _check_partner_required(self):
        for line in self:
            message = line._check_partner_required_msg()
            if message:
                raise ValidationError(message)