python实现压缩与解压

压缩文件的格式有zip、rar、tar.gz等，这里主要介绍tar.gz,zip文件的压缩与解压。

文件压缩与解压一般分为三个步骤：

1.打开文件

2.添加/解压文件

3.关闭文件

一、zip文件压缩与解压

1、压缩

将原文件夹中的数据内容提取出来放到打开的zipfile中

import zipfile

def make_zip(source_dir, desc_dir):
    zfile = zipfile.ZipFile(desc_dir, "w")
    try:
        for root, dirnames, filenames in os.walk(source_dir):
            file_path = root.replace(source_dir, '')  # 去掉根路径，只对目标文件夹下的文件及文件夹进行压缩
            # 循环出一个个文件名
            for filename in filenames:
                zfile.write(os.path.join(root, filename), os.path.join(file_path, filename))
        return True
    except Exception as e:
        print(e)
    zip.close()

2、解压

def unzip(source_dir, desc_dir):
    try:
        zfile = zipfile.ZipFile(source_dir)
        zfile.extractall(desc_dir)
        return True
    except Exception as e:
        print(e)

二、tar文件压缩与解压

1.压缩

import os, tarfile


# 一次性打包整个根目录。空子目录会被打包。
# 如果只打包不压缩，将"w:gz"参数改为"w:"或"w"即可。
def make_targz(output_filename, source_dir):
    with tarfile.open(output_filename, "w:gz") as tar:
        tar.add(source_dir, arcname=os.path.basename(source_dir))


# 逐个添加文件打包，未打包空子目录。可过滤文件。
# 如果只打包不压缩，将"w:gz"参数改为"w:"或"w"即可。
def make_targz_one_by_one(output_filename, source_dir):
    tar = tarfile.open(output_filename, "w:gz")
    for root, dir, files in os.walk(source_dir):
        for file in files:
            pathfile = os.path.join(root, file)
            tar.add(pathfile)
    tar.close()

 

2.解压

def untar(fname, dirs):
    """
    解压tar.gz文件
    :param fname: 压缩文件名
    :param dirs: 解压后的存放路径
    :return: bool
    """
    try:
        t = tarfile.open(fname)
        t.extractall(path=dirs)
        return True
    except Exception as e:
        print(e)
        return False

 THANKS A LOT!