65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
# Database Module
|
|
# --------------------
|
|
|
|
from frappe.database.database import savepoint
|
|
|
|
|
|
def setup_database(force, source_sql=None, verbose=None, no_mariadb_socket=False):
|
|
import frappe
|
|
|
|
if frappe.conf.db_type == "postgres":
|
|
import frappe.database.postgres.setup_db
|
|
|
|
return frappe.database.postgres.setup_db.setup_database(force, source_sql, verbose)
|
|
else:
|
|
import frappe.database.mariadb.setup_db
|
|
|
|
return frappe.database.mariadb.setup_db.setup_database(
|
|
force, source_sql, verbose, no_mariadb_socket=no_mariadb_socket
|
|
)
|
|
|
|
|
|
def bootstrap_database(db_name, verbose=None, source_sql=None):
|
|
import frappe
|
|
|
|
if frappe.conf.db_type == "postgres":
|
|
import frappe.database.postgres.setup_db
|
|
|
|
return frappe.database.postgres.setup_db.bootstrap_database(db_name, verbose, source_sql)
|
|
else:
|
|
import frappe.database.mariadb.setup_db
|
|
|
|
return frappe.database.mariadb.setup_db.bootstrap_database(db_name, verbose, source_sql)
|
|
|
|
|
|
def drop_user_and_database(db_name, root_login=None, root_password=None):
|
|
import frappe
|
|
|
|
if frappe.conf.db_type == "postgres":
|
|
import frappe.database.postgres.setup_db
|
|
|
|
return frappe.database.postgres.setup_db.drop_user_and_database(
|
|
db_name, root_login, root_password
|
|
)
|
|
else:
|
|
import frappe.database.mariadb.setup_db
|
|
|
|
return frappe.database.mariadb.setup_db.drop_user_and_database(
|
|
db_name, root_login, root_password
|
|
)
|
|
|
|
|
|
def get_db(host=None, user=None, password=None, port=None):
|
|
import frappe
|
|
|
|
if frappe.conf.db_type == "postgres":
|
|
import frappe.database.postgres.database
|
|
|
|
return frappe.database.postgres.database.PostgresDatabase(host, user, password, port=port)
|
|
else:
|
|
import frappe.database.mariadb.database
|
|
|
|
return frappe.database.mariadb.database.MariaDBDatabase(host, user, password, port=port)
|