Du kannst nicht mehr als 25 Themen auswählen Themen müssen entweder mit einem Buchstaben oder einer Ziffer beginnen. Sie können Bindestriche („-“) enthalten und bis zu 35 Zeichen lang sein.
 
 
 
 
 
 

228 Zeilen
8.5 KiB

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 = "120"
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__":
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.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.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":
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!")