sliauth.py 文件源码

def gen_qr_code(text, border=4, pixel=15, raw_image=False, img_html=''):
    try:
        import qrcode
    except ImportError:
        if img_html:
            return 'Install <code>pillow/qrcode</code> packages for QR code'
        raise Exception('Please install pillow and qrcode packages, e.g., conda install pillow; pip install qrcode')

    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=pixel,
        border=border,
        )

    qr.add_data(text)
    qr.make(fit=True)

    img = qr.make_image()

    img_io = io.BytesIO()
    img.save(img_io, "png")
    img_io.seek(0)
    img_data = img_io.getvalue()

    if raw_image:
        return img_data
    else:
        data_uri = "data:image/gif;base64,"+img_data.encode("base64")
        if img_html:
            return img_html % data_uri
        else:
            return data_uri