从BytesIO解压缩bz2文件

时间:2015-03-24 15:09:46

标签: python csv paramiko

我想在服务器中读取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

1 个答案:

答案 0 :(得分:2)

bz2.BZ2File filename 作为第一个参数。不是实际数据。

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

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

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

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