19.0 vanilla

odoo-bringout-oca-ocb-base/odoo/cli/populate.py

@@ -1,15 +1,14 @@
-# Part of Odoo. See LICENSE file for full copyright and licensing details.
 import logging
 import optparse
 import sys
 import time
-from pathlib import Path
+
+from odoo import api
+from odoo.modules.registry import Registry
+from odoo.tools import config
+from odoo.tools.populate import populate_models
 
 from . import Command
-import odoo
-from odoo.modules.registry import Registry
-from odoo.tools.populate import populate_models
-from odoo.api import Environment
 
 DEFAULT_FACTOR = '10000'
 DEFAULT_SEPARATOR = '_'
@@ -22,8 +21,8 @@ class Populate(Command):
     """Populate database via duplication of existing data for testing/demo purposes"""
 
     def run(self, cmdargs):
-        parser = odoo.tools.config.parser
-        parser.prog = f'{Path(sys.argv[0]).name} {self.name}'
+        parser = config.parser
+        parser.prog = self.prog
         group = optparse.OptionGroup(parser, "Populate Configuration")
         group.add_option("--factors", dest="factors",
                          help="Comma separated list of factors for each model, or just a single factor."
@@ -39,7 +38,7 @@ class Populate(Command):
                          help="Single character separator for char/text fields.",
                          default=DEFAULT_SEPARATOR)
         parser.add_option_group(group)
-        opt = odoo.tools.config.parse_config(cmdargs, setup_logging=True)
+        opt = config.parse_config(cmdargs, setup_logging=True)
 
         # deduplicate models if necessary, and keep the last corresponding
         # factor for each model
@@ -53,14 +52,16 @@ class Populate(Command):
         except TypeError:
             raise ValueError("Separator must be a single Unicode character.")
 
-        dbname = odoo.tools.config['db_name']
-        registry = Registry(dbname)
+        dbnames = config['db_name']
+        if len(dbnames) > 1:
+            sys.exit("-d/--database/db_name has multiple database, please provide a single one")
+        registry = Registry(dbnames[0])
         with registry.cursor() as cr:
-            env = odoo.api.Environment(cr, odoo.SUPERUSER_ID, {'active_test': False})
+            env = api.Environment(cr, api.SUPERUSER_ID, {'active_test': False})
             self.populate(env, model_factors, separator_code)
 
     @classmethod
-    def populate(cls, env: Environment, modelname_factors: dict[str, int], separator_code: int):
+    def populate(cls, env: api.Environment, modelname_factors: dict[str, int], separator_code: int):
         model_factors = {
             model: factor
             for model_name, factor in modelname_factors.items()