# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import fields, models
from odoo.exceptions import UserError


class ResPartner(models.Model):
    _inherit = 'res.partner'

    grade_id = fields.Many2one('res.partner.grade', 'Partner Level', tracking=True, group_expand='_read_group_expand_full')

    def write(self, vals):
        if vals.get('grade_id'):
            grade = self.env['res.partner.grade'].browse(vals['grade_id'])
            if grade.default_pricelist_id:
                pricelist = vals.get('specific_property_product_pricelist') or vals.get('property_product_pricelist')
                if pricelist and pricelist != grade.default_pricelist_id.id:
                    raise UserError(self.env._(
                        "You are trying to assign two different pricelists (one directly and one from grade (%(grade_name)s)).",
                        grade_name=grade.name,
                    ))
                else:
                    vals['specific_property_product_pricelist'] = grade.default_pricelist_id.id
        return super().write(vals)