Python3写gzip文件 - memoryview:需要一个类似字节的对象,而不是'str'

Question

我想写一个文件.根据文件的名称,这可能会也可能不会被gzip模块压缩.这是我的代码:

import gzip
filename = 'output.gz'
opener = gzip.open if filename.endswith('.gz') else open
with opener(filename, 'wb') as fd:
    print('blah blah blah'.encode(), file=fd)

我正在以二进制模式打开可写文件并编码我要编写的字符串.但是我收到以下错误:

File "/usr/lib/python3.5/gzip.py", line 258, in write
  data = memoryview(data)
TypeError: memoryview: a bytes-like object is required, not 'str'

为什么我的对象不是字节？如果我打开文件'w'并跳过编码步骤,我会得到相同的错误.如果我'.gz'从文件名中删除,我也会得到同样的错误.

我在Ubuntu 16.04上使用Python3.5

Answer 1

对我来说，将 gzip 标志更改为'wt'就可以了。我可以编写原始字符串，而无需“字节”它。（在 ubuntu 16 上的 python 3.5、3.7 上测试）。

来自python 3 gzip doc - 引用：“...模式参数可以是'r'，'rb'，'a'，'ab'，'w'，'wb'，'x'或'xb'中的任何一个对于二进制模式，或 'rt'、'at'、'wt' 或 'xt' 用于文本模式......”

import gzip

filename = 'output.gz'
opener = gzip.open if filename.endswith('.gz') else open
with opener(filename, 'wt') as fd:
    print('blah blah blah', file=fd)

!zcat output.gz
> blah blah blah

Answer 2

你可以把它转换成这样的字节.

import gzip 
with gzip.open(filename, 'wb') as fd:
   fd.write('blah blah blah'.encode('utf-8'))

Answer 3

print是一个相对复杂的功能。它写入str文件，但不str写入您传递的文件，而是写入str呈现参数的结果。

如果已经有字节，则可以fd.write(bytes)直接使用，并在需要时注意添加换行符。

如果没有字节，请确保fd已打开以接收文本。