partner-contact/partner_address_version/models/res_partner.py

# Copyright 2018 Akretion - Benoît Guillot
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo import fields, models, api, exceptions, _
import hashlib
from collections import OrderedDict


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

    version_hash = fields.Char(readonly=True, copy=False)
    date_version = fields.Datetime(string="Date version", readonly=True)

    @api.multi
    def get_address_version(self):
        """
        Get a versioned partner corresponding to the partner fields.
        If no versioned partner exists, create a new one.
        """
        self.ensure_one()
        version_hash = self.get_version_hash()
        versioned_partner = self.with_context(active_test=False).search(
            [("version_hash", "=", version_hash)]
        )
        if versioned_partner:
            return versioned_partner
        default = {
            "active": False,
            "version_hash": version_hash,
            "parent_id": self.parent_id and self.parent_id.id or self.id,
            "date_version": fields.Datetime.now(),
            "name": self.name,
        }
        versioned_partner = self.copy(default=default)
        return versioned_partner

    def get_version_fields(self):
        return [
            "name",
            "street",
            "street2",
            "zip",
            "city",
            "country_id",
            "parent_id",
        ]

    def get_version_hash(self):
        version_fields = self.get_version_fields()
        version = OrderedDict()
        for field in version_fields:
            if field == "parent_id":
                parent_id = self.parent_id and self.parent_id.id or self.id
                version[field] = parent_id
            elif self[field]:
                version[field] = self[field]
        version_hash = hashlib.md5(str(version).encode("utf-8")).hexdigest()
        return version_hash

    @api.multi
    def write(self, vals):
        version_fields = self.get_version_fields()
        has_written_versioned_fields = any(
            (f in version_fields) for f in vals.keys()
        )
        for partner in self:
            if partner.version_hash and has_written_versioned_fields:
                raise exceptions.UserError(
                    _(
                        "You can't modify a versioned field %s on the "
                        "versioned partner %s."
                    )
                    % (version_fields, partner.name)
                )
        return super(ResPartner, self).write(vals)