Python标准库zipfile将文件夹加入压缩包代码操作方法

本篇文章小编给大家分享一下Python标准库zipfile将文件夹加入压缩包代码操作方法，文章代码介绍的很详细，小编觉得挺不错的，现在分享给大家供大家参考，有需要的小伙伴们可以来看看。

将 “文件” 加入压缩包

假设目录结构如下:

my_project
|- 唐诗三百首.txt  # 这是要打包的文件
|- demo.py  # 演示代码会在这里编写
|- _______  # 我们想要在这里生成一个名为 "output.zip" 的文件

“demo.py” 内容如下:

from zipfile import ZipFile

handle = ZipFile('output.zip', 'w')
handle.write('唐诗三百首.txt')
handle.close()

将 “文件夹” 加入压缩包

ZipFile 支持两种路径写入方式:

绝对路径

handle = ZipFile('output.zip', 'w')
handle.write('c:/aaa/bbb/唐诗三百首.txt')
handle.close()

会生成:

~/output.zip
|= aaa
    |= bbb
        |- 唐诗三百首.txt

相对路径

"""
假设目录结构为:

my_project
|= aaa
    |= bbb
        |= 唐宋诗词  # 我们想打包这个文件夹
            |- 唐诗三百首.txt
            |- 宋词三百首.txt
|- demo.py
|- _______  # 在这里生成 'output.zip' 文件
"""

# demo.py
import os
from zipfile import ZipFile

with ZipFile('output.zip', 'w') as handle:
    # 首先切到 "唐宋诗词" 的父目录
    os.chdir('./aaa/bbb')
    
    # 然后使用相对路径写入
    # 注意参数必须是文件的路径, 不能是文件夹路径
    # # handle.write('唐宋诗词')  # wrong!
    handle.write('唐宋诗词/唐诗三百首.txt')  # right
    handle.write('唐宋诗词/宋词三百首.txt')  # right

脚本封装

下面是封装好的脚本, 函数比较直观, 看代码就可以理解用法了:

import os
from zipfile import ZipFile

from lk_utils import findall_files  # pip install lk-utils


def pack_file(file_i: str, file_o: str) -> str:
    return pack_files([file_i], file_o)


def pack_files(files_i: list, file_o: str) -> str:
    backup = os.getcwd()
    
    with ZipFile(os.path.abspath(file_o), 'w') as handle:
        for file_i in map(os.path.abspath, files_i):
            dir_, filename = os.path.split(file_i)
            os.chdir(dir_)
            handle.write(filename)
    
    # restore
    os.chdir(backup)
    return file_o


def pack_dir(dir_i: str, file_o: str) -> str:
    return pack_dirs([dir_i], file_o)


def pack_dirs(dirs_i: list, file_o: str) -> str:
    backup = os.getcwd()
    
    with ZipFile(os.path.abspath(file_o), 'w') as handle:
        
        for dir_i in map(os.path.abspath, dirs_i):
            dir_ii = os.path.dirname(dir_i)
            os.chdir(dir_ii)
            
            for file in findall_files(dir_i):
                handle.write(os.path.relpath(file, dir_ii))        
    
    # restore
    os.chdir(backup)
    return file_o


if __name__ == '__main__':
    pack_file('aaa/bbb/唐宋诗词/唐诗三百首.txt', 'output1.zip')
    pack_dir('aaa/bbb/唐宋诗词', 'output2.zip')