在一个视图中，我使用一个简单的csv编写器创建了一个完全由csv组成的Django HttpResponse对象：
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="foobar.csv"'
writer = csv.writer(response)
table_headers = ['Foo', 'Bar']
writer.writerow(table_headers)
bunch_of_rows = [['foo', 'bar'], ['foo2', 'bar2']]
for row in bunch_of_rows:
writer.writerow(row)
return response
在单元测试中，我想测试此csv的某些方面，因此我需要阅读它。我正在尝试这样做：
response = views.myview(args)
reader = csv.reader(response.content)
headers = next(reader)
row_count = 1 + sum(1 for row in reader)
self.assertEqual(row_count, 3) # header + 1 row for each attempt
self.assertIn('Foo', headers)
但是测试失败并显示以下内容headers = next(reader)：
nose.proxy.Error: iterator should return strings, not int (did you open the file in text mode?)
我在HttpResponse源中看到response.content将字符串拆分为字节字符串，但是我不确定如何正确处理csv.reader该文件以正确读取文件。我以为我可以替换response.content为response（因为您写入对象本身，而不是对象的内容），但这只会导致错误稍有变化：
_csv.Error: iterator should return strings, not bytes (did you open the file in text mode?)
这似乎更接近，但显然仍然是错误的。阅读csv文档，我认为我无法正确打开文件。我如何“打开”这个类似文件的对象，以便csv.reader可以对其进行解析？
response.content提供字节。您需要将其解码为字符串：
foo = response.content.decode('utf-8')
然后使用io.StringIO将此字符串传递给csv阅读器：
import io
reader = csv.reader(io.StringIO(foo))
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句