# -*- coding: utf-8 -*-
# © 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"

    @api.model
    def _get_partner_policies(self):
        """This is the method to be inherited for adding policies"""
        return [('optional', _('Optional')),
                ('always', _('Always')),
                ('never', _('Never'))]

    partner_policy = fields.Selection(
        _get_partner_policies, 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)