[REF] move postgres code into dedicated tools file

This commit is contained in:
Sylvain LE GAL 2024-03-23 10:05:34 +01:00
parent e5b0ebe6e8
commit 9e9ac18a17
2 changed files with 10 additions and 13 deletions

View File

@ -2,10 +2,7 @@ import shutil
from loguru import logger
from odoo_openupgrade_wizard.tools.tools_postgres import (
ensure_database,
execute_sql_request,
)
from odoo_openupgrade_wizard.tools.tools_postgres import ensure_database
def copydb(ctx, source, dest):
@ -13,9 +10,7 @@ def copydb(ctx, source, dest):
ensure_database(ctx, dest, state="absent")
# Copy database
logger.info(f"Copy database {source} into {dest} ...")
request = f"CREATE DATABASE {dest} WITH TEMPLATE {source};"
execute_sql_request(ctx, request)
ensure_database(ctx, dest, state="present", template=source)
main_path = ctx.obj["filestore_folder_path"] / "filestore"
source_path = main_path / source

View File

@ -137,7 +137,7 @@ def execute_psql_command(
return docker_result.output.decode("utf-8")
def ensure_database(ctx, database: str, state="present"):
def ensure_database(ctx, database: str, state="present", template: str = ""):
"""
- Connect to postgres container.
- Check if the database exist.
@ -152,11 +152,13 @@ def ensure_database(ctx, database: str, state="present"):
if [database] in result:
return
logger.info("Create database '%s' ..." % database)
request = "CREATE DATABASE {database} owner odoo;".format(
database=database
)
execute_sql_request(ctx, request)
if template:
logger.info(f"Copy database '{template}' into '{database}' ...")
request = f"CREATE DATABASE {database} WITH TEMPLATE {template};"
else:
logger.info(f"Create database '{database}' ...")
request = f"CREATE DATABASE {database} owner odoo;"
else:
if [database] not in result:
return