fix: added delete without filters
This commit is contained in:
Aradhya-Tripathi
parent
a1ed8dc68f
commit
3fefc6a10d
1 changed files with 59 additions and 74 deletions
|
|
@ -7,34 +7,33 @@ import frappe
|
|||
class Query:
|
||||
def __init__(self):
|
||||
self.operator_map = {
|
||||
"+": operator.add,
|
||||
"=": operator.eq,
|
||||
"-": operator.sub,
|
||||
"!=": operator.ne,
|
||||
"<": operator.lt,
|
||||
">": operator.gt,
|
||||
"<=": operator.le,
|
||||
">=": operator.ge,
|
||||
"in": self.func_in,
|
||||
"not in": self.func_not_in,
|
||||
"like": self.like,
|
||||
"not like": self.not_like,
|
||||
"regex": self.func_regex,
|
||||
"between": self.func_between
|
||||
"+": operator.add,
|
||||
"=": operator.eq,
|
||||
"-": operator.sub,
|
||||
"!=": operator.ne,
|
||||
"<": operator.lt,
|
||||
">": operator.gt,
|
||||
"<=": operator.le,
|
||||
">=": operator.ge,
|
||||
"in": self.func_in,
|
||||
"not in": self.func_not_in,
|
||||
"like": self.like,
|
||||
"not like": self.not_like,
|
||||
"regex": self.func_regex,
|
||||
"between": self.func_between
|
||||
}
|
||||
|
||||
self.sql_functions = ["sum", "now", "interval"]
|
||||
|
||||
@staticmethod
|
||||
def like(key: str, value: str) -> frappe.qb:
|
||||
"""Wrapper method for `LIKE`
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb: `frappe.qb object with `LIKE`
|
||||
frappe.qb: `frappe.qb object with `LIKE`
|
||||
"""
|
||||
return frappe.qb.Field(key).like(value)
|
||||
|
||||
|
|
@ -43,11 +42,11 @@ class Query:
|
|||
"""Wrapper method for `IN`
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb: `frappe.qb object with `IN`
|
||||
frappe.qb: `frappe.qb object with `IN`
|
||||
"""
|
||||
return frappe.qb.Field(key).isin(value)
|
||||
|
||||
|
|
@ -56,11 +55,11 @@ class Query:
|
|||
"""Wrapper method for `NOT LIKE`
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb: `frappe.qb object with `NOT LIKE`
|
||||
frappe.qb: `frappe.qb object with `NOT LIKE`
|
||||
"""
|
||||
return frappe.qb.Field(key).not_like(value)
|
||||
|
||||
|
|
@ -69,11 +68,11 @@ class Query:
|
|||
"""Wrapper method for `NOT IN`
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb: `frappe.qb object with `NOT IN`
|
||||
frappe.qb: `frappe.qb object with `NOT IN`
|
||||
"""
|
||||
return frappe.qb.Field(key).notin(value)
|
||||
|
||||
|
|
@ -82,11 +81,11 @@ class Query:
|
|||
"""Wrapper method for `REGEX`
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb: `frappe.qb object with `REGEX`
|
||||
frappe.qb: `frappe.qb object with `REGEX`
|
||||
"""
|
||||
return frappe.qb.Field(key).regex(value)
|
||||
|
||||
|
|
@ -95,66 +94,52 @@ class Query:
|
|||
"""Wrapper method for `BETWEEN`
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb: `frappe.qb object with `BETWEEN`
|
||||
frappe.qb: `frappe.qb object with `BETWEEN`
|
||||
"""
|
||||
return frappe.qb.Field(key)[slice(*value)]
|
||||
|
||||
@staticmethod
|
||||
def get_func_obj(func: str) -> frappe.qb.functions:
|
||||
"""Wrapper method for generating custom functions
|
||||
|
||||
Args:
|
||||
key (str): field
|
||||
value (ANy): criterion
|
||||
|
||||
Returns:
|
||||
frappe.qb.functions
|
||||
"""
|
||||
return frappe.qb.functions(func)
|
||||
|
||||
def make_function(self, key: Any, value: Union[int, str]):
|
||||
return self.operator_map[value[0]](key, value[1])
|
||||
|
||||
def dict_query(self, filters: Dict[str, Union[str, int]], table: str):
|
||||
conditions = frappe.qb.from_(table)
|
||||
for key in filters:
|
||||
value = filters.get(key)
|
||||
_operator = self.operator_map["="]
|
||||
|
||||
if not isinstance(key, str):
|
||||
conditions = conditions.where(self.make_function(key, value))
|
||||
continue
|
||||
if isinstance(value, (list, tuple)):
|
||||
if isinstance(value[1], (list, tuple)) or value[0] in list(self.operator_map.keys())[-4:]:
|
||||
_operator = self.operator_map[value[0]]
|
||||
conditions = conditions.where(_operator(key, value[1]))
|
||||
else:
|
||||
_operator = self.operator_map[value[0]]
|
||||
conditions = conditions.where(_operator(frappe.qb.Field(key), value[1]))
|
||||
else:
|
||||
conditions = conditions.where(_operator(frappe.qb.Field(key), value))
|
||||
|
||||
def dict_query(self, table: str, filters: Dict[str, Union[str, int]]=None):
|
||||
conditions = frappe.qb.from_(table)
|
||||
if not filters:
|
||||
return conditions
|
||||
|
||||
def build_conditions(self, filters: Union[Dict[str, Union[str, int]], str, int], table: str) -> frappe.qb:
|
||||
for key in filters:
|
||||
value = filters.get(key)
|
||||
_operator = self.operator_map["="]
|
||||
|
||||
if not isinstance(key, str):
|
||||
conditions = conditions.where(self.make_function(key, value))
|
||||
continue
|
||||
if isinstance(value, (list, tuple)):
|
||||
if isinstance(value[1], (list, tuple)) or value[0] in list(self.operator_map.keys())[-4:]:
|
||||
_operator = self.operator_map[value[0]]
|
||||
conditions = conditions.where(_operator(key, value[1]))
|
||||
else:
|
||||
_operator = self.operator_map[value[0]]
|
||||
conditions = conditions.where(_operator(frappe.qb.Field(key), value[1]))
|
||||
else:
|
||||
conditions = conditions.where(_operator(frappe.qb.Field(key), value))
|
||||
|
||||
return conditions
|
||||
|
||||
def build_conditions(self, table: str, filters: Union[Dict[str, Union[str, int]], str, int]=None) -> frappe.qb:
|
||||
"""Build conditions for sql query
|
||||
|
||||
Args:
|
||||
filters (Union[Dict[str, Union[str, int]], str, int]): conditions built from filters provided
|
||||
table (str): DocType
|
||||
filters (Union[Dict[str, Union[str, int]], str, int]): conditions built from filters provided
|
||||
table (str): DocType
|
||||
|
||||
Returns:
|
||||
frappe.qb: frappe.qb conditions object
|
||||
frappe.qb: frappe.qb conditions object
|
||||
"""
|
||||
if isinstance(filters, list):
|
||||
pass
|
||||
|
||||
if isinstance(filters, int) or isinstance(filters, str):
|
||||
filters = {"name": str(filters)}
|
||||
|
||||
if filters and isinstance(filters, dict):
|
||||
return self.dict_query(filters, table)
|
||||
return self.dict_query(filters=filters, table=table)
|
||||
Loading…
Add table
Reference in a new issue