vertical-cooperative/easy_my_coop_api/models/external_id_mixin.py

# Copyright 2020 Coop IT Easy SCRL fs
#   Robin Keunen <robin@coopiteasy.be>
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).

from psycopg2 import IntegrityError

from odoo import api, fields, models


class ExternalIdMixin(models.AbstractModel):
    _name = "external.id.mixin"
    _description = "External ID Mixin"

    _sql_constraints = [
        (
            "_api_external_id_uniq",
            "unique(_api_external_id)",
            "API External ID must be unique!",
        )
    ]

    # do not access directly, always use get_api_external_id method
    _api_external_id = fields.Integer(
        string="External ID", index=True, required=False
    )
    external_id_sequence_id = fields.Many2one(
        comodel_name="ir.sequence",
        string="External ID Sequence",
        required=False,
    )

    @api.multi
    def set_external_sequence(self):
        self.ensure_one()
        code = "%s.external.id" % self._name
        Sequence = self.env["ir.sequence"]
        # check if code was created for that model
        sequence = Sequence.search([("code", "=", code)])
        if not sequence:
            sequence = Sequence.sudo().create(
                {"name": code, "code": code, "number_next": 100}
            )

        self.sudo().write({"external_id_sequence_id": sequence.id})
        return True

    @api.multi
    def get_api_external_id(self):
        self.ensure_one()
        if not self.external_id_sequence_id:
            self.set_external_sequence()
        if not self._api_external_id:
            # pass already allocated ids
            n = 100
            while True:
                try:
                    self.sudo().write(
                        {
                            "_api_external_id": self.external_id_sequence_id._next()
                        }
                    )
                    break
                except IntegrityError as e:
                    if n > 0:
                        continue
                    else:
                        raise e
        return self._api_external_id


class ResPartner(models.Model):
    _name = "res.partner"
    _inherit = ["res.partner", "external.id.mixin"]


class SubscriptionRequest(models.Model):
    _name = "subscription.request"
    _inherit = ["subscription.request", "external.id.mixin"]


class AccountAccount(models.Model):
    _name = "account.account"
    _inherit = ["account.account", "external.id.mixin"]


class AccountJournal(models.Model):
    _name = "account.journal"
    _inherit = ["account.journal", "external.id.mixin"]


class AccountInvoice(models.Model):
    _name = "account.invoice"
    _inherit = ["account.invoice", "external.id.mixin"]


class AccountPayment(models.Model):
    _name = "account.payment"
    _inherit = ["account.payment", "external.id.mixin"]


class ProductTemplate(models.Model):
    _name = "product.template"
    _inherit = ["product.template", "external.id.mixin"]