Python 从Google驱动器拉入文件-urllib

我正试图让这段代码正常工作，但我不确定到底发生了什么。我的顾问编写了其中的一部分，我的任务是添加一些其他if语句，从不同的文件夹中提取数据。以下是URL已更改的代码：

import os
import urllib.request
import fabio
import matplotlib.pyplot as plt


def get_test_pattern(sample=None):

    if sample is None:
        data_url = 'https://drive.google.com/a/....'
        file_name = 'file_name1'
        data_dir = os.path.join('tempdata', 'drive')

    elif sample is 'water':
        data_url = 'https://drive.google.com/a/....'
        file_name = 'file_name_2'
        data_dir = os.path.join('tempdata', 'drive')

    else:
        raise ValueError('Name is invalid')

    file_path = os.path.join(data_dir, file_name)

    if not os.path.isfile(file_path):
        print('Fetching example data file')
        os.makedirs(data_dir, exist_ok=True)
        urllib.request.urlretrieve(data_url, file_path)

    image = fabio.open(file_path)
    data = image.data

    plt.imshow(data)
    plt.show()
    return data

图像打印只是为了看看我是否能得到文件，我最终会删除它。Fabio软件包只是打开了我们正在使用的特定文件类型。我留下来准备完成

当我使用参数

sample=none

运行它时，它工作正常，但当我切换到

sample='water'

时，它抛出一个“无效数据流”错误。我对Python还是相当陌生的，所以我的问题可能非常基本