Python 標準庫zipfile將文件夾加入壓縮包的操作方法

更新時間：2021年09月15日 10:33:48 作者：Likianta Me

Python zipfile 庫可用于壓縮/解壓 zip 文件. 本文介紹一下如何創(chuàng)建壓縮包，對Python zipfile壓縮包相關知識感興趣的朋友一起看看吧

zipfile模塊是python中一個處理壓縮文件的模塊，解決了不少我們平常需要處理壓縮文件的需求。大家還知道Python zipfile 庫可用于壓縮/解壓 zip 文件. 本文介紹一下如何創(chuàng)建壓縮包.

將 “文件” 加入壓縮包

假設目錄結(jié)構如下:

my_project
|- 唐詩三百首.txt  # 這是要打包的文件
|- demo.py  # 演示代碼會在這里編寫
|- _______  # 我們想要在這里生成一個名為 "output.zip" 的文件

“demo.py” 內(nèi)容如下:

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

相對路徑

"""
假設目錄結(jié)構為:

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')
    
    # 然后使用相對路徑寫入
    # 注意參數(shù)必須是文件的路徑, 不能是文件夾路徑
    # # handle.write('唐宋詩詞')  # wrong!
    handle.write('唐宋詩詞/唐詩三百首.txt')  # right
    handle.write('唐宋詩詞/宋詞三百首.txt')  # right

在這里插入圖片描述

腳本封裝

下面是封裝好的腳本, 函數(shù)比較直觀, 看代碼就可以理解用法了:

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')

在這里插入圖片描述