chore: Move files from build/common

build/frappe-worker/commands/new.py

@@ -0,0 +1,118 @@
+import os
+import frappe
+import semantic_version
+
+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,
+)
+
+# 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()