Update browser/driver.py

browser/driver.py

@@ -0,0 +1,107 @@
+"""
+WebDriver management and initialization
+"""
+import os
+import logging
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.chrome.options import Options
+from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.core.os_manager import ChromeType
+from config.settings import CHROMIUM_PATHS, CHROME_OPTIONS, DEFAULT_PAGE_LOAD_TIMEOUT, DEFAULT_IMPLICIT_WAIT
+
+logger = logging.getLogger(__name__)
+
+# Global browser session for persistence
+_driver = None
+
+def create_driver(persistent=False):
+    """Create Chrome driver with proper configuration"""
+    global _driver
+    
+    # If persistent and driver exists, return it
+    if persistent and _driver:
+        try:
+            # Check if driver is still alive
+            _driver.title
+            return _driver
+        except:
+            # Driver is dead, create new one
+            _driver = None
+    
+    try:
+        chrome_options = Options()
+        
+        # Detect Chromium binary location
+        chromium_binary = None
+        for path in CHROMIUM_PATHS:
+            if os.path.exists(path):
+                chromium_binary = path
+                logger.info(f"Found Chromium at: {path}")
+                break
+        
+        if chromium_binary:
+            chrome_options.binary_location = chromium_binary
+        
+        # Add all Chrome options
+        for option in CHROME_OPTIONS:
+            chrome_options.add_argument(option)
+        
+        # Try multiple driver installation methods
+        try:
+            # Method 1: Use system chromium-driver if available
+            if os.path.exists("/usr/bin/chromedriver"):
+                service = Service("/usr/bin/chromedriver")
+                logger.info("Using system chromedriver")
+            else:
+                # Method 2: Use webdriver-manager with ChromeType.CHROMIUM
+                service = Service(ChromeDriverManager(chrome_type=ChromeType.CHROMIUM).install())
+                logger.info("Using webdriver-manager chromedriver")
+        except Exception as e:
+            logger.warning(f"ChromeType.CHROMIUM failed: {e}")
+            # Method 3: Fallback to regular Chrome driver
+            service = Service(ChromeDriverManager().install())
+            logger.info("Using fallback Chrome driver")
+        
+        driver = webdriver.Chrome(service=service, options=chrome_options)
+        driver.set_page_load_timeout(DEFAULT_PAGE_LOAD_TIMEOUT)
+        driver.implicitly_wait(DEFAULT_IMPLICIT_WAIT)
+        
+        if persistent:
+            _driver = driver
+        
+        return driver
+        
+    except Exception as e:
+        logger.error(f"Failed to create driver: {e}")
+        raise
+
+def close_persistent_driver():
+    """Close the persistent driver if it exists"""
+    global _driver
+    if _driver:
+        try:
+            _driver.quit()
+        except:
+            pass
+        _driver = None
+        return "Persistent browser closed"
+    return "No persistent browser to close"
+
+def get_driver(url=None, use_persistent=False):
+    """Get driver and optionally navigate to URL"""
+    driver = create_driver(persistent=use_persistent)
+    
+    if url and (not use_persistent or driver.current_url != url):
+        logger.info(f"Navigating to: {url}")
+        driver.get(url)
+    
+    return driver
+
+def cleanup_driver(driver, use_persistent=False):
+    """Cleanup driver if not persistent"""
+    if not use_persistent and driver:
+        try:
+            driver.quit()
+        except:
+            pass