tools.py 文件源码

def download_and_extract(f,subfolder=''):   
    print("downloading %s..."%f)


    local_filename, headers = urlretrieve(f)
    print("decompressin the zip file %s"%local_filename)
    if not os.path.isdir(extraction_directory):
        os.mkdir(extraction_directory)
    if f[-3:]=='zip':
        with zipfile.ZipFile(local_filename) as zf:
            zf.extractall(extraction_directory+subfolder)
    elif f[-2:]=='gz':
        with tarfile.open(local_filename, "r") as tar:
            tar.extractall(extraction_directory+subfolder)
    else:
        print('unrecognized archive type')
        raise
    print("done")