From 68592c97683ba3fb9a17bbb45ea7f83d394a5ffe Mon Sep 17 00:00:00 2001 From: Nathaniel Phillips Date: Sat, 29 Jun 2024 19:21:43 -0500 Subject: [PATCH] Update screenshot_downloader.py --- video_creation/screenshot_downloader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/video_creation/screenshot_downloader.py b/video_creation/screenshot_downloader.py index 12a844d..f2e2ade 100644 --- a/video_creation/screenshot_downloader.py +++ b/video_creation/screenshot_downloader.py @@ -181,6 +181,7 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int): location[i] = float("{:.2f}".format(location[i] * zoom)) page.screenshot(clip=location, path=postcontentpath) else: + page.wait_for_load_state('load') page.locator('[data-test-id="post-content"]').screenshot(path=postcontentpath) except Exception as e: print_substep("Something went wrong!", style="red")