88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
import csv
|
|
|
|
import click
|
|
from loguru import logger
|
|
|
|
from odoo_openupgrade_wizard.cli_options import (
|
|
database_option,
|
|
get_migration_step_from_options,
|
|
)
|
|
from odoo_openupgrade_wizard.tools_odoo import kill_odoo, run_odoo
|
|
from odoo_openupgrade_wizard.tools_odoo_instance import OdooInstance
|
|
from odoo_openupgrade_wizard.tools_postgres import ensure_database
|
|
|
|
|
|
@click.command()
|
|
@database_option
|
|
@click.pass_context
|
|
def install_from_csv(ctx, database):
|
|
migration_step = get_migration_step_from_options(ctx, 1)
|
|
ensure_database(database, state="present")
|
|
|
|
# Get modules list from the CSV file
|
|
csv_path = ctx.obj["module_file_path"]
|
|
logger.info("Reading '%s' file ..." % csv_path)
|
|
module_names = []
|
|
csvfile = open(csv_path, "r")
|
|
spamreader = csv.reader(csvfile, delimiter=",", quotechar='"')
|
|
for row in spamreader:
|
|
# Try to guess that a line is not correct
|
|
if not row[0]:
|
|
continue
|
|
if " " in row[0]:
|
|
continue
|
|
if any([x.isupper() for x in row[0]]):
|
|
continue
|
|
module_names.append(row[0])
|
|
|
|
module_names.sort()
|
|
logger.info("Found %d modules." % (len(module_names)))
|
|
logger.debug(module_names)
|
|
|
|
try:
|
|
logger.info("Install 'base' module on %s database ..." % (database))
|
|
run_odoo(
|
|
ctx,
|
|
migration_step,
|
|
database=database,
|
|
detached_container=True,
|
|
init="base",
|
|
)
|
|
odoo_instance = OdooInstance(ctx, database)
|
|
|
|
default_country_code = ctx.obj["config"].get(
|
|
"default_country_code", False
|
|
)
|
|
if "account" in module_names:
|
|
# Then, set correct country to the company of the current user
|
|
# Otherwise, due to poor design of Odoo, when installing account
|
|
# the US localization will be installed.
|
|
# (l10n_us + l10n_generic_coa)
|
|
|
|
countries = odoo_instance.browse_by_search(
|
|
"res.country",
|
|
[("code", "=", default_country_code)],
|
|
)
|
|
if len(countries) != 1:
|
|
raise Exception(
|
|
"Unable to find a country, based on the code %s."
|
|
" countries found : %s "
|
|
% (
|
|
default_country_code,
|
|
", ".join([x.name for x in countries]),
|
|
)
|
|
)
|
|
logger.info(
|
|
"Configuring country of the main company with #%d - %s"
|
|
% (countries[0].id, countries[0].name)
|
|
)
|
|
odoo_instance.env.user.company_id.country_id = countries[0].id
|
|
|
|
# Install modules
|
|
odoo_instance.install_modules(module_names)
|
|
|
|
except (KeyboardInterrupt, SystemExit):
|
|
logger.info("Received Keyboard Interrupt or System Exiting...")
|
|
finally:
|
|
kill_odoo(ctx, migration_step)
|