Python 2.x到3.x转换的代码在IO.StringIO转换中失败

时间:2018-03-06 13:55:12

标签: python python-3.x io

python 2.x代码

sdata = StringIO.StringIO(c.data)

python 3.x代码

sdata = io.StringIO(c.data)

完整代码

def downloadCSV(c, d):
    filename = c.getFilename(d)
    reqstr = c.getReqStr(d)

    print(("Downloading %s ..." % (filename)))
    if c.getResponse(reqstr) == -1:
        return -1
    sdata = io.StringIO(c.data)
    z = zipfile.ZipFile(sdata)

错误:

Traceback (most recent call last):
  File "xxx.py", line 166, in <module>
    main(sys.argv[1:])
  File "xxxx.py", line 158, in main
    getMonth(c, args[1], args[2])
  File "xxxx.py", line 133, in getMonth
    if downloadCSV(c, d) > -1:
  File "xxxx.py", line 73, in downloadCSV
    sdata = io.StringIO(c.data)
TypeError: initial_value must be str or None, not bytes

对3.x python版本进行正确导入,sdata的转换应该自动发生在这里?为什么会出现上述错误,以及在python3.x中纠正此错误的方法是什么。试过在这个论坛上发布的其他答案,但在这种情况下似乎没有任何工作。

1 个答案:

答案 0 :(得分:2)

Python 3现在可以区分strbytes

您的请求返回了二进制数据,因此为了能够将其存储在io对象中,您必须创建BytesIO对象而不是StringIO对象。所以:

sdata = io.BytesIO(c.data) 

请注意,代码仍与Python 2兼容。