From 0633e17998807e6d3c4c564b103003e93df39b98 Mon Sep 17 00:00:00 2001 From: Patrick Robertson Date: Tue, 4 Feb 2025 14:18:46 +0100 Subject: [PATCH] Close the facebook 'login' window if it's there - to allow for proper screenshots --- src/auto_archiver/utils/webdriver.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/auto_archiver/utils/webdriver.py b/src/auto_archiver/utils/webdriver.py index 005f49d..db26d04 100644 --- a/src/auto_archiver/utils/webdriver.py +++ b/src/auto_archiver/utils/webdriver.py @@ -66,6 +66,13 @@ class CookieSettingDriver(webdriver.Firefox): logger.warning(f'Failed on fb accept cookies.', e) # now get the actual URL super(CookieSettingDriver, self).get(url) + if self.facebook_accept_cookies: + # try and click the 'close' button on the 'login' window to close it + close_button = self.find_element(By.XPATH, "//div[@role='dialog']//div[@aria-label='Close']") + if close_button: + close_button.click() + + class Webdriver: def __init__(self, width: int, height: int, timeout_seconds: int,