153 lines
4.6 KiB
Python
153 lines
4.6 KiB
Python
from pathlib import Path
|
|
|
|
import docker
|
|
from loguru import logger
|
|
|
|
|
|
# WIP
|
|
def get_odoo_addons_path(ctx, odoo_version: dict, migration_step: dict) -> str:
|
|
pass
|
|
# repo_file = Path(
|
|
# self._current_directory,
|
|
# CUSTOMER_CONFIG_FOLDER,
|
|
# "repo_files",
|
|
# "%s.yml" % step["version"],
|
|
# )
|
|
# folder = Path(self._current_directory, step["local_path"])
|
|
# base_module_folder = get_base_module_folder(step)
|
|
# stream = open(repo_file, "r")
|
|
# data = yaml.safe_load(stream)
|
|
|
|
# addons_path = []
|
|
# for key in data.keys():
|
|
# path = os.path.join(folder, key)
|
|
# if path.endswith(get_odoo_folder(step)):
|
|
# # Add two folder for odoo folder
|
|
# addons_path.append(os.path.join(path, "addons"))
|
|
# addons_path.append(
|
|
# os.path.join(path, base_module_folder, "addons")
|
|
# )
|
|
# elif skip_path(step, path):
|
|
# pass
|
|
# else:
|
|
# addons_path.append(path)
|
|
|
|
# return ",".join(addons_path)
|
|
|
|
|
|
def get_odoo_env_path(ctx, odoo_version: dict) -> Path:
|
|
folder_name = "env_%s" % str(odoo_version["release"]).rjust(4, "0")
|
|
return ctx.obj["src_folder_path"] / folder_name
|
|
|
|
|
|
def get_docker_image_tag(ctx, odoo_version: dict) -> str:
|
|
"""Return a docker image tag, based on project name and odoo release"""
|
|
return "odoo-openupgrade-wizard-image-%s-%s" % (
|
|
ctx.obj["config"]["project_name"],
|
|
str(odoo_version["release"]).rjust(4, "0"),
|
|
)
|
|
|
|
|
|
def get_docker_container_name(ctx, migration_step: dict) -> str:
|
|
"""Return a docker container name, based on project name,
|
|
odoo release and migration step"""
|
|
return "odoo-openupgrade-wizard-container-%s-%s-step-%s" % (
|
|
ctx.obj["config"]["project_name"],
|
|
str(migration_step["release"]).rjust(4, "0"),
|
|
str(migration_step["name"]).rjust(2, "0"),
|
|
)
|
|
|
|
|
|
def get_odoo_version_from_migration_step(ctx, migration_step: dict) -> dict:
|
|
for odoo_version in ctx.obj["config"]["odoo_versions"]:
|
|
if odoo_version["release"] == migration_step["release"]:
|
|
return odoo_version
|
|
# TODO, improve exception
|
|
raise Exception
|
|
|
|
|
|
def generate_odoo_command(
|
|
ctx,
|
|
migration_step: dict,
|
|
database: str,
|
|
update_all: bool,
|
|
stop_after_init: bool,
|
|
shell: bool,
|
|
) -> str:
|
|
# TODO, make it dynamic
|
|
addons_path = (
|
|
"/container_env/src/odoo/addons," "/container_env/src/odoo/odoo/addons"
|
|
)
|
|
database_cmd = database and "--database %s" % database or ""
|
|
update_all_cmd = update_all and "--update_all" or ""
|
|
stop_after_init_cmd = stop_after_init and "-- stop-after-init" or ""
|
|
shell_cmd = shell and "shell" or ""
|
|
return (
|
|
f"/container_env/src/odoo/odoo-bin"
|
|
f" --db_host db"
|
|
f" --db_port 5432"
|
|
f" --db_user odoo"
|
|
f" --db_password odoo"
|
|
f" --workers 0"
|
|
f" --addons-path {addons_path}"
|
|
f" {database_cmd}"
|
|
f" {update_all_cmd}"
|
|
f" {stop_after_init_cmd}"
|
|
f" {shell_cmd}"
|
|
)
|
|
|
|
|
|
def run_odoo(
|
|
ctx,
|
|
migration_step: dict,
|
|
database: str = False,
|
|
stop_after_init: bool = False,
|
|
shell: bool = False,
|
|
update_all: bool = False,
|
|
):
|
|
client = docker.from_env()
|
|
odoo_version = get_odoo_version_from_migration_step(ctx, migration_step)
|
|
folder_path = get_odoo_env_path(ctx, odoo_version)
|
|
|
|
command = generate_odoo_command(
|
|
ctx,
|
|
migration_step,
|
|
database=database,
|
|
stop_after_init=stop_after_init,
|
|
shell=shell,
|
|
update_all=update_all,
|
|
)
|
|
|
|
image_name = get_docker_image_tag(ctx, odoo_version)
|
|
container_name = get_docker_container_name(ctx, migration_step)
|
|
logger.info(
|
|
"Launching Odoo Docker container named %s based on image '%s'."
|
|
% (container_name, image_name)
|
|
)
|
|
container = client.containers.run(
|
|
image_name,
|
|
name=container_name,
|
|
command=command,
|
|
ports={"8069": 8069, "5432": 5432},
|
|
volumes=["%s:/container_env/" % (folder_path)],
|
|
links={"db": "db"},
|
|
detach=True,
|
|
auto_remove=True,
|
|
)
|
|
logger.info("Container Launched. Command executed : %s" % command)
|
|
return container
|
|
|
|
|
|
def kill_odoo(ctx, migration_step: dict):
|
|
client = docker.from_env()
|
|
containers = client.containers.list(
|
|
all=True,
|
|
filters={"name": get_docker_container_name(ctx, migration_step)},
|
|
)
|
|
for container in containers:
|
|
logger.info(
|
|
"Stop container %s, based on image '%s'."
|
|
% (container.name, ",".join(container.image.tags))
|
|
)
|
|
container.stop()
|