Update src/browser/custom_context.py

src/browser/custom_context.py

@@ -13,71 +13,63 @@ class CustomBrowserContext:
         self.driver = None
 
     def create_context(self):
-        """Crea un nuevo contexto de navegador con medidas anti-detección y carga cookies si están disponibles."""
         chrome_options = Options()
 
-        # Configuración de la ventana del navegador
         if self.config.get("browser_window_size"):
             chrome_options.add_argument(f"--window-size={self.config['browser_window_size'][0]},{self.config['browser_window_size'][1]}")
 
-        # Configuración de seguridad
         if self.config.get("disable_security", False):
             chrome_options.add_argument("--disable-web-security")
             chrome_options.add_argument("--disable-site-isolation-trials")
             chrome_options.add_argument("--disable-features=IsolateOrigins,site-per-process")
 
-        # Configuración del user-agent
         user_agent = (
             "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
             "(KHTML, like Gecko) Chrome/85.0.4183.102 Safari/537.36"
         )
         chrome_options.add_argument(f"user-agent={user_agent}")
 
-        # Inicializar el navegador
         self.driver = webdriver.Chrome(options=chrome_options)
 
-        # Cargar cookies si existen
         if self.config.get("cookies_file") and os.path.exists(self.config["cookies_file"]):
-            with open(self.config["cookies_file"], "r") as f:
-                cookies = json.load(f)
-                logger.info(f"Loaded {len(cookies)} cookies from {self.config['cookies_file']}")
-                for cookie in cookies:
-                    self.driver.add_cookie(cookie)
+            try:
+                with open(self.config["cookies_file"], "r") as f:
+                    cookies = json.load(f)
+                    logger.info(f"Loaded {len(cookies)} cookies from {self.config['cookies_file']}")
+                    for cookie in cookies:
+                        self.driver.add_cookie(cookie)
+            except Exception as e:
+                logger.error(f"Error loading cookies: {e}")
 
-        # Ejecutar scripts anti-detección
-        self.driver.execute_script(
-            """
-            // Webdriver property
-            Object.defineProperty(navigator, 'webdriver', {
-                get: () => undefined
-            });
-
-            // Languages
-            Object.defineProperty(navigator, 'languages', {
-                get: () => ['en-US', 'en']
-            });
-
-            // Plugins
-            Object.defineProperty(navigator, 'plugins', {
-                get: () => [1, 2, 3, 4, 5]
-            });
-
-            // Chrome runtime
-            window.chrome = { runtime: {} };
-
-            // Permissions
-            const originalQuery = window.navigator.permissions.query;
-            window.navigator.permissions.query = (parameters) => (
-                parameters.name === 'notifications' ?
-                    Promise.resolve({ state: Notification.permission }) :
-                    originalQuery(parameters)
-            );
-            """
-        )
+        try:
+            self.driver.execute_script(
+                """
+                Object.defineProperty(navigator, 'webdriver', {
+                    get: () => undefined
+                });
+                Object.defineProperty(navigator, 'languages', {
+                    get: () => ['en-US', 'en']
+                });
+                Object.defineProperty(navigator, 'plugins', {
+                    get: () => [1, 2, 3, 4, 5]
+                });
+                window.chrome = { runtime: {} };
+                const originalQuery = window.navigator.permissions.query;
+                window.navigator.permissions.query = (parameters) => (
+                    parameters.name === 'notifications' ?
+                        Promise.resolve({ state: Notification.permission }) :
+                        originalQuery(parameters)
+                );
+                """
+            )
+        except Exception as e:
+            logger.error(f"Error executing anti-detection scripts: {e}")
 
         return self.driver
 
     def close(self):
-        """Cierra el contexto del navegador."""
         if self.driver:
-            self.driver.quit()
+            try:
+                self.driver.quit()
+            except Exception as e:
+                logger.error(f"Error closing the browser: {e}")