from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
import time
import run

if __name__ == '__main__':
    options = webdriver.ChromeOptions()
    options.add_argument('lang=zh-CN')
    options.add_argument(
        'user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36')
    # options.add_argument(r'--user-data-dir=E:\Repository\skcks.cn\python-selenium-spider\tmp\UserData')
    options.add_argument('--disable-gpu')  # 如果不加这个选项，有时定位会出现问题
    # options.add_argument('--headless')  # 增加无界面选项

    service = webdriver.ChromeService(r"./driver/chromedriver.exe")
    driver = webdriver.Chrome(service=service,options=options)
    # driver.maximize_window()
    # driver.execute_cdp_cmd("Page.addScriptToEvaluateOnNewDocument", {
    #     "source": """
    #             Object.defineProperty(navigator, 'webdriver', {
    #               get: () => undefined
    #             })
    #             """
    # })

    try:
        with open("./stealth.min.js") as f:
            driver.execute_cdp_cmd("Page.addScriptToEvaluateOnNewDocument", {
                "source": f.read()
            })

        run.run(driver=driver)
        time.sleep(5)
    finally:
        driver.quit()