mirror of
https://github.com/NaiboWang/EasySpider.git
synced 2025-04-12 03:27:08 +08:00
256 lines
9.7 KiB
Python
256 lines
9.7 KiB
Python
import json
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import requests
|
|
import platform
|
|
import shutil
|
|
import zipfile
|
|
import urllib.request
|
|
if sys.platform == "win32":
|
|
import winreg
|
|
import re
|
|
|
|
def get_processor_info():
|
|
if os.uname().sysname == 'Darwin':
|
|
processor_info = subprocess.check_output(['sysctl', '-n', 'machdep.cpu.brand_string']).strip()
|
|
processor_info = str(processor_info)
|
|
if 'Intel' in processor_info:
|
|
return 'Intel'
|
|
elif 'Apple' in processor_info:
|
|
return 'Apple'
|
|
else:
|
|
return 'Unknown'
|
|
else:
|
|
return 'This method is only implemented for macOS.'
|
|
|
|
def download_and_extract_zip(url, destination_folder):
|
|
# 下载ZIP文件
|
|
urllib.request.urlretrieve(url, "temp.zip")
|
|
|
|
# 解压ZIP文件
|
|
with zipfile.ZipFile("temp.zip", "r") as zip_ref:
|
|
zip_ref.extractall(destination_folder)
|
|
|
|
# 删除临时ZIP文件
|
|
os.remove("temp.zip")
|
|
|
|
|
|
def copy_file(source_file, destination_file):
|
|
# 使用copy2()函数复制文件
|
|
shutil.copy2(source_file, destination_file)
|
|
|
|
|
|
def copy_folder(source_folder, destination_folder):
|
|
# 使用copytree()函数复制文件夹及其内容
|
|
shutil.copytree(source_folder, destination_folder)
|
|
|
|
|
|
def get_chrome_version():
|
|
version = "131"
|
|
if sys.platform == "win32":
|
|
version_re = re.compile(r"^[1-9]\d*\.\d*.\d*")
|
|
try:
|
|
key = winreg.OpenKey(
|
|
winreg.HKEY_CURRENT_USER, r"Software\Google\Chrome\BLBeacon"
|
|
)
|
|
_v, type = winreg.QueryValueEx(key, "version")
|
|
return version_re.findall(_v)[0][:3]
|
|
except WindowsError as e:
|
|
print("check Chrome failed:{}".format(e))
|
|
else:
|
|
return version
|
|
|
|
chrome_version = get_chrome_version() # 要更新的chromedriver版本
|
|
|
|
print("Detected your chrome version is: ", chrome_version)
|
|
|
|
chrome_driver_url = "https://googlechromelabs.github.io/chrome-for-testing/known-good-versions-with-downloads.json"
|
|
win64_chrome_path = "C:\\Program Files\\Google\\Chrome\\Application"
|
|
win32_chrome_path = "C:\\Program Files\\Google\\Chrome\\Application"
|
|
mac_chrome_path = "/Applications/Google Chrome.app"
|
|
linux_chrome_path = "/opt/google/chrome"
|
|
old_driver_version = {
|
|
"100":"100.0.4896.60",
|
|
"101":"101.0.4951.41",
|
|
"102":"102.0.5005.61",
|
|
"103":"103.0.5060.134",
|
|
"104":"104.0.5112.79",
|
|
"105":"105.0.5195.52",
|
|
"106":"106.0.5249.61",
|
|
"107":"107.0.5304.62",
|
|
"108":"108.0.5359.71",
|
|
"109":"109.0.5414.74",
|
|
"110":"110.0.5481.77",
|
|
"111":"111.0.5563.64",
|
|
"112":"112.0.5615.49",
|
|
"113":"113.0.5672.63",
|
|
"114":"114.0.5735.90",
|
|
}
|
|
|
|
if __name__ == "__main__":
|
|
os.system("npm install -g extract-stealth-evasions") # 安装stealth.min.js
|
|
os.system("npx extract-stealth-evasions") # 提取stealth.min.js
|
|
driver_downloads = []
|
|
response = requests.get(chrome_driver_url)
|
|
if response.status_code == 200:
|
|
versions = json.loads(response.content)["versions"]
|
|
versions = versions[::-1] # 倒序排列数组
|
|
for info in versions:
|
|
version = info["version"].split(".")[0]
|
|
if version.find(chrome_version) == 0:
|
|
downloads = info["downloads"]
|
|
if "chromedriver" in downloads:
|
|
print(info["version"])
|
|
driver_downloads = downloads["chromedriver"]
|
|
break
|
|
else:
|
|
print("Error: " + response.status_code)
|
|
exit(1)
|
|
if not driver_downloads and int(chrome_version) < 115:
|
|
if chrome_version not in old_driver_version:
|
|
print("没有可用的chromedriver")
|
|
exit(1)
|
|
full_version = old_driver_version[chrome_version]
|
|
driver_downloads = [
|
|
{
|
|
"platform": "linux64",
|
|
"url": f"http://chromedriver.storage.googleapis.com/{full_version}/chromedriver_linux64.zip",
|
|
},
|
|
{
|
|
"platform": "mac-arm64",
|
|
"url": f"http://chromedriver.storage.googleapis.com/{full_version}/chromedriver_mac_arm64.zip",
|
|
},
|
|
{
|
|
"platform": "mac-x64",
|
|
"url": f"http://chromedriver.storage.googleapis.com/{full_version}/chromedriver_mac64.zip",
|
|
},
|
|
{
|
|
"platform": "win32",
|
|
"url": f"http://chromedriver.storage.googleapis.com/{full_version}/chromedriver_win32.zip",
|
|
},
|
|
{
|
|
"platform": "win64",
|
|
"url": f"http://chromedriver.storage.googleapis.com/{full_version}/chromedriver_win32.zip",
|
|
},
|
|
]
|
|
|
|
if os.path.exists("./chromedrivers"):
|
|
shutil.rmtree("./chromedrivers")
|
|
os.mkdir("./chromedrivers")
|
|
if sys.platform == "win32" and platform.architecture()[0] == "64bit":
|
|
for download in driver_downloads:
|
|
if download["platform"] == "win64":
|
|
url = download["url"]
|
|
print("ChromeDriver will be downloaded from: ", url)
|
|
break
|
|
download_and_extract_zip(url, "./chromedrivers")
|
|
if os.path.exists("./chrome_win64"):
|
|
shutil.rmtree("./chrome_win64")
|
|
copy_folder(win64_chrome_path, "./chrome_win64")
|
|
for folder in os.listdir("./chrome_win64"):
|
|
if folder[0].isdigit() and os.path.isdir("./chrome_win64/"+folder):
|
|
shutil.rmtree("./chrome_win64/"+folder+"/Installer") # 删除Installer文件夹
|
|
copy_file("./execute_win64.bat", "./chrome_win64/execute_win64.bat")
|
|
copy_file("./stealth.min.js", "./chrome_win64/stealth.min.js")
|
|
try:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver-win64/chromedriver.exe",
|
|
"./chrome_win64/chromedriver_win64.exe",
|
|
)
|
|
except:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver.exe",
|
|
"./chrome_win64/chromedriver_win64.exe",
|
|
)
|
|
finally:
|
|
shutil.rmtree("./chromedrivers")
|
|
elif sys.platform == "win32" and platform.architecture()[0] == "32bit":
|
|
for download in driver_downloads:
|
|
if download["platform"] == "win32":
|
|
url = download["url"]
|
|
print("ChromeDriver will be downloaded from: ", url)
|
|
break
|
|
download_and_extract_zip(url, "./chromedrivers")
|
|
if os.path.exists("./chrome_win32"):
|
|
shutil.rmtree("./chrome_win32")
|
|
copy_folder(win64_chrome_path, "./chrome_win32")
|
|
for folder in os.listdir("./chrome_win32"):
|
|
if folder[0].isdigit() and os.path.isdir("./chrome_win32/"+folder):
|
|
shutil.rmtree("./chrome_win32/"+folder+"/Installer") # 删除Installer文件夹
|
|
copy_file("./execute_win32.bat", "./chrome_win32/execute_win32.bat")
|
|
copy_file("./stealth.min.js", "./chrome_win32/stealth.min.js")
|
|
try:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver-win32/chromedriver.exe",
|
|
"./chrome_win32/chromedriver_win32.exe",
|
|
)
|
|
except:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver.exe",
|
|
"./chrome_win32/chromedriver_win64.exe",
|
|
)
|
|
finally:
|
|
shutil.rmtree("./chromedrivers")
|
|
elif sys.platform == "linux" and platform.architecture()[0] == "64bit":
|
|
for download in driver_downloads:
|
|
if download["platform"] == "linux64":
|
|
url = download["url"]
|
|
print("ChromeDriver will be downloaded from: ", url)
|
|
break
|
|
download_and_extract_zip(url, "./chromedrivers")
|
|
if os.path.exists("./chrome_linux64"):
|
|
shutil.rmtree("./chrome_linux64")
|
|
copy_folder(linux_chrome_path, "./chrome_linux64")
|
|
copy_file("./execute_linux64.sh", "./chrome_linux64/execute_linux64.sh")
|
|
copy_file("./stealth.min.js", "./chrome_linux64/stealth.min.js")
|
|
try:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver-linux64/chromedriver",
|
|
"./chrome_linux64/chromedriver_linux64",
|
|
)
|
|
except:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver",
|
|
"./chrome_linux64/chromedriver_linux64",
|
|
)
|
|
finally:
|
|
# Change Linux file permissions
|
|
os.chmod("./chrome_linux64/chromedriver_linux64", 0o755)
|
|
os.chmod("./chrome_linux64/execute_linux64.sh", 0o755)
|
|
shutil.rmtree("./chromedrivers")
|
|
elif sys.platform == "darwin" and platform.architecture()[0] == "64bit":
|
|
processor = get_processor_info()
|
|
if processor == "Intel":
|
|
driver_arch = "mac-x64"
|
|
elif processor == "Apple":
|
|
driver_arch = "mac-arm64"
|
|
for download in driver_downloads:
|
|
if download["platform"] == driver_arch:
|
|
url = download["url"]
|
|
print("ChromeDriver will be downloaded from: ", url)
|
|
break
|
|
download_and_extract_zip(url, "./chromedrivers")
|
|
if os.path.exists("./chrome_mac64.app"):
|
|
shutil.rmtree("./chrome_mac64.app")
|
|
# copy_folder(mac_chrome_path, "./chrome_mac64.app")
|
|
subprocess.call(["cp", "-R", mac_chrome_path, "./chrome_mac64.app"])
|
|
try:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver-%s/chromedriver" % driver_arch,
|
|
"./chromedriver_mac64",
|
|
)
|
|
except:
|
|
copy_file(
|
|
"./chromedrivers/chromedriver",
|
|
"./chromedriver_mac64",
|
|
)
|
|
finally:
|
|
shutil.rmtree("./chromedrivers")
|
|
os.chmod("./chromedriver_mac64", 0o755)
|
|
os.chmod("./chrome_mac64.app", 0o555)
|
|
os.chmod("./chrome_mac64.app/Contents/MacOS/Google Chrome", 0o555)
|
|
|
|
print("Done and don't forget to generate executestage EXEcutable program!")
|