42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import shutil
|
|
|
|
from loguru import logger
|
|
|
|
from odoo_openupgrade_wizard.tools.tools_postgres import (
|
|
ensure_database,
|
|
execute_sql_request,
|
|
)
|
|
|
|
|
|
def copydb(ctx, source, dest):
|
|
# drop database if exist
|
|
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)
|
|
|
|
main_path = ctx.obj["filestore_folder_path"] / "filestore"
|
|
source_path = main_path / source
|
|
dest_path = main_path / dest
|
|
# Drop filestore if exist
|
|
logger.info(f"Remove filestore of '{dest}' if exists.")
|
|
shutil.rmtree(dest_path, ignore_errors=True)
|
|
|
|
# Copy Filestore
|
|
logger.info(f"Copy filestore of '{source}' into '{dest}' folder ...")
|
|
shutil.copytree(source_path, dest_path)
|
|
|
|
|
|
def dropdb(ctx, database):
|
|
"""Drop a database and its filestore"""
|
|
# Drop database
|
|
logger.info(f"Drop database '{database}' if it exists...")
|
|
ensure_database(ctx, database, state="absent")
|
|
# Drop filestore
|
|
root_filestore_path = ctx.obj["filestore_folder_path"] / "filestore"
|
|
filestore_path = root_filestore_path / database
|
|
logger.info(f"Remove filestore of '{database}' if exists...")
|
|
shutil.rmtree(filestore_path, ignore_errors=True)
|