Simple change in table.py to return query from find

Browse Source
This commit is contained in:
Jameson Merkow 2016-04-09 15:08:31 -07:00
parent 38aef065cb
commit 2fa5f5f1d4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dataset/persistence/table.py
View File

@ -365,6 +365,7 @@ class Table(object):
_step = kwargs.pop('_step', 5000)
order_by = kwargs.pop('order_by', 'id')
return_count = kwargs.pop('return_count', False)
return_query = kwargs.pop('return_query', False)
_filter = kwargs
self._check_dropped()
@ -391,6 +392,8 @@ class Table(object):
query = self.table.select(whereclause=args, limit=_limit,
offset=_offset, order_by=order_by)
if return_query:
return query
return ResultIter(self.database.executable.execute(query),
row_type=self.database.row_type, step=_step)