Как использовать CSV Writers с файлами GZIP в Python 3?

Question

Я пытаюсь перенести некоторый код из Python 2.7 в Python 3. Инструмент 2to3 отлично работает для базового синтаксиса и изменений пакета, но теперь мы сталкиваемся с некоторыми странными побочными эффектами.

У меня есть следующий блок кода. Он открывает временное имя файла, используя модуль gzip.

f = NamedTemporaryFile(delete=False) 
f.close() 
fn = f.name + '.gz' 
os.rename(f.name, fn) 
fz = gzip.open(fn, 'wb') 
writer = csv.writer(fz, delimiter='\t', lineterminator=lt) 
for row in table: 
    writer.writerow(row) 
fz.close() 

Проблема заключается в том, что выполнение этого дает мне следующую ошибку:

File "/usr/local/Cellar/python3/3.4.2_1/Frameworks/Python.framework/Versions/3.4/lib/python3.4/gzip.py", line 343, in write 
self.crc = zlib.crc32(data, self.crc) & 0xffffffff 
TypeError: 'str' does not support the buffer interface 

Я попытался открыть файл GZIP как «ш» вместо «термометра», но безрезультатно. Я предполагаю, что модуль gzip ожидает массив байтов, но CSV Writer не предоставляет или не предоставляет ничего, кроме строки.

Как люди делают это в Python 3?

Редактировать: Следует упомянуть, что этот блок кода выполняется без проблем в Python 2.7.

Answer 1

Вам нужно изменить режим gzip к wt:

fz = gzip.open(fn, 'wt') 

Также малоизвестная особенность gzip.open() и bz2.open() является то, что они могут быть многоуровневыми поверх существующего файла, открытого в бинарном режиме. Например, это работает:

import gzip 
f = open('somefile.gz', 'rb') 
with gzip.open(f, 'rt') as g: 
    text = g.read() 

Это позволяет GZIP и BZ2 модули для работы с различными файловыми как объекты, такие как розетки, трубы, и в памяти файлов.