mirror of
https://github.com/frappe/frappe_docker.git
synced 2024-11-09 23:00:56 +00:00
133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
import os
|
|
|
|
import frappe
|
|
import semantic_version
|
|
from constants import COMMON_SITE_CONFIG_FILE, RDS_DB, RDS_PRIVILEGES
|
|
from frappe.installer import update_site_config
|
|
from utils import get_config, get_password, get_site_config, run_command
|
|
|
|
# try to import _new_site from frappe, which could possibly
|
|
# exist in either commands.py or installer.py, and so we need
|
|
# to maintain compatibility across all frappe versions.
|
|
try:
|
|
# <= version-{11,12}
|
|
from frappe.commands.site import _new_site
|
|
except ImportError:
|
|
# >= version-13 and develop
|
|
from frappe.installer import _new_site
|
|
|
|
|
|
def main():
|
|
config = get_config()
|
|
db_type = "mariadb"
|
|
db_port = config.get("db_port", 3306)
|
|
db_host = config.get("db_host")
|
|
site_name = os.environ.get("SITE_NAME", "site1.localhost")
|
|
db_root_username = os.environ.get("DB_ROOT_USER", "root")
|
|
mariadb_root_password = get_password("MYSQL_ROOT_PASSWORD", "admin")
|
|
postgres_root_password = get_password("POSTGRES_PASSWORD")
|
|
db_root_password = mariadb_root_password
|
|
|
|
if postgres_root_password:
|
|
db_type = "postgres"
|
|
db_host = os.environ.get("POSTGRES_HOST")
|
|
db_port = 5432
|
|
db_root_password = postgres_root_password
|
|
if not db_host:
|
|
db_host = config.get("db_host")
|
|
print("Environment variable POSTGRES_HOST not found.")
|
|
print("Using db_host from common_site_config.json")
|
|
|
|
sites_path = os.getcwd()
|
|
common_site_config_path = os.path.join(sites_path, COMMON_SITE_CONFIG_FILE)
|
|
update_site_config(
|
|
"root_login",
|
|
db_root_username,
|
|
validate=False,
|
|
site_config_path=common_site_config_path,
|
|
)
|
|
update_site_config(
|
|
"root_password",
|
|
db_root_password,
|
|
validate=False,
|
|
site_config_path=common_site_config_path,
|
|
)
|
|
|
|
force = True if os.environ.get("FORCE", None) else False
|
|
install_apps = os.environ.get("INSTALL_APPS", None)
|
|
install_apps = install_apps.split(",") if install_apps else []
|
|
frappe.init(site_name, new_site=True)
|
|
|
|
if semantic_version.Version(frappe.__version__).major > 11:
|
|
_new_site(
|
|
None,
|
|
site_name,
|
|
mariadb_root_username=db_root_username,
|
|
mariadb_root_password=db_root_password,
|
|
admin_password=get_password("ADMIN_PASSWORD", "admin"),
|
|
verbose=True,
|
|
install_apps=install_apps,
|
|
source_sql=None,
|
|
force=force,
|
|
db_type=db_type,
|
|
reinstall=False,
|
|
db_host=db_host,
|
|
db_port=db_port,
|
|
)
|
|
else:
|
|
_new_site(
|
|
None,
|
|
site_name,
|
|
mariadb_root_username=db_root_username,
|
|
mariadb_root_password=db_root_password,
|
|
admin_password=get_password("ADMIN_PASSWORD", "admin"),
|
|
verbose=True,
|
|
install_apps=install_apps,
|
|
source_sql=None,
|
|
force=force,
|
|
reinstall=False,
|
|
)
|
|
|
|
if db_type == "mariadb":
|
|
site_config = get_site_config(site_name)
|
|
db_name = site_config.get("db_name")
|
|
db_password = site_config.get("db_password")
|
|
|
|
mysql_command = [
|
|
"mysql",
|
|
f"-h{db_host}",
|
|
f"-u{db_root_username}",
|
|
f"-p{mariadb_root_password}",
|
|
"-e",
|
|
]
|
|
|
|
# Drop User if exists
|
|
command = mysql_command + [
|
|
f"DROP USER IF EXISTS '{db_name}'; FLUSH PRIVILEGES;"
|
|
]
|
|
run_command(command)
|
|
|
|
# Grant permission to database and set password
|
|
grant_privileges = "ALL PRIVILEGES"
|
|
|
|
# for Amazon RDS
|
|
if config.get(RDS_DB) or site_config.get(RDS_DB):
|
|
grant_privileges = RDS_PRIVILEGES
|
|
|
|
command = mysql_command + [
|
|
f"\
|
|
CREATE USER IF NOT EXISTS '{db_name}'@'%' IDENTIFIED BY '{db_password}'; \
|
|
GRANT {grant_privileges} ON `{db_name}`.* TO '{db_name}'@'%'; \
|
|
FLUSH PRIVILEGES;"
|
|
]
|
|
run_command(command)
|
|
|
|
if frappe.redis_server:
|
|
frappe.redis_server.connection_pool.disconnect()
|
|
|
|
exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|