From edf68edaa24674a51be92e15d65ee40c0c2a6918 Mon Sep 17 00:00:00 2001 From: Noah Levitt Date: Fri, 28 Sep 2018 12:27:13 -0700 Subject: [PATCH] pass through extra args to run() --- doublethink/rethinker.py | 5 +++-- setup.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/doublethink/rethinker.py b/doublethink/rethinker.py index a0442fe..8270e1e 100644 --- a/doublethink/rethinker.py +++ b/doublethink/rethinker.py @@ -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) diff --git a/setup.py b/setup.py index 7511e2c..fe99bbb 100644 --- a/setup.py +++ b/setup.py @@ -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',