vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

fix: naming and provision to use same db credentails for secondary

Browse source
This commit is contained in:
Saurabh 2019-05-03 18:17:04 +05:30
parent de58a59402
commit 5e284db206
1 changed files with 14 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
frappe/__init__.py
View file

@ -187,14 +187,19 @@ def connect(site=None, db_name=None):
local.db = get_db(user=db_name or local.conf.db_name)
set_user("Administrator")
def connect_read_only():
def connect_secondary():
from frappe.database import get_db
user = local.conf.db_name
password = local.conf.db_password
local.read_only_db = get_db(host=local.conf.slave_host, user=local.conf.slave_db_name,
password=local.conf.slave_db_password)
if local.conf.different_credentials_for_secondary:
user = local.conf.secondary_db_name
password = local.conf.secondary_db_password
local.read_only_db = get_db(host=local.conf.secondary_host, user=user, password=password)
# swap db connections
local.master_db = local.db
local.primary_db = local.db
local.db = local.read_only_db
def get_site_config(sites_path=None, site_path=None):
@ -495,16 +500,17 @@ def whitelist(allow_guest=False, xss_safe=False):
def read_only():
def innfn(fn):
def wrapper_fn(*args, **kwargs):
if conf.use_slave_for_read_only:
connect_read_only()
if conf.read_from_secondary:
connect_secondary()
try:
retval = fn(*args, **get_newargs(fn, kwargs))
except:
raise
finally:
if local and hasattr(local, 'master_db'):
if local and hasattr(local, 'primary_db'):
local.db.close()
local.db = local.master_db
local.db = local.primary_db
return retval
return wrapper_fn