backup/warcprox
1
0
Fork 0
mirror of https://github.com/internetarchive/warcprox.git synced 2025-01-18 13:22:09 +01:00
Code Issues Projects Releases Activity

pass through extra args to run()

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
doublethink/rethinker.py
View File

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

2
setup.py
View File

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