Python学习

Python解压文件

import zipfile
import os
def unzip(zip_filepath, dest_path):
    """
        解压zip文件
    """
    with zipfile.ZipFile(zip_filepath) as zf:
        zf.extractall(path=dest_path)


def get_dataset_filename(zip_filepath):
    """
            获取数据库文件名
    """
    with zipfile.ZipFile(zip_filepath) as zf:
        return zf.namelist()[0]

	

	# 声明变量
	    dataset_path = './data'  # 数据集路径
	    zip_filename = 'open-food-facts.zip'  # zip文件名
	    zip_filepath = os.path.join(dataset_path, zip_filename)  # zip文件路径
	    dataset_filename = get_dataset_filename(zip_filepath)  # 数据集文件名（在zip中）
	    dataset_filepath = os.path.join(dataset_path, dataset_filename)  # 数据集文件路径
	
	    print('解压zip...', end='')
	    unzip(zip_filepath, dataset_path)
	    print('完成.')