fixup

polars_db_process/__manifest__.py

@@ -22,11 +22,12 @@
     "data": [
         "security/ir.model.access.xml",
         "wizards/df_process.xml",
-        "views/dataframe.xml",
+        "views/model_map.xml",
         "views/df_field.xml",
         "views/df_source.xml",
         "views/db_config.xml",
-        "data/demo.xml",
+        "data/action.xml",
+        "data/demo.xml",  # TODO remove
     ],
     "demo": [
         "data/demo.xml",

polars_db_process/data/action.xml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<odoo>
+    <record id="remove_related_uidstring_record" model="ir.actions.server">
+        <field name="name">Remove ir_model_data created from there</field>
+        <field name="model_id" ref="polars_db_process.model_model_map" />
+        <field name="binding_model_id" ref="polars_db_process.model_model_map" />
+        <field name="state">code</field>
+        <field name="code">env['model.map']._remove_related_uidstring_record()</field>
+    </record>
+</odoo>

polars_db_process/data/demo.xml

@@ -6,7 +6,7 @@
         >sqlite://.../polars_db_schema/tests/files/chinook.sqlite</field>
     </record>
 
-    <record id="contact_chinook" model="dataframe">
+    <record id="contact_chinook" model="model.map">
         <field name="model_id" ref="base.model_res_partner" />
         <field name="code">Chinook Customers</field>
     </record>

polars_db_process/models/__init__.py

@@ -1,2 +1,3 @@
 from . import df_source
 from . import db_config
+from . import model_map

polars_db_process/models/db_config.py

@@ -42,3 +42,6 @@ def _read_sql(self, query):
             raise exceptions.ValidationError(err) from err
         except Exception as err:
             raise exceptions.ValidationError(err) from err
+
+    def _set_uidstring_module_name(self):
+        return "polars"

polars_db_process/models/df_source.py

@@ -8,7 +8,7 @@
 
 HELP = """Supported files: .xlsx and .sql
 Sql files may contains a comment on first line
-to be mapped automatically with dataframe, i.e:\n
+to be mapped automatically with model_map, i.e:\n
 -- {'model_id': 'product.product', 'db_conf_id': mydb}
 -- {'code': 'my_delivery_address', 'db_conf_id': mydb}
 """
@@ -24,7 +24,7 @@ class DfSource(models.Model):
     )
 
     def _file_hook(self, file):
-        "Map sql file with the right Odoo model via dataframe and the right db.config"
+        "Map sql file with the right Odoo model via model_map and the right db.config"
         vals = super()._file_hook(file)
         if ".sql" in file:
             # TODO: improve
@@ -36,31 +36,31 @@ def _file_hook(self, file):
                 model_name = metadata.get("model")
                 model = self.env["ir.model"].search([("model", "=", model_name)])
                 if model_name:
-                    # we don't want to use these dataframes
-                    dataframes = (
+                    # we don't want to use these model_maps
+                    model_maps = (
                         self.env["df.source"]
                         .search([])
                         .filtered(lambda s: not s.db_conf_id)
-                        .mapped("dataframe_id")
+                        .mapped("model_map_id")
                     )
-                    dataframe = self.env["dataframe"].search(
+                    model_map = self.env["model.map"].search(
                         [
-                            ("id", "not in", dataframes.ids),
+                            ("id", "not in", model_maps.ids),
                             ("model_id", "=", model_name),
                         ]
                     )
-                    if dataframe:
+                    if model_map:
                         # TODO use first
-                        vals["dataframe_id"] = dataframe[0].id
+                        vals["model_map_id"] = model_map[0].id
                         db_config = self.env["db.config"].search(
                             [("name", "ilike", metadata.get("db_conf_id"))]
                         )
                         vals["db_conf_id"] = db_config and db_config[0].id or False
                     else:
-                        df = self.env["dataframe"].create(
+                        df = self.env["model.map"].create(
                             {"code": model.name, "model_id": model and model[0].id}
                         )
-                        vals["dataframe_id"] = df.id
+                        vals["model_map_id"] = df.id
             vals["query"] = content
         return vals
 

polars_db_process/models/model_map.py

@@ -0,0 +1,20 @@
+from odoo import models
+
+
+class ModelMap(models.Model):
+    _inherit = "model.map"
+
+    def _remove_related_uidstring_record(self):
+        self = self.browse(self.env.context.get("active_ids"))
+        self = self and self[0] or False
+        res = self.env["ir.model.data"].search(
+            [
+                ("model", "=", self.model_id.model),
+                ("module", "=", self._get_uidstring_module_name()),
+            ]
+        )
+        res.reference.unlink()
+        return True
+
+    def _get_uidstring_module_name(self):
+        return "polars"

polars_db_process/views/db_config.xml

@@ -34,7 +34,7 @@
         </field>
     </record>
 
-    <record id="dataframe_list" model="ir.ui.view">
+    <record id="model_map_list" model="ir.ui.view">
         <field name="model">db.config</field>
         <field name="arch" type="xml">
             <list>

polars_db_process/wizards/df_process.py

@@ -1,25 +1,50 @@
+import logging
+
 from odoo import _, exceptions, models
 
+logger = logging.getLogger(__name__)
+
 
 class DfProcessWiz(models.TransientModel):
     _inherit = "df.process.wiz"
 
     def _pre_process(self):
         res = super()._pre_process()
         if not self.file:
-            self._pre_process_sql()
+            self._pre_process_query()
         return res
 
-    def _pre_process_sql(self):
+    def _pre_process_query(self):
         "You may inherit to set your own behavior"
         if not self.df_source_id.db_conf_id:
             raise exceptions.ValidationError(
                 _("Missing database configuration in your df source ")
             )
-        self._process_sql()
+        self._process_query()
 
-    def _process_sql(self):
+    def _process_query(self):
         self.ensure_one()
         df = self.df_source_id.db_conf_id._read_sql(self.df_source_id.query)
-        if self.dataframe_id:
-            self.env[self.dataframe_id.model_id.model].create(df.to_dicts())
+        if self.model_map_id:
+            model = self.model_map_id.model_id.model
+            vals_list = df.to_dicts()
+            mapper = {}
+            for vals in vals_list:
+                uidstring = vals.pop("id")
+                if "parent_id" in vals:
+                    vals["parent_id"] = mapper.get(vals["parent_id"])
+                rec = self.env[model].create(vals)
+                mapper[uidstring] = rec.id
+                logger.info("  >>>  ", vals)
+                self._set_uidstring(uidstring, rec, model)
+
+    def _set_uidstring(self, uidstring, record, model):
+        self.env["ir.model.data"].create(
+            {
+                "res_id": record.id,
+                "model": model,
+                "module": self.model_map_id._get_uidstring_module_name(),
+                "name": uidstring,
+                "noupdate": False,
+            }
+        )