proxy_handler.py

import requests
from typing import Dict, Optional

class ProxyHandler:
    def __init__(self, proxy_url: str = None):
        self.proxy_url = proxy_url

    def get_proxy_config(self):
        """Return proxy configuration for requests."""
        if not self.proxy_url:
            return None
        return {
            'http': self.proxy_url,
            'https': self.proxy_url
        }

    def get_selenium_proxy_options(self):
        """Return proxy configuration for Selenium."""
        from selenium.webdriver.common.proxy import Proxy, ProxyType
        if not self.proxy_url:
            return None
        
        proxy = Proxy()
        proxy.proxy_type = ProxyType.MANUAL
        proxy.http_proxy = self.proxy_url
        proxy.ssl_proxy = self.proxy_url
        return proxy