Scrittura su io.BytesIO in csv non riesce in python3

Question

Sto provando a scrivere codice python 2/3 compatibile per scrivere stringhe sull'oggetto file csv. Questo codice:

line_as_list = [line.encode() for line in line_as_list] 
writer_file = io.BytesIO() 
writer = csv.writer(writer_file, dialect=dialect, delimiter=self.delimiter) 
for line in line_as_list: 
    assert isinstance(line,bytes) 
    writer.writerow(line) 

dà questo errore su python3:

>   writer.writerow(line) 
E   TypeError: a bytes-like object is required, not 'str' 

Ma affermare non ha alcun problema con il tipo, quindi perché è csv creazione di un errore?

Non è possibile utilizzare BytesIO solo per Python 2 e 3? Dov'è il problema qui?

Answer 1

In Python3 csv.writer si aspetta un oggetto simile a un file aperto in modalità testo. In Python2, csv.writer si aspetta un oggetto simile a un file aperto in modalità binaria.

Pertanto, in python3, utilizzare io.StringIO, mentre in python2 utilizzare io.BytesIO:

import io 
import csv 
import sys 
PY3 = sys.version_info[0] == 3 

line_as_list = [u'foo', u'bar'] 
encoding = 'utf-8' 

if PY3: 
    writer_file = io.StringIO() 
else: 
    writer_file = io.BytesIO() 
    line_as_list = [line.encode(encoding) for line in line_as_list] 

writer = csv.writer(writer_file, dialect='excel', delimiter=',') 
writer.writerow(line_as_list) 
content = writer_file.getvalue() 

if PY3: 
    content = content.encode(encoding) 

print(type(content)) 
print(repr(content)) 

---

In python3 codice precedente stampe

<class 'bytes'> 
b'foo,bar\r\n' 

In python2 codice precedente stampe

<type 'str'> 
'foo,bar\r\n'