47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import shutil
|
|
|
|
from loguru import logger
|
|
|
|
from odoo_openupgrade_wizard.tools.tools_postgres import (
|
|
check_db_exist,
|
|
ensure_database,
|
|
)
|
|
|
|
|
|
def copydb(ctx, source, dest):
|
|
# check if source exists
|
|
logger.info(f"Check if source database '{source}' exists...")
|
|
check_db_exist(ctx, source, raise_exception=True)
|
|
|
|
# drop database if exist
|
|
ensure_database(ctx, dest, state="absent")
|
|
|
|
# Copy database
|
|
ensure_database(ctx, dest, state="present", template=source)
|
|
|
|
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}' directory...")
|
|
shutil.copytree(source_path, dest_path)
|
|
|
|
|
|
def dropdb(ctx, database):
|
|
"""Drop a database and its filestore"""
|
|
# Check if database exists
|
|
logger.info(f"Check if database '{database}' exists...")
|
|
check_db_exist(ctx, database, raise_exception=True)
|
|
# Drop database
|
|
logger.info(f"Drop database '{database}'...")
|
|
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)
|