Проблема с tmpfile и gzip - PullRequest
Новая
       13

Проблема с tmpfile и gzip

8 голосов
/ 09 апреля 2010

У меня проблема с этим кодом: 

file = tempfile.TemporaryFile(mode='wrb')
file.write(base64.b64decode(data))
file.flush()
os.fsync(file)
# file.seek(0)
f = gzip.GzipFile(mode='rb', fileobj=file)
print f.read()

Я не знаю, почему он ничего не распечатывает. Если я раскомментирую file.seek, тогда произойдет ошибка: 

  File "/usr/lib/python2.5/gzip.py", line 263, in _read
    self._read_gzip_header()
  File "/usr/lib/python2.5/gzip.py", line 162, in _read_gzip_header
    magic = self.fileobj.read(2)
IOError: [Errno 9] Bad file descriptor

Только для информации, эта версия отлично работает: 

x = open("test.gzip", 'wb')
x.write(base64.b64decode(data))
x.close()
f = gzip.GzipFile('test.gzip', 'rb')
print f.read()

РЕДАКТИРОВАТЬ: Для проблемы wrb. Это не дает мне ошибку при инициализации. Python 2.5.2. 

>>> t = tempfile.TemporaryFile(mode="wrb")
>>> t.write("test")
>>> t.seek(0)
>>> t.read()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IOError: [Errno 9] Bad file descriptor

1 Ответ

11 голосов
/ 09 апреля 2010

'wrb' не является допустимым режимом.

Это отлично работает: 

import tempfile
import gzip

with tempfile.TemporaryFile(mode='w+b') as f:
    f.write(data.decode('base64'))
    f.flush()
    f.seek(0)
    gzf = gzip.GzipFile(mode='rb', fileobj=f)
    print gzf.read()
...