retoor/dataset
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add find operators

Browse Source
This commit is contained in:
Unknown
2018-06-13 02:29:32 +02:00
parent 38a59b921e
commit f9bb652034
3 changed files with 52 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dataset/table.py
+19 -1
View File
@@ -2,7 +2,7 @@ import logging
import warnings
import threading
from sqlalchemy.sql import and_, expression
from sqlalchemy.sql import and_, between, expression
from sqlalchemy.sql.expression import ClauseElement
from sqlalchemy.schema import Column, Index
from sqlalchemy import func, select, false
@@ -293,6 +293,24 @@ class Table(object):
clauses.append(false())
elif isinstance(value, (list, tuple)):
clauses.append(self.table.c[column].in_(value))
elif isinstance(value, dict):
key = list(value.keys())[0]
if key in ['like']:
clauses.append(self.table.c[column].like(value[key]))
elif key in ['>', 'gt']:
clauses.append(self.table.c[column] > value[key])
elif key in ['<', 'lt']:
clauses.append(self.table.c[column] < value[key])
elif key in ['>=', 'gte']:
clauses.append(self.table.c[column] >= value[key])
elif key in ['<=', 'lte']:
clauses.append(self.table.c[column] <= value[key])
elif key in ['!=', '<>', 'not']:
clauses.append(self.table.c[column] != value[key])
elif key in ['between', '..']:
clauses.append(self.table.c[column].between(value[key][0], value[key][1]))
else:
clauses.append(false)
else:
clauses.append(self.table.c[column] == value)
return and_(*clauses)