python下载并且解压

# -*- coding: UTF-8 -*-
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'

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)

'''
函数说明
getfile：
参数:

fname: 下载后你想把这个文件保存成什么名字

origin: 下载的地址链接

cache_subdir: 模型保存在哪个文件夹下
'''