python—压缩文件

import os
import zipfile
#挑选文件夹中.log结尾的文件
def find_file(logsdir):
    file_list=[]
    #处理logsdir路径下文件
    for files in os.listdir(logsdir):
        #如果文件以.log结尾则添加到列表中
        if files.endswith(".log"):
            file_list.append(files)
    return file_list
#压缩文件
def compress_file(logdir,file_list):
    #压缩后文件夹名
    fileName = "logall.zip"
    #压缩后文件夹路径
    storagePath = logsdir + os.sep + fileName
    #创建实例
    z = zipfile.ZipFile(storagePath, 'w')
    #将列表中每一个文件写入到添加到压缩文件中
    for file in file_list:
        z.write(logsdir + os.sep + file)
    z.close()
if __name__ == '__main__':
    logsdir = r'/Users/**/'
    file_list = find_file(logsdir)
    compress_file(logsdir, file_list)