下载压缩文件并解压的经典代码

下载压缩文件并解压的经典代码

摘取自Tensorflow下载数据库的源码，笔者较为赞同这段代码，写代码时可以借鉴与参考

代码详情

1.文件名分割，本地创建文件夹
2.定义下载情况显示情况函数
3.下载文件
4.解压

def download_and_uncompress_tarball(tarball_url, dataset_dir):
    """Downloads the `tarball_url` and uncompresses it locally.
    Args:
        tarball_url: The URL of a tarball file.
        dataset_dir: The directory where the temporary files are stored.
    """
    filename = tarball_url.split('/')[-1]
    filepath = os.path.join(dataset_dir, filename)

    def _progress(count, block_size, total_size):
        sys.stdout.write('\r>> Downloading %s %.1f%%' % (
            filename, float(count * block_size) / float(total_size) * 100.0))
        sys.stdout.flush()
    filepath, _ = urllib.request.urlretrieve(tarball_url, filepath, _progress)
    print()
    statinfo = os.stat(filepath)
    print('Successfully downloaded', filename, statinfo.st_size, 'bytes.')
    tarfile.open(filepath, 'r:gz').extractall(dataset_dir)