python实现下载压缩包并且解压

from keras.utils.data_utils import get_file
import zipfile
import os
# 下载代码
file='data.zip'
dir='/data1/home/qlj/code/pytorch'
url='https://download.pytorch.org/tutorial/data.zip'

'''
函数说明 getfile：
参数:
fname: 下载后你想把这个文件保存成什么名字
origin: 下载的地址链接
cache_subdir: 模型保存在哪个文件夹下
'''
os.chdir(dir) # 将当前工作目录改变为要解压的目录
try:
    path = get_file(fname=file,
                    origin=url,cache_subdir=dir) #
except:
    print('Error')
    raise

# 解压当前目录的某个文件到当前目录下
def un_zip(file_name):  
    """unzip zip file"""  
    zip_file = zipfile.ZipFile(file_name)
    '''
    if os.path.isdir(file_name.split(".")[0]):  
        pass  
    else:  
        os.mkdir(file_name.split(".")[0])
    '''
    for names in zip_file.namelist():  
        zip_file.extract(names)  #加入到某个文件夹中 zip_file.extract(names,file_name.split(".")[0])
    zip_file.close()


un_zip(file)