Python判断IEDriverServer是否最新版本并自动更新

之前写了一篇判断Chromedriver是否最新并自动下载的文章，这次尝试实现了IEDriverServer的自动更新。
火狐的geckodriver 是在github，暂时还没想到怎么实现。

这次与上一篇的不同，主要在于页面接口返回的是XML，所以使用了lxml进行解析。同时使用了之前自己写的一个比较版本号大小的函数。

2019.03.17更新：
原本是使用的xml.etree.ElementTree进行解析，但是使用xpath获取元素列表的时候，发现每个节点都需要使用xml的namespace作为前缀，但是ElementTree模块貌似没有获取namespace的方法，搜索好久之后，发现lxml模块有个方法可以获取（lxml.tree.nsmap[None]），所以改用了lxml模块。

简单说下思路：
1、获取版本号（解析xml，并通过xpath获取）
xpath为：’.//{name_space}CommonPrefixes/{name_space}Prefix’
因为该xml有namespace，所以每个节点都需要加前缀，比较麻烦，所以将namespace提取出来作为变量

2、拼接下载地址
如图，获取到最新的版本号后，下载地址里面只需要相应修改2个地方就可以了

代码如下：

#!/usr/bin/env python
# coding=UTF-8
'''
@Description: 用于将IEDriverServer更新为最新版本
@Author: wjx
@LastEditors: Please set LastEditors
@Date: 2019-03-15 10:56:34
@LastEditTime: 2019-03-17 00:52:33

依据网页上最新版本的命名规则获取最新的文件，所以如果网站命名规则改变的话，可能会获取到旧版
如果获取的版本不是最新，请手动到网站（http://selenium-release.storage.googleapis.com/index.html/）获取
'''

import requests
import re
import os
import zipfile
from lxml import etree


IEDRIVERSERVER_URL = 'http://selenium-release.storage.googleapis.com/'
IEDRIVERSERVER_API = f'{IEDRIVERSERVER_URL}?delimiter=/&prefix='  # 下载地址
PATH = ".\\drivers\\"

def get_latest_version(url):
    '''查询最新的IEDriverServer版本'''
    rep = requests.get(url)
    tree = etree.fromstring(rep.content)   # 接口返回的是XML文件，需要进行解析
    name_space = '{' + tree.nsmap[None] + '}' # 获取XML的namespace
    latest_version = '0'
    for tag in tree.findall(f'.//{name_space}CommonPrefixes/{name_space}Prefix'):  # 通过xpath定位，每个节点都有namespace
        version = tag.text[:-1]
        if version.replace('.', '').isdigit():
            latest_version = compare(latest_version, version) or latest_version  # 如果返回为None，说明相等，则直接拿上次的结果  
    return latest_version

def download_driver(download_url):
    '''下载文件'''
    file = requests.get(download_url)
    with open(f"{PATH}IEDriverServer.zip", 'wb') as zip_file:        # 保存文件到脚本所在目录
        zip_file.write(file.content)
        print('下载成功')

def get_version():
    '''查询指定位置的IEDriverServer版本'''
    # path = os.getcwd() # 获取当前工作路径
    outstd2 = os.popen(f'{PATH}IEDriverServer.exe --version').read()
    return outstd2.split(' ')[1]

def unzip_driver():
    '''解压IEDriverServer压缩包到指定目录'''
    f = zipfile.ZipFile(f"{PATH}IEDriverServer.zip", 'r')
    for file in f.namelist():
        f.extract(file, PATH)

def compare(a: str, b: str):
    '''比较两个版本的大小，需要按.分割后比较各个部分的大小'''
    lena = len(a.split('.'))  # 获取版本字符串的组成部分
    lenb = len(b.split('.'))
    a2 = a + '.0' * (lenb-lena)  # b比a长的时候补全a
    b2 = b + '.0' * (lena-lenb)
    for i in range(max(lena, lenb)):  # 对每个部分进行比较，需要转化为整数进行比较
        if int(a2.split('.')[i]) > int(b2.split('.')[i]):
            return a
        elif int(a2.split('.')[i]) < int(b2.split('.')[i]):
            return b
        else:						# 比较到最后都相等，则返回None用于判断相等的情况
            if i == max(lena, lenb)-1:
                return None

if __name__ == "__main__":
    latest_version = get_latest_version(IEDRIVERSERVER_API)
    print('最新的IEDriverServer版本为：', latest_version)
    version = get_version()
    print('当前系统内的IEDriverServer版本为：', version)
    if compare(version, latest_version):  # 如果有返回值，说明不相等
        print('当前系统内的IEDriverServer不是最新的，需要进行更新')
        download_url = IEDRIVERSERVER_URL + latest_version + '/IEDriverServer_x64_' + latest_version + '.0.zip'  # 拼接下载链接
        print('文件下载路径：', download_url)
        download_driver(download_url)
        unzip_driver()
        print('更新后的IEDriverServer版本为：', get_version())
    else:
        print('当前系统内的IEDriverServer已经是最新的')