from playwright.async_api import async_playwright
from auto_media_publisher.uploaders.login_chrome import LoginChrome

class LoginChromeCDP(LoginChrome):
    def __init__(self, user,platform,proxy=None):
        if proxy is None:
            proxy = "http://127.0.0.1:7890"
        super().__init__(user,platform,proxy)
      
    async def _init_browser_context(self):
        """
        实现平台登录操作，如扫码、验证码登录等
        """
        async with async_playwright() as playwright:
            self.browser = await playwright.chromium.connect_over_cdp(self.cdp_endpoint)
            self.context = self.browser.contexts[0]  # 复用第一个上下文