Prevent exceptions because of prefetching non existing fields

Author: fkantelberg

setup.cfg

@@ -6,7 +6,7 @@ author_email = info@initos.com
 description = Management tool for Odoo installations
 long_description = file: README.md
 long_description_content_type = text/markdown
-license = Apache License 2.0
+license_file = file: LICENSE.md
 keywords = odoo environment management
 url = https://github.com/initos/dob-lib
 classifiers =

src/doblib/__init__.py

@@ -1 +1 @@
-VERSION = "0.20.4"
+VERSION = "0.20.5"

src/doblib/env.py

@@ -262,7 +262,7 @@ def odoo_version(self):
         return odoo.release.version_info
 
     @contextmanager
-    def env(self, db_name, rollback=False):
+    def env(self, db_name, rollback=False, minimal=False):
         """Create an environment from a registry"""
         # pylint: disable=C0415,E0401,W0611
         # ruff: noqa: F401
@@ -278,7 +278,19 @@ def env(self, db_name, rollback=False):
         # Get all installed modules
         reg = Registry(db_name)
         with closing(reg.cursor()) as cr:
-            yield Environment(cr, SUPERUSER_ID, {})
+            uid = SUPERUSER_ID
+            env = Environment(cr, uid, {})
+
+            if minimal:
+                # Prevent prefetching fields when creating a context
+                ctx = (
+                    Environment(cr, uid, {})["res.users"]
+                    .with_context(prefetch_fields=False)
+                    .context_get()
+                )
+                env = Environment(cr, uid, ctx)
+
+            yield env
 
             if rollback:
                 cr.rollback()

src/doblib/module.py

@@ -61,7 +61,7 @@ def _run_migration(self, db_name, script_name):
             sys.path = path
 
         utils.info(f"Executing {script.__name__.replace('_', ' ')} script")
-        with self.env(db_name) as env:
+        with self.env(db_name, minimal=True) as env:
             version = utils.Version(
                 env["ir.config_parameter"].get_param("db_version", False)
             )
@@ -85,7 +85,7 @@ def _run_migration_sql(self, db_name, script_name):
 
     def _get_installed_modules(self, db_name):
         """Return the list of modules which are installed"""
-        with self.env(db_name, False) as env:
+        with self.env(db_name, False, minimal=True) as env:
             domain = [("state", "=", "installed")]
             installed = env["ir.module.module"].search(domain).mapped("name")
             return set(installed).union(["base"])
@@ -120,7 +120,7 @@ def check_auto_install(self, db_name):
         """Install auto installable modules if the dependencies are installed"""
         states = frozenset(("installed", "to install", "to upgrade"))
 
-        with self.env(db_name, False) as env:
+        with self.env(db_name, False, minimal=True) as env:
             countries = env["res.company"].search([]).mapped("country_id")
 
             domain = [("state", "=", "uninstalled"), ("auto_install", "=", True)]
@@ -158,7 +158,7 @@ def check_auto_install(self, db_name):
 
     def update_checksums(self, db_name):
         """Only update the module checksums in the database"""
-        with self.env(db_name, False) as env:
+        with self.env(db_name, False, minimal=True) as env:
             model = env["ir.module.module"]
             if hasattr(model, "_save_installed_checksums"):
                 utils.info("Updating module checksums")
@@ -202,7 +202,7 @@ def update_specific(
     def update_changed(self, db_name, blacklist=None):
         """Update only changed modules"""
         utils.info("Updating changed modules")
-        with self.env(db_name, False) as env:
+        with self.env(db_name, False, minimal=True) as env:
             model = env["ir.module.module"]
             if hasattr(model, "upgrade_changed_checksum"):
                 # Initialize `res.company`, `res.partner` and `res.users` to prevent