autoUpload/utils/anti_detection.py

# -*- coding: utf-8 -*-
"""
反自动化检测工具模块
提供统一的反检测措施，供各个社交媒体上传器使用
"""

from typing import Dict, Any, Optional, List
from playwright.async_api import BrowserContext, Browser, Playwright


class AntiDetectionConfig:
    """反检测配置类"""
    
    # 标准的反检测浏览器参数
    STANDARD_BROWSER_ARGS = [
        '--no-sandbox',
        '--disable-blink-features=AutomationControlled',  # 核心：禁用自动化控制标识
        '--disable-web-security',
        '--disable-features=VizDisplayCompositor',
        '--disable-dev-shm-usage',
        '--disable-infobars',
        '--disable-extensions',
        '--disable-gpu',
        '--disable-dev-shm-usage',
        '--no-first-run',
        '--no-default-browser-check',
        '--lang=zh-CN'
    ]
    
    # 真实的用户代理字符串
    REAL_USER_AGENTS = [
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36',
        'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/121.0'
    ]
    
    # 默认的上下文选项
    DEFAULT_CONTEXT_OPTIONS = {
        'viewport': {'width': 1920, 'height': 1080},
        'locale': 'zh-CN',
        'timezone_id': 'Asia/Shanghai',
        'device_scale_factor': 1,
        'has_touch': False,
        'is_mobile': False
    }


async def create_stealth_browser(
    playwright: Playwright,
    headless: bool = True,
    executable_path: Optional[str] = None,
    custom_args: Optional[List[str]] = None
) -> Browser:
    """
    创建具有反检测功能的浏览器实例
    
    Args:
        playwright: Playwright实例
        headless: 是否使用无头模式
        executable_path: 自定义浏览器可执行文件路径
        custom_args: 自定义浏览器参数（会与标准参数合并）
    
    Returns:
        配置好的Browser实例
    """
    # 合并浏览器参数
    browser_args = AntiDetectionConfig.STANDARD_BROWSER_ARGS.copy()
    if custom_args:
        browser_args.extend(custom_args)
    
    # 浏览器启动选项
    launch_options = {
        'headless': headless,
        'args': browser_args
    }
    
    if executable_path:
        launch_options['executable_path'] = executable_path
    
    return await playwright.chromium.launch(**launch_options)


async def create_stealth_context(
    browser: Browser,
    account_file: str,
    headless: bool = True,
    custom_options: Optional[Dict[str, Any]] = None
) -> BrowserContext:
    """
    创建具有反检测功能的浏览器上下文
    
    Args:
        browser: Browser实例
        account_file: Cookie文件路径
        headless: 是否为无头模式
        custom_options: 自定义上下文选项
    
    Returns:
        配置好的BrowserContext实例
    """
    # 基础上下文选项
    context_options = {
        'storage_state': account_file,
    }
    
    # 为无头模式添加额外的反检测措施
    if headless:
        context_options.update(AntiDetectionConfig.DEFAULT_CONTEXT_OPTIONS)
        
        # 使用随机的真实用户代理
        import random
        user_agent = random.choice(AntiDetectionConfig.REAL_USER_AGENTS)
        context_options['user_agent'] = user_agent
    
    # 合并自定义选项
    if custom_options:
        context_options.update(custom_options)
    
    return await browser.new_context(**context_options)


async def setup_stealth_page(context: BrowserContext, url: str):
    """
    创建并设置具有反检测功能的页面
    
    Args:
        context: BrowserContext实例
        url: 要访问的URL
    
    Returns:
        配置好的Page实例
    """
    from utils.base_social_media import set_init_script
    
    # 应用stealth脚本
    context = await set_init_script(context)
    
    # 创建页面
    page = await context.new_page()
    
    # 访问URL（使用更温和的等待策略）
    try:
        await page.goto(url, wait_until='domcontentloaded', timeout=30000)
    except Exception as e:
        print(f"页面导航警告: {e}")
        # 即使导航有问题，也继续返回页面对象
    
    return page


class SocialMediaUploader:
    """
    社交媒体上传器基类
    提供统一的反检测功能
    """
    
    def __init__(self, headless: bool = True, executable_path: Optional[str] = None):
        self.headless = headless
        self.executable_path = executable_path
        self.browser = None
        self.context = None
        self.page = None
    
    async def create_browser_session(
        self, 
        playwright: Playwright, 
        account_file: str,
        target_url: str,
        custom_browser_args: Optional[List[str]] = None,
        custom_context_options: Optional[Dict[str, Any]] = None
    ):
        """
        创建完整的浏览器会话（浏览器+上下文+页面）
        
        Args:
            playwright: Playwright实例
            account_file: Cookie文件路径
            target_url: 目标URL
            custom_browser_args: 自定义浏览器参数
            custom_context_options: 自定义上下文选项
        
        Returns:
            tuple: (browser, context, page)
        """
        # 创建浏览器
        self.browser = await create_stealth_browser(
            playwright=playwright,
            headless=self.headless,
            executable_path=self.executable_path,
            custom_args=custom_browser_args
        )
        
        # 创建上下文
        self.context = await create_stealth_context(
            browser=self.browser,
            account_file=account_file,
            headless=self.headless,
            custom_options=custom_context_options
        )
        
        # 创建页面
        self.page = await setup_stealth_page(self.context, target_url)
        
        return self.browser, self.context, self.page
    
    async def close_session(self):
        """关闭浏览器会话"""
        if self.context:
            await self.context.close()
        if self.browser:
            await self.browser.close()


def get_anti_detection_info() -> Dict[str, Any]:
    """
    获取反检测配置信息，用于调试
    
    Returns:
        包含配置信息的字典
    """
    return {
        'browser_args_count': len(AntiDetectionConfig.STANDARD_BROWSER_ARGS),
        'browser_args': AntiDetectionConfig.STANDARD_BROWSER_ARGS,
        'user_agents_count': len(AntiDetectionConfig.REAL_USER_AGENTS),
        'sample_user_agent': AntiDetectionConfig.REAL_USER_AGENTS[0],
        'default_viewport': AntiDetectionConfig.DEFAULT_CONTEXT_OPTIONS['viewport'],
        'locale': AntiDetectionConfig.DEFAULT_CONTEXT_OPTIONS['locale'],
        'timezone': AntiDetectionConfig.DEFAULT_CONTEXT_OPTIONS['timezone_id']
    }


# 便捷函数，用于快速设置
async def quick_setup_stealth_browser(
    playwright: Playwright,
    account_file: str,
    target_url: str,
    headless: bool = True,
    executable_path: Optional[str] = None
):
    """
    快速设置具有反检测功能的完整浏览器环境
    
    Args:
        playwright: Playwright实例
        account_file: Cookie文件路径
        target_url: 目标URL
        headless: 是否使用无头模式
        executable_path: 浏览器可执行文件路径
    
    Returns:
        tuple: (browser, context, page)
    """
    uploader = SocialMediaUploader(headless=headless, executable_path=executable_path)
    return await uploader.create_browser_session(
        playwright=playwright,
        account_file=account_file,
        target_url=target_url
    )


# 调试工具
def print_anti_detection_config():
    """打印当前的反检测配置，用于调试"""
    info = get_anti_detection_info()
    print("🛡️ 反检测配置信息:")
    print("=" * 50)
    print(f"浏览器参数数量: {info['browser_args_count']}")
    print(f"用户代理数量: {info['user_agents_count']}")
    print(f"默认视口: {info['default_viewport']}")
    print(f"语言区域: {info['locale']}")
    print(f"时区: {info['timezone']}")
    print("\n📋 浏览器参数:")
    for i, arg in enumerate(info['browser_args'], 1):
        print(f"  {i:2d}. {arg}")
    print(f"\n🌐 示例用户代理:\n  {info['sample_user_agent']}")


if __name__ == "__main__":
    # 测试和演示
    print_anti_detection_config()