快捷導(dǎo)航

Python文件的壓縮與解壓

更新時(shí)間：2022年04月15日 13:48:47 作者：淺若清風(fēng)cyf?

這篇文章主要介紹了Python文件的壓縮與解壓，Python進(jìn)行文件、文件夾壓縮與解壓，用到的是zipfile的第三方依賴庫(kù)。根據(jù)不同應(yīng)用場(chǎng)景封裝了幾個(gè)方法，下文相關(guān)內(nèi)容需要的小伙伴可以參考一下

前言：

Python在人工智能，后臺(tái)服務(wù)等領(lǐng)域中得到了廣泛應(yīng)用。由于python有著大量的第三方開(kāi)源包以及其易用的語(yǔ)言特性，這門(mén)語(yǔ)言在數(shù)據(jù)處理、批處理等方法具有強(qiáng)大的優(yōu)勢(shì)，使用起來(lái)相當(dāng)便捷。本文為大家?guī)?lái)的是使用Python進(jìn)行文件、文件夾壓縮與解壓，用到的是zipfile的第三方依賴庫(kù)。根據(jù)不同應(yīng)用場(chǎng)景封裝了幾個(gè)方法，方便根據(jù)實(shí)際需要使用相應(yīng)的方法，能夠篩選出需要壓縮的文件。

安裝：

pip install zipfile

代碼：

1、壓縮整個(gè)文件夾

import zipfile
import os
from tqdm import tqdm

def zip_file(src_dir, save_name='default'):
    '''
    壓縮文件夾下所有文件及文件夾
    默認(rèn)壓縮文件名：文件夾名
    默認(rèn)壓縮文件路徑：文件夾上層目錄
    '''
    if save_name == 'default':
        zip_name = src_dir + '.zip'
    else:
        if save_name is None or save_name == '':
            zip_name = src_dir + '.zip'
        else:
            zip_name = save_name + '.zip'

    z = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED)
    for dirpath, dirnames, filenames in os.walk(src_dir):
        fpath = dirpath.replace(src_dir, '')
        fpath = fpath and fpath + os.sep or ''
        for filename in filenames:
            z.write(os.path.join(dirpath, filename), fpath + filename)
    z.close()
    return True

2、壓縮指定擴(kuò)展名文件

def zip_file_by_extension(src_dir, extension='dll', save_name='default'):
    '''
    根據(jù)文件擴(kuò)展名篩選需要壓縮的文件到壓縮包
    注：針對(duì)單層目錄情況，文件夾下的子文件不會(huì)被壓縮
    '''
    if save_name == 'default':
        zip_name = src_dir + '.zip'
    else:
        if save_name is None or save_name == '':
            zip_name = src_dir + '.zip'
        else:
            zip_name = save_name + '.zip'

    z = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED)
    file_list = os.listdir(src_dir)
    for filename in tqdm(file_list):
        if not os.path.isfile(os.path.join(src_dir, filename)):
            continue

        if filename.split('.')[-1] == extension:
            print(r'正在壓縮文件：', filename, end='')
            z.write(os.path.join(src_dir, filename), filename)
    z.close()
    return True

3、獲取指定類(lèi)型的待壓縮文件列表

def get_zip_file_list(src_dir, extension='dll'):
    '''
    獲取需要壓縮的文件列表
    '''
    file_list = os.listdir(src_dir)
    zip_list = []
    for filename in file_list:
        if not os.path.isfile(os.path.join(src_dir, filename)):
            continue

        if filename.split('.')[-1] == extension:
            zip_list.append(filename)
    print(zip_list)

4、解壓文件

def unzip_file(zip_src, dst_dir):
    '''
    解壓縮
    '''
    r = zipfile.is_zipfile(zip_src)
    if r:
        fz = zipfile.ZipFile(zip_src, 'r')
        for file in fz.namelist():
            fz.extract(file, dst_dir)
    else:
        print('This is not zip')
        return False
    return True

到此這篇關(guān)于Python文件的壓縮與解壓的文章就介紹到這了,更多相關(guān)Python文件壓縮解壓內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: