#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
生成小红书登录二维码
"""

import json
import base64
from pathlib import Path
from playwright.sync_api import sync_playwright

# 加载 cookies 配置
CONFIG_FILE = Path(__file__).parent / "beijing-exhibitions" / "config" / "xiaohongshu_cookies.json"

def get_login_qr():
    """获取登录二维码"""
    with sync_playwright() as p:
        browser = p.chromium.launch(
            headless=True,
            executable_path='/usr/bin/google-chrome',
            args=['--disable-blink-features=AutomationControlled']
        )
        
        context = browser.new_context()
        page = context.new_page()
        
        # 访问小红书登录页
        print("访问小红书登录页...")
        page.goto('https://www.xiaohongshu.com/login', wait_until='domcontentloaded', timeout=60000)
        page.wait_for_timeout(5000)
        
        # 截图保存二维码
        qr_path = Path(__file__).parent / "wiki" / "xiaohongshu_login_qr_new.png"
        qr_path.parent.mkdir(exist_ok=True)
        
        # 尝试截取登录二维码区域
        print("截取二维码...")
        page.screenshot(path=str(qr_path), full_page=False)
        
        print(f"二维码已保存到：{qr_path}")
        print(f"请用小红书 App 扫码登录")
        
        browser.close()
        return str(qr_path)

if __name__ == "__main__":
    get_login_qr()