diff --git a/src/wikiq/__init__.py b/src/wikiq/__init__.py index 18bcefd..34e1892 100755 --- a/src/wikiq/__init__.py +++ b/src/wikiq/__init__.py @@ -49,12 +49,12 @@ class PersistMethod: async def diff_async(differ, last_text, text): try: + loop = asyncio.get_running_loop() return await asyncio.wait_for( asyncio.to_thread(differ.inline_json_diff, last_text, text), timeout=DIFF_TIMEOUT ) - except TimeoutError: - print(f"WARNING! wikidiff2 timeout for rev: {row_buffer['revid'][i]}. Falling back to default limits.", file=sys.stderr) + except TimeoutError as e: return None @@ -681,6 +681,7 @@ class WikiqParser: for i, text in enumerate(row_buffer["text"]): diff = asyncio.run(diff_async(differ, last_text, text)) if diff is None: + print(f"WARNING! wikidiff2 timeout for rev: {row_buffer['revid'][i]}. Falling back to default limits.", file=sys.stderr) diff = fast_differ.inline_json_diff(last_text, text) new_diffs.append(diff) last_text = text