fixed dark mode bug

3 years ago · 77f68b847c
parent d746fabb6b
commit 77f68b847c
1 changed files with 8 additions and 5 deletions
--- a/video_creation/screenshot_downloader.py
+++ b/video_creation/screenshot_downloader.py
@ -22,11 +22,14 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num, theme):

        browser = p.chromium.launch()
        context = browser.new_context()
-
-        if theme.casefold() == "dark":
-            cookie_file = open('video_creation/cookies.json')
-            cookies = json.load(cookie_file)
-            context.add_cookies(cookies)
+        
+        try:
+            if theme.casefold() == "dark":
+                cookie_file = open('video_creation/cookies.json')
+                cookies = json.load(cookie_file)
+                context.add_cookies(cookies)
+        except AttributeError:
+            pass

        # Get the thread screenshot
        page = context.new_page()