Skip to content

Commit 5657a32

Browse files
committed
add is_null() & is_not_null() methods, not_null() as a synonym
1 parent 36f5733 commit 5657a32

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

simple_query_builder/querybuilder.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -392,6 +392,24 @@ def not_like(self, cond: Union[str, tuple, list] = ()):
392392
self.where([[cond[0], "NOT LIKE", cond[1]]])
393393
return self
394394

395+
def is_null(self, field: str = ""):
396+
if not field:
397+
self.set_error(f"Empty field in {inspect.stack()[0][3]} method")
398+
return self
399+
self.where([[field, "IS NULL"]])
400+
return self
401+
402+
def is_not_null(self, field: str):
403+
if not field:
404+
self.set_error(f"Empty field in {inspect.stack()[0][3]} method")
405+
return self
406+
self.where([[field, "IS NOT NULL"]])
407+
return self
408+
409+
def not_null(self, field: str):
410+
self.is_not_null(field)
411+
return self
412+
395413
def limit(self, limit: int = 1):
396414
self._sql += f" LIMIT {limit}"
397415
return self

0 commit comments

Comments
 (0)