browser/cookie_validator.py

import time
import sys
from playwright.sync_api import TimeoutError, Error as PlaywrightError


class CookieValidator:
    """Cookie验证器，负责定期验证Cookie的有效性。"""

    def __init__(self, page, context, logger):
        """
        初始化Cookie验证器

        Args:
            page: 主页面实例
            context: 浏览器上下文
            logger: 日志记录器
        """
        self.page = page
        self.context = context
        self.logger = logger

    
    def validate_cookies_in_main_thread(self):
        """
        在主线程中执行Cookie验证（由主线程调用）

        Returns:
            bool: Cookie是否有效
        """
        validation_page = None
        try:
            # 创建新标签页（在主线程中执行）
            self.logger.info("开始Cookie验证...")
            validation_page = self.context.new_page()

            # 访问验证URL
            validation_url = "https://aistudio.google.com/apps"
            validation_page.goto(validation_url, wait_until='domcontentloaded', timeout=30000)

            # 等待页面加载
            validation_page.wait_for_timeout(2000)

            # 获取最终URL
            final_url = validation_page.url

            # 检查是否被重定向到登录页面
            if "accounts.google.com/v3/signin/identifier" in final_url:
                self.logger.error("Cookie验证失败: 被重定向到登录页面")
                return False

            if "accounts.google.com/v3/signin/accountchooser" in final_url:
                self.logger.error("Cookie验证失败: 被重定向到账户选择页面")
                return False

            # 如果没有跳转到登录页面，就算成功
            self.logger.info("Cookie验证成功")
            return True

        except TimeoutError:
            self.logger.error("Cookie验证失败: 页面加载超时")
            return False

        except PlaywrightError as e:
            self.logger.error(f"Cookie验证失败: {e}")
            return False

        except Exception as e:
            self.logger.error(f"Cookie验证失败: {e}")
            return False

        finally:
            # 关闭验证标签页
            if validation_page:
                try:
                    validation_page.close()
                except Exception:
                    pass  # 忽略关闭错误

    def shutdown_instance_on_cookie_failure(self):
        """
        因Cookie失效而关闭实例
        """
        self.logger.error("Cookie失效，关闭实例")
        time.sleep(1)
        sys.exit(1)