schreiben Nicht-Unicode mit csv-Modul

Question

Während Python Migration 3, bemerkte ich einige Dateien, die wir den eingebauten in csv jetzt b' Präfix haben um jede Saite erzeugen mit ...

Hier ist der Code, das sollte eine generieren CSV für eine Liste von dogs, nach einigen von export_fields definierten Parameter (also gibt immer Unicode-Daten):

file_content = StringIO() 
csv_writer = csv.writer(
    file_content, delimiter='\t', quotechar='"', quoting=csv.QUOTE_MINIMAL 
) 
csv_writer.writerow([ 
    header_name.encode('cp1252') for _v, header_name in export_fields 
]) 
# Write content 
for dog in dogs: 
    csv_writer.writerow([ 
     get_value(dog).encode('cp1252') for get_value, _header in export_fields 
    ]) 

Das Problem ist, wenn ich file_content.getvalue() zurückkehrt, erhalte ich:

b'Does he bark?' b'Full  Name' b'Gender' 
b'Sometimes, yes' b'Woofy the dog' b'Male' 

Statt ^{_{(Vertiefung modifiziert wurde auf SO lesbar zu sein)}}:

'Does he bark?' 'Full  Name' 'Gender' 
'Sometimes, yes' 'Woofy the dog' 'Male' 

ich keine encoding Parameter im csv Modul gefunden. Ich möchte, dass die ganze Datei in cp1252 kodiert wird, also ist es mir egal, ob die Codierung durch die Iteration der Zeilen oder die selbst erstellte Datei erfolgt.

Also weiß jemand, wie man eine richtige Zeichenfolge erzeugt, die nur cp1252-codierte Zeichenfolgen enthält?

Answer 1

Die csv Modul befasst sich mit Text und wandelt alles, was mit str() keine Zeichenfolge in eine Zeichenfolge ist.

Nicht in bytes Objekte übergeben. Übergeben Sie str Objekte oder Typen, die sauber in Zeichenfolgen mit str() konvertieren. Das bedeutet, dass Sie keine Zeichenfolgen codieren sollten.

Wenn Sie cp1252 Ausgang benötigen, codieren die StringIO Wert:

file_content.getvalue().encode('cp1252') 

als StringIO Objekte auch nur in Text befassen.

noch besser, ein BytesIO object mit einem TextIOWrapper() verwenden für Sie die Codierung zu tun, wie das csv Modul an das Dateiobjekt schreibt:

from io import BytesIO, TextIOWrapper 

file_content = BytesIO() 
wrapper = TextIOWrapper(file_content, encoding='cp1252', line_buffering=True) 
csv_writer = csv.writer(
    wrapper, delimiter='\t', quotechar='"', quoting=csv.QUOTE_MINIMAL) 

# write rows 

result = file_content.getvalue() 

I Line-Pufferung auf der Verpackung aktiviert haben, so dass es‘ Wird bei jedem Schreiben einer Zeile automatisch an die BytesIO Instanz gesendet.

Jetzt produziert file_content.getvalue() Bytestring:

>>> from io import BytesIO, TextIOWrapper 
>>> import csv 
>>> file_content = BytesIO() 
>>> wrapper = TextIOWrapper(file_content, encoding='cp1252', line_buffering=True) 
>>> csv_writer = csv.writer(wrapper, delimiter='\t', quotechar='"', quoting=csv.QUOTE_MINIMAL) 
>>> csv_writer.writerow(['Does he bark?', 'Full  Name', 'Gender']) 
36 
>>> csv_writer.writerow(['Sometimes, yes', 'Woofy the dog', 'Male']) 
35 
>>> file_content.getvalue() 
b'Does he bark?\tFull  Name\tGender\r\nSometimes, yes\tWoofy the dog\tMale\r\n'