Python 3 でQRコードを読み書き

#!/usr/bin/env/python3
# coding: utf-8

import numpy as np
import cv2
import subprocess

# zbar command
ZBARIMG = 'zbarimg'
    
def passzbar(image): 
    # convert to bmp binary so that zbar can handle it
    retval, buf = cv2.imencode('.bmp', image)
    if retval == False:
        raise ValueError('The Given image could not be converted to BMP binary data')
    # convert buf from numpy.ndarray to bytes
    binbmp = buf.tobytes()
    optionargs = []
    
    args = [
        ZBARIMG,
        ':-', '-q'
    ] + optionargs
    
    p = subprocess.Popen(
        args,
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        shell=False
        )
        
    stdout, stderr = p.communicate(input=binbmp)
    if len(stderr) == 0:
        bindata = stdout
    else:
        raise RuntimeError('ZBar threw error:\n' + stderr.decode('utf-8'))
        
    type, data = bindata.split(b":", 1)
    return type, data
    
def main():
    testpic = cv2.imread('test.png', cv2.IMREAD_GRAYSCALE)
    bartype, bardata = passzbar(testpic)
    print(bardata.decode('utf-8'))

if __name__ == '__main__':
    main()

Python bindings からサクっと使うはずだったのに完全に予定が狂ってしまった…。

QRコード生成

さて、生成する方はずっと楽。
python-qrcode パッケージをインストールする。
qrcode 5.2.2 : Python Package Index

pip install qrcode

上で挙げた pypi のサイトに解説がある通りだが、簡単には次のようにできる。

import qrcode
img = qrcode.make("ここにデータが入ります")
img.save("qr.png")

更に細かく設定することもできる。内部で PIL を利用してるらしく、 qrcode.make_image() の出力は PIL.Image のようである。 img.save() に io.BytesIO などを与えることもできるようだ。

import qrcode

qr = qrcode.QRCode(error_correction=qrcode.constants.ERROR_CORRECT_L)
qr.add_data("ここにデータが入ります")
qr.make(fit=True)
img = qr.make_image()
img.save("qr.png")