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

fix: Remove cross import statements

Browse source
This commit is contained in:
Gavin D'souza 2021-08-12 17:44:14 +05:30
parent 233b42b945
commit 8fc2577e9a
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
frappe/query_builder/utils.py
View file

@ -1,5 +1,6 @@
from enum import Enum
from typing import Any, Callable, Dict, get_type_hints
from importlib import import_module
from pypika import Query
@ -19,6 +20,9 @@ class ImportMapper:
db = db_type_is(frappe.conf.db_type or "mariadb")
return self.func_map[db](*args, **kwds)
class BuilderIdentificationFailed(Exception):
def __init__(self):
super().__init__("Couldn't guess builder")
def get_query_builder(type_of_db: str) -> Query:
"""[return the query builder object]
@ -33,6 +37,10 @@ def get_query_builder(type_of_db: str) -> Query:
picks = {db_type_is.MARIADB: MariaDB, db_type_is.POSTGRES: Postgres}
return picks[db]
def get_attr(method_string):
modulename = '.'.join(method_string.split('.')[:-1])
methodname = method_string.split('.')[-1]
return getattr(import_module(modulename), methodname)
def patch_query_execute():
"""Patch the Query Builder with helper execute method
@ -43,12 +51,7 @@ def patch_query_execute():
def execute_query(query, **kwargs):
return frappe.db.sql(query, **kwargs)
class BuilderIdentificationFailed(Exception):
def __init__(self, message=None):
message = message or "Couldn't guess builder"
super().__init__(message)
query_class = frappe.get_attr(str(frappe.qb).split("'")[1])
query_class = get_attr(str(frappe.qb).split("'")[1])
builder_class = get_type_hints(query_class._builder).get('return')
if not builder_class: