从Python2升级到Python3时出现字符串/字节

时间:2014-09-28 19:47:16

标签: python python-3.x python-imaging-library qr-code pillow

我尝试使用Python 3以这种方式生成base64 img的QRCode:

def gen_qrcode(data):
    import base64
    import io
    import qrcode

    qrc = qrcode.QRCode(version=1,
                        error_correction=qrcode.constants.ERROR_CORRECT_Q,
                        box_size=8,
                        border=4)
    qrc.add_data(data)
    qrc.make(fit=True)
    img = qrc.make_image()

    output = io.StringIO()
    img.save(output, 'PNG') # This line is now a problem with Python 3
    output.seek(0)
    output_s = output.read()
    b64 = base64.b64encode(output_s)
    img_tag = '<img src="data:image/png;base64,{0}">'.format(b64)

    return img_tag

它适用于Python 2(唯一更改的代码是由IO替换的StringIO)但现在我有一个错误:

TypeError at /qrcode
string argument expected, got 'bytes'
-> img.save(output, 'PNG')

有什么想法吗? 感谢。

1 个答案:

答案 0 :(得分:5)

output = io.BytesIO

这将期望字节并产生输入到base64.b64encode的字节。

删除领先&#34; b&#34;从结果输出.decode()必须按照评论中提到的那样使用:

b64 = base64.b64encode(output_s).decode()