python - csv.reader 从请求流 : iterator should return strings, 中读取而不是字节

Question

我正在尝试使用 requests.get(url, stream=True) 将响应流式传输到 csv.reader 以处理相当大的数据源。我的代码适用于 python2.7。这是代码:

response = requests.get(url, stream=True)
ret = csv.reader(response.iter_lines(decode_unicode=True), delimiter=delimiter, quotechar=quotechar,
    dialect=csv.excel_tab)
for line in ret:
    line.get('name')

不幸的是，在迁移到 python3.6 之后，我收到了以下错误:

_csv.Error: iterator should return strings, not bytes (did you open the file in text mode?)

我试图找到一些包装器/装饰器来将 response.iter_lines() 迭代器的结果从字节转换为字符串，但没有成功。 我已经尝试使用 io 包和 codecs。使用 codecs.iterdecode 不会按行拆分数据，它可能只是按 chunk_size 拆分， 在这种情况下，csv.reader 以下列方式提示:

_csv.Error: new-line character seen in unquoted field - do you need to open the file in universal-newline mode?

Answer 1

我猜你可以将其包装在 genexp 中并向其提供解码行:

from contextlib import closing

with closing(requests.get(url, stream=True)) as r:
    f = (line.decode('utf-8') for line in r.iter_lines())
    reader = csv.reader(f, delimiter=',', quotechar='"')
    for row in reader:
        print(row)

使用 3.5 中的一些样本数据这会关闭 csv.reader，输入它的每一行都首先在 genexp 中被解码。另外，我正在使用 closing来自 contextlib原样generally suggested自动关闭响应。