selenium: fix autosave_cookies

TheShellLand · Oct 1, 2024 · 1bc0573 · 1bc0573
1 parent c36c249
commit 1bc0573
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/automon/integrations/seleniumWrapper/browser.py b/automon/integrations/seleniumWrapper/browser.py
@@ -169,6 +169,7 @@ def refresh(self) -> None:
 
     @property
     def url(self):
+        """alias to current_url"""
         return self.current_url
 
     @property
@@ -367,7 +368,8 @@ def add_cookie_from_url(self, url: str) -> bool:
             logger.info(f'add_cookie_from_url :: done')
             return add_cookie_from_url
 
-        raise Exception(f'add_cookie_from_url :: failed :: {cookie_file=}')
+        logger.error(f'add_cookie_from_url :: failed :: file not found :: {cookie_file=}')
+        raise Exception(f'add_cookie_from_url :: failed :: file not found :: {cookie_file=}')
 
     def add_cookie_from_base64(self, base64_str: str) -> bool:
         """add cookie from base64 string"""
@@ -385,14 +387,16 @@ def add_cookie_from_base64(self, base64_str: str) -> bool:
 
     def autosave_cookies(self) -> bool:
         """auto save cookies for current url"""
-        logger.debug(f'autosave_cookies')
+        logger.debug(f'autosave_cookies :: {self.current_url=}')
 
         if self.current_url:
-            logger.debug(f'autosave_cookies :: {self.current_url=}')
 
             if not self.autosaved:
                 logger.debug(f'autosave_cookies :: {self.autosaved=}')
-                self.add_cookie_from_current_url()
+                try:
+                    self.add_cookie_from_current_url()
+                except:
+                    logger.debug(f'autosave_cookies :: no cookies for {self.current_url=}')
                 self.refresh()
                 self.autosaved = True
                 logger.debug(f'autosave_cookies :: {self.autosaved=}')