From 2c19d599b5ee429c693fcb0fe521cd62aed23ad3 Mon Sep 17 00:00:00 2001 From: fred Date: Tue, 26 Sep 2023 13:52:35 +0200 Subject: [PATCH] maximum 30 sec --- tools/page_screenshot.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tools/page_screenshot.py b/tools/page_screenshot.py index da94696f..2644a332 100755 --- a/tools/page_screenshot.py +++ b/tools/page_screenshot.py @@ -27,8 +27,13 @@ def main(): loop = asyncio.new_event_loop() asyncio.set_event_loop(loop) - loop.run_until_complete(take_screenshot(url, output_file, width, height)) - loop.close() + + try: + loop.run_until_complete(asyncio.wait_for(take_screenshot(url, output_file, width, height), timeout=30)) + except asyncio.TimeoutError: + print("Timeout: The operation took too long to complete and has been terminated.") + finally: + loop.close() if __name__ == "__main__": main()