From 64e52737646d1661e0d295868fc85aa0da47b1f2 Mon Sep 17 00:00:00 2001
From: Gavin D'souza <gavin18d@gmail.com>
Date: Fri, 10 Jun 2022 17:42:47 +0530
Subject: [PATCH] perf: Patch qb only once - not on every init

---
 frappe/__init__.py            | 7 +++++--
 frappe/query_builder/utils.py | 5 ++++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/frappe/__init__.py b/frappe/__init__.py
index 51c6ba3a74..37a03e412c 100644
--- a/frappe/__init__.py
+++ b/frappe/__init__.py
@@ -44,6 +44,7 @@ local = Local()
 STANDARD_USERS = ("Guest", "Administrator")
 
 _dev_server = int(sbool(os.environ.get("DEV_SERVER", False)))
+_qb_patched = False
 
 if _dev_server:
 	warnings.simplefilter("always", DeprecationWarning)
@@ -236,8 +237,10 @@ def init(site, sites_path=None, new_site=False):
 	local.qb = get_query_builder(local.conf.db_type or "mariadb")
 
 	setup_module_map()
-	patch_query_execute()
-	patch_query_aggregation()
+
+	if not _qb_patched:
+		patch_query_execute()
+		patch_query_aggregation()
 
 	local.initialised = True
 
diff --git a/frappe/query_builder/utils.py b/frappe/query_builder/utils.py
index 69aee9b350..c75e380d49 100644
--- a/frappe/query_builder/utils.py
+++ b/frappe/query_builder/utils.py
@@ -64,7 +64,6 @@ def patch_query_execute():
 	This excludes the use of `frappe.db.sql` method while
 	executing the query object
 	"""
-	from frappe.utils.safe_exec import check_safe_sql_query
 
 	def execute_query(query, *args, **kwargs):
 		query, params = prepare_query(query)
@@ -73,6 +72,8 @@ def patch_query_execute():
 	def prepare_query(query):
 		import inspect
 
+		from frappe.utils.safe_exec import check_safe_sql_query
+
 		param_collector = NamedParameterWrapper()
 		query = query.get_sql(param_wrapper=param_collector)
 		if frappe.flags.in_safe_exec and not check_safe_sql_query(query, throw=False):
@@ -103,6 +104,7 @@ def patch_query_execute():
 
 	builder_class.run = execute_query
 	builder_class.walk = prepare_query
+	frappe._qb_patched = True
 
 
 def patch_query_aggregation():
@@ -113,3 +115,4 @@ def patch_query_aggregation():
 	frappe.qb.min = _min
 	frappe.qb.avg = _avg
 	frappe.qb.sum = _sum
+	frappe._qb_patched = True