from DrissionPage import Chromium, ChromiumOptions

from app.config import gAppConfig
from app.utils import random_ua

class DPEngine:
    """DrissionPage 引擎"""

    def __init__(self, is_wap: bool = False, no_img: bool = True):
        chrome_opts = ChromiumOptions()
        chrome_opts.mute(True)  # 静音
        chrome_opts.headless(True)  # 无头模式
        chrome_opts.no_image(no_img)  # 不加载图片
        chrome_opts.set_argument("--disable-gpu")  # 禁用GPU
        chrome_opts.set_argument('--ignore-certificate-errors')  # 忽略证书错误
        chrome_opts.set_argument('--start-maximized')   # 最大化启动
        chrome_opts.set_argument(f'--user-agent={random_ua(is_wap)}') # 设置 UA
        # 设置代理
        if gAppConfig.chrome.proxy:
            chrome_opts.set_argument('--proxy-server', gAppConfig.chrome.proxy)
        
        # 创建浏览器对象
        self.browser = Chromium(addr_or_opts=chrome_opts)