Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-server-tools-database_cleanup/database_cleanup/models/purge_data.py

@@ -0,0 +1,92 @@
+# Copyright 2014-2016 Therp BV <http://therp.nl>
+# Copyright 2021 Camptocamp <https://camptocamp.com>
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+from odoo import _, api, fields, models
+from odoo.exceptions import UserError
+
+from ..identifier_adapter import IdentifierAdapter
+
+
+class CleanupPurgeLineData(models.TransientModel):
+    _inherit = "cleanup.purge.line"
+    _name = "cleanup.purge.line.data"
+    _description = "Cleanup Purge Line Data"
+
+    data_id = fields.Many2one("ir.model.data", "Data entry")
+    wizard_id = fields.Many2one(
+        "cleanup.purge.wizard.data", "Purge Wizard", readonly=True
+    )
+
+    def purge(self):
+        """Unlink data entries upon manual confirmation."""
+        if self:
+            objs = self
+        else:
+            objs = self.env["cleanup.purge.line.data"].browse(
+                self._context.get("active_ids")
+            )
+        to_unlink = objs.filtered(lambda x: not x.purged and x.data_id)
+        self.logger.info("Purging data entries: %s", to_unlink.mapped("name"))
+        to_unlink.mapped("data_id").unlink()
+        return to_unlink.write({"purged": True})
+
+
+class CleanupPurgeWizardData(models.TransientModel):
+    _inherit = "cleanup.purge.wizard"
+    _name = "cleanup.purge.wizard.data"
+    _description = "Purge data"
+
+    @api.model
+    def find(self):
+        """Collect all rows from ir_model_data that refer
+        to a nonexisting model, or to a nonexisting
+        row in the model's table."""
+        res = []
+        data_ids = []
+        unknown_models = []
+        self.env.cr.execute("""SELECT DISTINCT(model) FROM ir_model_data""")
+        for (model,) in self.env.cr.fetchall():
+            if not model:
+                continue
+            if model not in self.env:
+                unknown_models.append(model)
+                continue
+            self.env.cr.execute(
+                """
+                SELECT id FROM ir_model_data
+                WHERE model = %s
+                AND res_id IS NOT NULL
+                AND NOT EXISTS (
+                    SELECT id FROM %s WHERE id=ir_model_data.res_id)
+                """,
+                (model, IdentifierAdapter(self.env[model]._table)),
+            )
+            data_ids.extend(data_row for data_row, in self.env.cr.fetchall())
+        data_ids += (
+            self.env["ir.model.data"]
+            .search(
+                [
+                    ("model", "in", unknown_models),
+                ]
+            )
+            .ids
+        )
+        for data in self.env["ir.model.data"].browse(data_ids):
+            res.append(
+                (
+                    0,
+                    0,
+                    {
+                        "data_id": data.id,
+                        "name": "%s.%s, object of type %s"
+                        % (data.module, data.name, data.model),
+                    },
+                )
+            )
+        if not res:
+            raise UserError(_("No orphaned data entries found"))
+        return res
+
+    purge_line_ids = fields.One2many(
+        "cleanup.purge.line.data", "wizard_id", "Data to purge"
+    )