python selenium+Chromedriver 自动 配置脚本

这个脚本能自动安装selenium库和自动安装chromedriver 只需要输入chrome.exe所在目录路径即可

import os
import requests
import time
import zipfile
import sys

def setting():
    print("\n开始安装selenium库,请耐心等待... 中途不要关闭程序\n")
    os.system('pip install selenium')      #调用dos来安装selenium
    print('\nselenium库已经安装完成！\n')
    print('-------------------------------------------------------------------------------')
    print('下面将安装chromedriver,请按照提示完成操作:\n')

    chrome_path=input('请输入chrome浏览器根目录(即chrome.exe所在文件夹 例如:C:\Program Files\Google\Chrome\Application):')
    dirs=os.listdir(chrome_path)

    flags=0
    for each in dirs:
        for one in range(70,98):
            v_str=str(one)
            if each.startswith(v_str):
                chrome_version=each
                flags=1
                break
        if flags==1:
            break                                    #获取chrome的版本号

    print('您的谷歌浏览器版本号是:',chrome_version)
    print('正在下载对应版本的chromedriver...请稍后')
    download_path='http://chromedriver.storage.googleapis.com/'+chrome_version+'/chromedriver_win32.zip'
    params={
    'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
    'Accept-Encoding': 'gzip, deflate',
    'Accept-Language': 'zh-CN,zh;q=0.9',
    'Cache-Control': 'max-age=0',
    'Connection': 'keep-alive',
    'Host': "chromedriver.storage.googleapis.com",
    "Upgrade-Insecure-Requests": "1"
    }

    def zifunpack():           #解压下载到的压缩包
        z=zipfile.ZipFile('chromedriver_win32.zip','r')
        path=sys.prefix           #获取python根目录
        z.extract(member='chromedriver.exe',path=path)
        z.close()

    while True:
        try:
            url_response=requests.get(url=download_path,params=params)
        except:
            time.sleep(5)
            continue                      #请求失败时多次尝试
        else:
            file=open('chromedriver_win32.zip','wb')
            file.write(url_response.content)
            file.close()
            print('下载完成！')
            print('--------------------------------------------------')
            break
    zifunpack()
    print('selenium已安装\n===================================\n所有环境已配置完成，您可以退出程序')

if __name__=='__main__':
    py_path=sys.prefix
    py_path_files=os.listdir(py_path)
    if 'chromedriver.exe' not in py_path_files:
        setting()
        time.sleep(0.5)
    else:
        print('环境已配置\n---------------------------------')

欢迎白嫖￣へ￣
计算机小白一枚，请大佬多多指教