diff --git a/video_creation/screenshot_downloader.py b/video_creation/screenshot_downloader.py index da2adbf..581f92e 100644 --- a/video_creation/screenshot_downloader.py +++ b/video_creation/screenshot_downloader.py @@ -40,7 +40,6 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int): print_substep("Launching Headless Browser...") browser = p.chromium.launch(headless=True) # headless=False will show the browser for debugging purposes - context = browser.new_context() # Device scale factor (or dsf for short) allows us to increase the resolution of the screenshots # When the dsf is 1, the width of the screenshot is 600 pixels # so we need a dsf such that the width of the screenshot is greater than the final resolution of the video @@ -66,6 +65,9 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int): bgcolor = (0, 0, 0, 0) txtcolor = (255, 255, 255) transparent = True + cookie_file = open( + "./video_creation/data/cookie-dark-mode.json", encoding="utf-8" + ) else: # Switch to dark theme cookie_file = open( @@ -90,18 +92,19 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int): context.add_cookies(cookies) # load preference cookies # Login to Reddit - print_substep("Logging in to Reddit...") - page = context.new_page() - page.goto("https://www.reddit.com/login", timeout=0) - page.set_viewport_size(ViewportSize(width=1920, height=1080)) - page.wait_for_load_state() - - page.locator('[name="username"]').fill(settings.config["reddit"]["creds"]["username"]) - page.locator('[name="password"]').fill(settings.config["reddit"]["creds"]["password"]) - page.locator("button:has-text('Log In')").click() - - page.wait_for_load_state() # Wait for page to fully load and add 5 seconds - page.wait_for_timeout(5000) + if storymode: + print_substep("Logging in to Reddit...") + page = context.new_page() + page.goto("https://www.reddit.com/login", timeout=0) + page.set_viewport_size(ViewportSize(width=1920, height=1080)) + page.wait_for_load_state() + + page.locator('[name="username"]').fill(settings.config["reddit"]["creds"]["username"]) + page.locator('[name="password"]').fill(settings.config["reddit"]["creds"]["password"]) + page.locator("button[class$='m-full-width']").click() + + page.wait_for_load_state() # Wait for page to fully load and add 5 seconds + page.wait_for_timeout(5000) # Get the thread screenshot page = context.new_page()