pass through extra args to run()

This commit is contained in:
Noah Levitt 2018-09-28 12:27:13 -07:00
parent a43fb74464
commit edf68edaa2
2 changed files with 4 additions and 3 deletions

View File

@ -66,13 +66,14 @@ class RethinkerWrapper(object):
result.close()
conn.close()
def run(self, db=None):
def run(self, db=None, **kwargs):
self.wrapped.run # raise AttributeError early
while True:
conn = self.rr._random_server_connection()
is_iter = False
try:
result = self.wrapped.run(conn, db=db or self.rr.dbname)
result = self.wrapped.run(
conn, db=db or self.rr.dbname, **kwargs)
if hasattr(result, '__next__'):
is_iter = True
g = self._result_iter(conn, result)

View File

@ -10,7 +10,7 @@ except:
setuptools.setup(
name='doublethink',
version='0.2.0.dev90',
version='0.2.0.dev91',
packages=['doublethink'],
classifiers=[
'Programming Language :: Python :: 2.7',