選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。
 
 
 
 
 
 

188 行
6.8 KiB

import json
import os
import re
import sys
import requests
import platform
import shutil
import zipfile
import urllib.request
import winreg
import re
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():
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))
elif sys.platform == "linux":
pass
elif sys.platform == "darwin":
pass
return "115"
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/Contents/MacOS"
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__":
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"]
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.bat", "./chrome_win64/execute.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.bat", "./chrome_win32/execute.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":
pass
elif sys.platform == "darwin" and platform.architecture()[0] == "64bit":
pass
print("Done and don't forget to generate executestage EXEcutable program!")