UI自动化 判断文件是否下载成功-Part 2

3、

# -*- coding:UTF-8 -*-
import os
from time import sleep
import hashlib
import glob

DOWNLOAD_PATH = 'D:\\download_file'

# 获取文件MD5值
def get_file_md5(f):
    m = hashlib.md5()
    with open(f, 'rb') as f:
        for line in f:
            m.update(line)
    md5code = m.hexdigest()
    return md5code

# 检查下载文件
def check_download_file():

    try:

        '''
        判断文件夹为空2种方法
        1、os.listdir()函数可以获取指定文件夹中的所有文件名，如果函数返回列表长度为0,说明此文件夹为空
        2、glob
        '''

        if len(os.listdir(DOWNLOAD_PATH)) == 0:
            print("文件夹为空")

        # if len(glob.glob(DOWNLOAD_PATH + '/*')) == 0:
        #     print("文件夹为空")

        else:
            if os.path.exists(DOWNLOAD_PATH):
                sleep(1.2)
                for file in os.listdir(DOWNLOAD_PATH):
                    file_md5 = get_file_md5(DOWNLOAD_PATH + "\\" + file)
                    if file_md5 == "72825077ee51ae9da45ae71834832325":
                        print("下载文件的MD5正确\n文件名:{0}\nMD5值:{1}".format(file, file_md5))
                    else:
                        print(file_md5)
                        print("下载文件的MD5错误")
    except:
        print("路径不存在")

if __name__ == '__main__':
    check_download_file()