Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-iot-iot_rule/iot_rule/models/iot_key.py

@@ -0,0 +1,122 @@
+# Copyright 2019 Creu Blanca
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+import uuid
+from datetime import datetime
+
+from odoo import _, api, fields, models
+from odoo.exceptions import ValidationError
+
+
+class IotKey(models.Model):
+    _name = "iot.key"
+    _description = "IoT Key"
+    _rec_name = "unique_virtual_key"
+
+    name = fields.Char()
+    key_type = fields.Char()
+    unique_virtual_key = fields.Char(
+        readonly=True,
+        default="/",
+        required=True,
+    )
+    expiration_date = fields.Datetime()
+    rule_ids = fields.Many2many("iot.rule", string="Rules")
+    action_ids = fields.One2many("iot.key.action", inverse_name="key_id")
+    active = fields.Boolean(default=True)
+    res_id = fields.Integer(index=True)
+    res_model = fields.Char(index=True)
+
+    _sql_constraints = [
+        (
+            "unique_virtual_key_uniq",
+            "UNIQUE(unique_virtual_key)",
+            "Key must be unique",
+        ),
+    ]
+
+    def _get_unique_virtual_key(self, vals):
+        """Hook that can be used to define the key according to needs"""
+        return uuid.uuid4()
+
+    @api.model_create_multi
+    def create(self, mvals):
+        for vals in mvals:
+            if vals.get("unique_virtual_key", "/") == "/":
+                vals["unique_virtual_key"] = self._get_unique_virtual_key(vals)
+        return super().create(mvals)
+
+    def view_actions(self):
+        self.ensure_one()
+        action = self.env.ref("iot_rule.iot_key_action_act_window").read()[0]
+        action["domain"] = [("key_id", "=", self.id)]
+        return action
+
+    def get_iot_rule_values(self):
+        result = []
+        for key in self:
+            result.append(key._get_iot_rule_values())
+        return result
+
+    def _get_iot_rule_values(self):
+        if self.expiration_date:
+            expiration_timestamp = int(datetime.timestamp(self.expiration_date))
+        else:
+            expiration_timestamp = False
+        return {
+            "key_type": self.key_type,
+            "expiration_date": expiration_timestamp,
+            "unique_virtual_key": self.unique_virtual_key,
+            "id": self.id,
+        }
+
+    def _get_unique_key_models(self):
+        return []
+
+    @api.constrains("res_model", "res_id")
+    def _constrain_key_model(self):
+        unique_models = self._get_unique_key_models()
+        for record in self:
+            if record.res_model not in unique_models:
+                continue
+            if self.with_context(active_test=False).search(
+                [
+                    ("id", "!=", record.id),
+                    ("res_id", "=", record.res_id),
+                    ("res_model", "=", record.res_model),
+                ],
+                limit=1,
+            ):
+                raise ValidationError(_("Only one key can be assigned to this model"))
+
+
+class IotKeyMixin(models.AbstractModel):
+    _name = "iot.key.mixin"
+    _description = "Mixin for relation between record and Iot Keys"
+
+    iot_key_ids = fields.One2many(
+        "iot.key",
+        inverse_name="res_id",
+        domain=lambda r: [("res_model", "=", r._name)],
+    )
+    iot_key_count = fields.Integer(compute="_compute_iot_key_count")
+
+    @api.depends("iot_key_ids")
+    def _compute_iot_key_count(self):
+        for record in self:
+            record.iot_key_count = len(record.iot_key_ids)
+
+    def action_view_iot_key(self):
+        self.ensure_one()
+        action = self.env["ir.actions.act_window"]._for_xml_id(
+            "iot_rule.iot_key_act_window"
+        )
+        action["domain"] = [
+            ("res_id", "=", self.id),
+            ("res_model", "=", self._name),
+        ]
+        action["context"] = {
+            "default_res_id": self.id,
+            "default_res_model": self._name,
+        }
+        return action