[12.0][ADD] kpi_dashboard

kpi_dashboard/models/kpi_kpi.py

@@ -0,0 +1,110 @@
+# Copyright 2020 Creu Blanca
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo import api, fields, models
+import ast
+
+
+class KpiKpi(models.Model):
+    _name = "kpi.kpi"
+    _description = "Kpi Kpi"
+
+    name = fields.Char(required=True)
+    active = fields.Boolean(default=True)
+    cron_id = fields.Many2one("ir.cron", readonly=True, copy=False)
+    computation_method = fields.Selection(
+        [("function", "Function")], required=True
+    )
+    value = fields.Serialized()
+    dashboard_item_ids = fields.One2many("kpi.dashboard.item", inverse_name="kpi_id")
+    model_id = fields.Many2one("ir.model",)
+    function = fields.Char()
+    args = fields.Char()
+    kwargs = fields.Char()
+    widget = fields.Selection(
+        [("number", "Number"), ("meter", "Meter"), ("graph", "Graph")],
+        required=True,
+        default="number",
+    )
+    value_last_update = fields.Datetime(readonly=True)
+    prefix = fields.Char()
+    suffix = fields.Char()
+    action_ids = fields.One2many(
+        "kpi.kpi.action",
+        inverse_name='kpi_id',
+        help="Actions that can be opened from the KPI"
+    )
+
+    def _cron_vals(self):
+        return {
+            "name": self.name,
+            "model_id": self.env.ref("kpi_dashboard.model_kpi_kpi").id,
+            "interval_number": 1,
+            "interval_type": "hours",
+            "state": "code",
+            "code": "model.browse(%s).compute()" % self.id,
+            "active": True,
+        }
+
+    def compute(self):
+        for record in self:
+            record._compute()
+        return True
+
+    def _compute(self):
+        self.write(
+            {
+                "value": getattr(
+                    self, "_compute_value_%s" % self.computation_method
+                )()
+            }
+        )
+        notifications = []
+        for dashboard_item in self.dashboard_item_ids:
+            channel = "kpi_dashboard_%s" % dashboard_item.dashboard_id.id
+            notifications.append([channel, dashboard_item._read_dashboard()])
+        if notifications:
+            self.env["bus.bus"].sendmany(notifications)
+
+    def _compute_value_function(self):
+        obj = self
+        if self.model_id:
+            obj = self.env[self.model_id.model]
+        args = ast.literal_eval(self.args or "[]")
+        kwargs = ast.literal_eval(self.kwargs or "{}")
+        return getattr(obj, self.function)(*args, **kwargs)
+
+    def generate_cron(self):
+        self.ensure_one()
+        self.cron_id = self.env["ir.cron"].create(self._cron_vals())
+
+    @api.multi
+    def write(self, vals):
+        if "value" in vals:
+            vals["value_last_update"] = fields.Datetime.now()
+        return super().write(vals)
+
+
+class KpiKpiAction(models.Model):
+    _name = 'kpi.kpi.action'
+    _description = 'KPI action'
+
+    kpi_id = fields.Many2one('kpi.kpi', required=True, ondelete='cascade')
+    action = fields.Reference(
+        selection=[('ir.actions.report', 'ir.actions.report'),
+                   ('ir.actions.act_window', 'ir.actions.act_window'),
+                   ('ir.actions.act_url', 'ir.actions.act_url'),
+                   ('ir.actions.server', 'ir.actions.server'),
+                   ('ir.actions.client', 'ir.actions.client')],
+        required=True,
+    )
+
+    def read_dashboard(self):
+        result = []
+        for r in self:
+            result.append({
+                'id': r.action.id,
+                'type': r.action._name,
+                'name': r.action.name
+            })
+        return result