python-selenium-spider/main.py
2024-05-12 01:41:09 +08:00

48 lines
1.6 KiB
Python

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
import time
import run
download_dir = r'E:\Repository\python-selenium-spider\download'
proxies = {
"http": "socks5://127.0.0.1:1080",
"https": "socks5://127.0.0.1:1080",
}
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\python-selenium-spider\tmp\UserData')
options.add_argument('--disable-gpu') # 如果不加这个选项,有时定位会出现问题
# options.add_argument('--headless') # 增加无界面选项
options.add_experimental_option("prefs", {
"download.default_directory": download_dir
})
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.main(driver=driver, download_dir=download_dir, proxies=proxies)
time.sleep(86400)
finally:
pass
# driver.quit()