python - 从 BytesIO 解压缩 bz2 文件

Question

我想读取服务器中的bz2文件，将其解压并使用csv解析器读取，但仍然有错误；

    myfile = bz2.BZ2File(bio.read(), "rb")
TypeError: file() argument 1 must be encoded string without NULL bytes, not str

import paramiko
from config import config
import bz2
import csv
import StringIO
from io import BytesIO
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(config.get('mrc_ssh', 'host'), username=config.get('mrc_ssh', 'user'))
sftp_client = ssh.open_sftp()
_file = sftp_client.open('/home/myfile.bz2')
bio = BytesIO(_file.read())
print bio
myfile = bz2.BZ2File(bio.read(), "rb")
reader = csv.DictReader(myfile)
for row in reader:
    print row

Answer 1

bz2.BZ2File将文件名作为第一个参数。不是实际数据。

使用(如果您可以在本地存储文件):

myfile = bz2.BZ2File('/home/myfile.bz2', "rb")

或者使用一次性解压功能bz2.decompress

uncompressed_data = bz2.decompress(bio.read())