Python产生QRCode图片 - Python练习题一

Python练习题一

import qrcode
import sys
import os
 
# 需要传入5个参数
# msg: QRCode的内容
# box_size: 图里面的像素大小，也就是每个黑点要多大，大一点整个图也会变大，预设值为 10
# border: 旁边白框大小，预设值为 2
# file_name: QRCode的图片档名, 不包含附档名
# path: QRCode的图片存档路径, 分隔字元采用斜线（/）

arg_cnt = len(sys.argv)
if arg_cnt < 2:
    print("需要传入5个参数:")
    print("qtcode_gen msg [box_size [border [file_name [path]]]")
    print("  msg: QRCode的内容")
    print("  box_size: 图里面的像素大小，也就是每个黑点要多大，大一点整个图也会变大，预设值为 10")
    print("  border: 旁边白框大小，预设值为 2")
    print("  file_name: QRCode的图片档名, 不包含附档名")
    print("  path: QRCode的图片存档路径, 分隔字元用斜线（/）")
    sys.exit(1)

# QRCode 的内容
msg = sys.argv[1]
# 二维码矩阵大小, 1~40
ver = 1
# 二维码每个小格像素值, 预设10
box_size=10 
if arg_cnt >= 3 and sys.argv[2].isdigit():
    box_size = int(sys.argv[2])

# 外框像素, 预设2
border=2
if arg_cnt >= 4 and sys.argv[3].isdigit():
    border = int(sys.argv[3])

# file_name, 图片格式固定为 png
file_name = 'qrcode_tmp.png'
if arg_cnt >= 5:
    file_name = sys.argv[4] + '.png'
# path
path = os.getcwd() #目前工作路径
if arg_cnt >= 6:
    path = sys.argv[5]

if path[-1] != '/':
    path += '/'

# 产生一个 qrcode 物件
# error_correction: 错误修正程度，设置为 High
qr = qrcode.QRCode(
    version=ver,
    error_correction=qrcode.constants.ERROR_CORRECT_H,
    box_size=box_size,
    border=border,
)

# 产生图档
qr.add_data(msg)
qr.make(fit=True) #忽略version
img = qr.make_image()

# 存档
img.save(path+file_name)
print("完成\n"+path+file_name)