mirror of
https://github.com/frappe/frappe_docker.git
synced 2024-09-21 03:29:01 +00:00
115 lines
3.9 KiB
Python
115 lines
3.9 KiB
Python
import os
|
|
import frappe
|
|
import semantic_version
|
|
|
|
from frappe.commands.site import _new_site
|
|
from frappe.installer import update_site_config
|
|
from constants import COMMON_SITE_CONFIG_FILE, RDS_DB, RDS_PRIVILEGES
|
|
from utils import (
|
|
run_command,
|
|
get_config,
|
|
get_site_config,
|
|
get_password,
|
|
)
|
|
|
|
|
|
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{mariadb_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)
|
|
|
|
# update User's host to '%' required to connect from any container
|
|
command = mysql_command + [f"UPDATE mysql.user SET Host = '%' where User = '{db_name}'; FLUSH PRIVILEGES;"]
|
|
run_command(command)
|
|
|
|
# Set db password
|
|
command = mysql_command + [f"ALTER USER '{db_name}'@'%' IDENTIFIED BY '{db_password}'; FLUSH PRIVILEGES;"]
|
|
run_command(command)
|
|
|
|
# Grant permission to database
|
|
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"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()
|