Fixing screenshot cmd --wait

medialab · Apr 16, 2024 · 67518bc · 67518bc
1 parent 3cef9d2
commit 67518bc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/minet/cli/fetch/fetch.py b/minet/cli/fetch/fetch.py
@@ -461,7 +461,6 @@ def worker_callback(
 
     # Screenshot
     elif cli_args.action == "screenshot":
-        import asyncio
         from playwright.async_api import (
             Error as PlaywrightError,
             TimeoutError as PlaywrightTimeoutError,
@@ -523,7 +522,7 @@ async def screenshot(
                     )
 
                     if cli_args.wait is not None:
-                        await asyncio.sleep(cli_args.wait)
+                        await page.wait_for_timeout(cli_args.wait * 1000)
 
                     await page.screenshot(
                         path=pathjoin(cli_args.output_dir, filename),