将 CSV 写入临时文件时，“类型错误：需要类似字节的对象，而不是‘str’”

Question

我正在尝试测试一个使用 .csv 文件将数据写入 CSV 文件的函数tempfile.TemporaryFile。这是我正在尝试做的简化版本：

import csv
import tempfile


def write_csv(csvfile):
    writer = csv.DictWriter(csvfile, fieldnames=['foo', 'bar'])

    writer.writeheader()
    writer.writerow({'foo': 1, 'bar': 2})


def test_write_csv():
    with tempfile.TemporaryFile() as csvfile:
        write_csv(csvfile)

这似乎与csv.DictWriter记录方式一致，但是当我运行测试（使用pytest）时，出现以下错误：

============================================================ FAILURES ============================================================
_________________________________________________________ test_write_csv _________________________________________________________

    def test_write_csv():
        with tempfile.TemporaryFile() as csvfile:
>           write_csv(csvfile)

csvtest.py:14: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
csvtest.py:8: in write_csv
    writer.writeheader()
/usr/local/Cellar/python/3.7.3/Frameworks/Python.framework/Versions/3.7/lib/python3.7/csv.py:144: in writeheader
    self.writerow(header)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <csv.DictWriter object at 0x103bc46a0>, rowdict = {'bar': 'bar', 'foo': 'foo'}

    def writerow(self, rowdict):
>       return self.writer.writerow(self._dict_to_list(rowdict))
E       TypeError: a bytes-like object is required, not 'str'

知道是什么原因造成的吗？它似乎发生在rowdictis 时{'foo': 'foo', 'bar': 'bar'}，但我无法进一步确定它。

Answer 1

tempfile.TemporaryFile()默认情况下以二进制模式打开文件。您需要明确指定模式。

with tempfile.TemporaryFile(mode = "w") as csvfile: