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: 模型保存在哪個文件夾下
'''