6b2bb9a2ab
This avoids having to manipulate config files in brittle bash entrypoints that need to react to dynamic service discovery. This significantly improves the operability of various bench sites.
112 lines
3.2 KiB
Python
112 lines
3.2 KiB
Python
import os
|
|
|
|
import frappe
|
|
|
|
|
|
def setup_database(force, source_sql=None, verbose=False):
|
|
root_conn = get_root_connection(frappe.flags.root_login, frappe.flags.root_password)
|
|
root_conn.commit()
|
|
root_conn.sql("end")
|
|
root_conn.sql(f"DROP DATABASE IF EXISTS `{frappe.conf.db_name}`")
|
|
root_conn.sql(f"DROP USER IF EXISTS {frappe.conf.db_name}")
|
|
root_conn.sql(f"CREATE DATABASE `{frappe.conf.db_name}`")
|
|
root_conn.sql(f"CREATE user {frappe.conf.db_name} password '{frappe.conf.db_password}'")
|
|
root_conn.sql("GRANT ALL PRIVILEGES ON DATABASE `{0}` TO {0}".format(frappe.conf.db_name))
|
|
root_conn.close()
|
|
|
|
bootstrap_database(frappe.conf.db_name, verbose, source_sql=source_sql)
|
|
frappe.connect()
|
|
|
|
|
|
def bootstrap_database(db_name, verbose, source_sql=None):
|
|
frappe.connect(db_name=db_name)
|
|
import_db_from_sql(source_sql, verbose)
|
|
frappe.connect(db_name=db_name)
|
|
|
|
if "tabDefaultValue" not in frappe.db.get_tables():
|
|
import sys
|
|
|
|
from click import secho
|
|
|
|
secho(
|
|
"Table 'tabDefaultValue' missing in the restored site. "
|
|
"This may be due to incorrect permissions or the result of a restore from a bad backup file. "
|
|
"Database not installed correctly.",
|
|
fg="red",
|
|
)
|
|
sys.exit(1)
|
|
|
|
|
|
def import_db_from_sql(source_sql=None, verbose=False):
|
|
from shutil import which
|
|
from subprocess import PIPE, run
|
|
|
|
# we can't pass psql password in arguments in postgresql as mysql. So
|
|
# set password connection parameter in environment variable
|
|
subprocess_env = os.environ.copy()
|
|
subprocess_env["PGPASSWORD"] = str(frappe.conf.db_password)
|
|
|
|
# bootstrap db
|
|
if not source_sql:
|
|
source_sql = os.path.join(os.path.dirname(__file__), "framework_postgres.sql")
|
|
|
|
pv = which("pv")
|
|
|
|
_command = (
|
|
f"psql {frappe.conf.db_name} "
|
|
f"-h {frappe.conf.db_host} -p {str(frappe.conf.db_port)} "
|
|
f"-U {frappe.conf.db_name}"
|
|
)
|
|
|
|
if pv:
|
|
command = f"{pv} {source_sql} | " + _command
|
|
else:
|
|
command = _command + f" -f {source_sql}"
|
|
|
|
print("Restoring Database file...")
|
|
if verbose:
|
|
print(command)
|
|
|
|
restore_proc = run(command, env=subprocess_env, shell=True, stdout=PIPE)
|
|
|
|
if verbose:
|
|
print(
|
|
f"\nSTDOUT by psql:\n{restore_proc.stdout.decode()}\nImported from Database File: {source_sql}"
|
|
)
|
|
|
|
|
|
def get_root_connection(root_login=None, root_password=None):
|
|
if not frappe.local.flags.root_connection:
|
|
if not root_login:
|
|
root_login = frappe.conf.get("root_login") or None
|
|
|
|
if not root_login:
|
|
root_login = input("Enter postgres super user: ")
|
|
|
|
if not root_password:
|
|
root_password = frappe.conf.get("root_password") or None
|
|
|
|
if not root_password:
|
|
from getpass import getpass
|
|
|
|
root_password = getpass("Postgres super user password: ")
|
|
|
|
frappe.local.flags.root_connection = frappe.database.get_db(
|
|
user=root_login, password=root_password
|
|
)
|
|
|
|
return frappe.local.flags.root_connection
|
|
|
|
|
|
def drop_user_and_database(db_name, root_login, root_password):
|
|
root_conn = get_root_connection(
|
|
frappe.flags.root_login or root_login, frappe.flags.root_password or root_password
|
|
)
|
|
root_conn.commit()
|
|
root_conn.sql(
|
|
"SELECT pg_terminate_backend (pg_stat_activity.pid) FROM pg_stat_activity WHERE pg_stat_activity.datname = %s",
|
|
(db_name,),
|
|
)
|
|
root_conn.sql("end")
|
|
root_conn.sql(f"DROP DATABASE IF EXISTS {db_name}")
|
|
root_conn.sql(f"DROP USER IF EXISTS {db_name}")
|