手把手教你用Python实现批量文件的压缩处理

一、前言

大家好，我是崔艳飞。接到项目求助，需要对上千个文件夹中的文件进行压缩处理，并要删除源文件，只保留压缩后的压缩文件，数据量大，手动完成耗时耗力，用Python处理再合适不过了。

二、项目目标

批量对文件夹的内容进行压缩处理，满足客户要求。

三、项目准备

软件：PyCharm

需要的库：os， shutil，zipfile

四、项目分析

1）如何读取源文件？

利用OS库，获取文件夹名list，利用for循环，轻松拿到要压缩的源文件。

2）如何进行压缩处理？

利用zipfile库中的zipfile.ZipFile()对获取到的文件进行压缩处理。

3）如何删除源文件？

先利用os库的remove()删除文件，再利用shutil库的rmtree()删除空文件夹。

五、项目实现

1、第一步导入需要的三个库


import os as os

import shutil

import zipfile

2、第二步定义删除文件函数和压缩文件函数


def del_(rootdir):
    filelist = []
    filelist = os.listdir(rootdir)  # 列出该目录下的所有文件名
    for f in filelist:
        filepath = os.path.join(rootdir, f)  # 将文件名映射成绝对路劲
        if os.path.isfile(filepath):  # 判断该文件是否为文件或者文件夹
            os.remove(filepath)  # 若为文件，则直接删除
        elif os.path.isdir(filepath):
            shutil.rmtree(filepath, True)  # 若为文件夹，则删除该文件夹及文件夹内所有文件
    shutil.rmtree(rootdir, True)
def zipDir(dirpath,outFullName):
    zip = zipfile.ZipFile(outFullName,"w",zipfile.ZIP_DEFLATED)
    for path,dirnames,filenames in os.walk(dirpath):
        # 去掉目标跟路径，只对目标文件夹下边的文件及文件夹进行压缩
        fpath = path.replace(dirpath,'')

        for filename in filenames:
            zip.write(os.path.join(path,filename),os.path.join(fpath,filename))
    zip.close()

3、第三步创建主函数


def main():
    path_end = 'D:/a/h/'
    date= os.listdir(path_end)
    # 获取目标文件夹所有文件夹名列表
    for f in date:
        ljbc='D:/a/h/'+f+'/'+'查询信息.zip'
        ljbc2 = 'D:/a/h/' + f + '/' + '下发修改.zip'
        #以上两行是创建压缩后的文件名
        ljcx='D:/a/h/'+f+'/查询信息'
        ljxf = 'D:/a/h/' + f + '/下发修改'
        #以上两行是要压缩的源文件
        zipDir(ljcx,ljbc)
        zipDir(ljxf, ljbc2)
        del_(ljcx)
        del_(ljxf)

if __name__ == '__main__':
    main()