249 lines
11 KiB
Python
249 lines
11 KiB
Python
|
# =================================================
|
|||
|
|
# =============== 二维码 - 任务管理器 ===============
|
|||
|
|
# =================================================
|
|||
|
|
|
|||
|
|
import base64
|
|||
|
|
from PIL import Image, ImageEnhance, ImageFilter
|
|||
|
|
from io import BytesIO
|
|||
|
|
from .mission import Mission
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import zxingcpp
|
|||
|
|
except Exception as e:
|
|||
|
|
zxingcpp = None
|
|||
|
|
zxingcppErr = str(e)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class _MissionQRCodeClass(Mission):
|
|||
|
|
|
|||
|
|
def createImage(self, text, format="QRCode", w=0, h=0, quiet_zone=-1, ec_level=-1):
|
|||
|
|
"""
|
|||
|
|
生成二维码图片
|
|||
|
|
format: "Aztec","Codabar","Code128","Code39","Code93","DataBar","DataBarExpanded","DataMatrix","EAN13","EAN8","ITF","LinearCodes","MatrixCodes","MaxiCode","MicroQRCode","PDF417","QRCode","UPCA","UPCE",
|
|||
|
|
quiet_zone: 四周的空闲区域
|
|||
|
|
ec_level：纠错等级，-1 - 自动, 1- L-7% , 0 - M-15%, 3 - Q-25%, 2 - H-30%
|
|||
|
|
纠错仅用于Aztec、PDF417和QRCode
|
|||
|
|
返回：成功返回PIL对象，失败返回错误原因字符串
|
|||
|
|
"""
|
|||
|
|
# 转整数
|
|||
|
|
w, h = round(w), round(h)
|
|||
|
|
quiet_zone, ec_level = round(quiet_zone), round(ec_level)
|
|||
|
|
# 生成格式对象
|
|||
|
|
bFormat = getattr(zxingcpp.BarcodeFormat, format, None)
|
|||
|
|
if not bFormat:
|
|||
|
|
return f"[Error] format {format} not in zxingcpp.BarcodeFormat!"
|
|||
|
|
try:
|
|||
|
|
bit = zxingcpp.write_barcode(bFormat, text, w, h, quiet_zone, ec_level)
|
|||
|
|
except Exception as e:
|
|||
|
|
return f"[Error] [{format}] {e}"
|
|||
|
|
try:
|
|||
|
|
img = Image.fromarray(bit, "L")
|
|||
|
|
except Exception as e:
|
|||
|
|
return f"[Error] Image.fromarray: {e}"
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
# msnInfo: { 回调函数"onXX" , 参数"argd":{ "preprocessing.xx" } }
|
|||
|
|
# msnList: [ { "path", "pil", "base64" } ]
|
|||
|
|
# def addMissionList(self, msnInfo, msnList): # 添加任务列表
|
|||
|
|
# return super().addMissionList(msnInfo, msnList)
|
|||
|
|
def msnTask(self, msnInfo, msn): # 执行msn
|
|||
|
|
# 导入库失败
|
|||
|
|
if not zxingcpp:
|
|||
|
|
return {
|
|||
|
|
"code": 901,
|
|||
|
|
"data": f"【Error】无法导入二维码解析器 zxingcpp 。\n[Error] Unable to import zxingcpp.\n{zxingcppErr}",
|
|||
|
|
}
|
|||
|
|
# 读入图片
|
|||
|
|
try:
|
|||
|
|
if "pil" in msn:
|
|||
|
|
img = msn["pil"]
|
|||
|
|
elif "path" in msn:
|
|||
|
|
imgPath = msn["path"]
|
|||
|
|
img = Image.open(imgPath)
|
|||
|
|
elif "base64" in msn:
|
|||
|
|
imgBase64 = msn["base64"]
|
|||
|
|
img = Image.open(BytesIO(base64.b64decode(imgBase64)))
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"code": 202,
|
|||
|
|
"data": f"【Error】图片读取失败。\n[Error] Image reading failed.\n {e}",
|
|||
|
|
}
|
|||
|
|
# 预处理
|
|||
|
|
if "argd" in msnInfo:
|
|||
|
|
try:
|
|||
|
|
img = self._preprocessing(img, msnInfo["argd"])
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"code": 203,
|
|||
|
|
"data": f"【Error】图片预处理失败。\n[Error] Image preprocessing failed.\n {e}",
|
|||
|
|
}
|
|||
|
|
# 二维码解析
|
|||
|
|
try:
|
|||
|
|
codes = zxingcpp.read_barcodes(img)
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"code": 204,
|
|||
|
|
"data": f"【Error】zxingcpp 二维码解析失败。\n[Error] zxingcpp read_barcodes failed.\n {e}",
|
|||
|
|
}
|
|||
|
|
# 结果解析
|
|||
|
|
try:
|
|||
|
|
return self._zxingcpp2dict(codes)
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"code": 205,
|
|||
|
|
"data": f"【Error】zxingcpp 结果解析失败。\n[Error] zxingcpp resule to dict failed.\n {e}",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 解析 zxingcpp 库的返回结果，转为字典。此函数允许发生异常。
|
|||
|
|
# 失败返回值： {"code":错误码, "data": "错误信息字符串"}
|
|||
|
|
# 成功返回值： {"code":100, "data": [每个码的数据] }
|
|||
|
|
# data: {"box":[包围盒], "score":1, "text": "文本"}
|
|||
|
|
def _zxingcpp2dict(self, codes):
|
|||
|
|
# 图中无码
|
|||
|
|
if not codes:
|
|||
|
|
return {
|
|||
|
|
"code": 101,
|
|||
|
|
"data": "QR code not found in the image.",
|
|||
|
|
}
|
|||
|
|
# 处理结果冲每一个二维码
|
|||
|
|
data = []
|
|||
|
|
for c in codes:
|
|||
|
|
if not c.valid: # 码无效
|
|||
|
|
continue
|
|||
|
|
d = {}
|
|||
|
|
# 方向
|
|||
|
|
d["orientation"] = c.orientation
|
|||
|
|
# 位置
|
|||
|
|
d["box"] = [
|
|||
|
|
[c.position.top_left.x, c.position.top_left.y],
|
|||
|
|
[c.position.top_right.x, c.position.top_right.y],
|
|||
|
|
[c.position.bottom_right.x, c.position.bottom_right.y],
|
|||
|
|
[c.position.bottom_left.x, c.position.bottom_left.y],
|
|||
|
|
]
|
|||
|
|
d["score"] = 1 # 置信度，兼容OCR格式，无意义
|
|||
|
|
d["end"] = "\n" # 间隔符，兼容OCR格式，无意义
|
|||
|
|
# 格式
|
|||
|
|
d["format"] = c.format.name
|
|||
|
|
# 内容为文本类型
|
|||
|
|
if c.content_type.name == "Text":
|
|||
|
|
d["text"] = c.text
|
|||
|
|
# 内容为其它格式
|
|||
|
|
# TODO: 现在是通用的处理方法，将二进制内容转为纯文本或base64
|
|||
|
|
# 或许对于某些格式的码，有更好的转文本方式
|
|||
|
|
else:
|
|||
|
|
text = f"type: {c.content_type.name}\ndata: "
|
|||
|
|
try:
|
|||
|
|
# 尝试将 bytes 转换为纯文本字符串
|
|||
|
|
t = c.bytes.decode("utf-8")
|
|||
|
|
text += t
|
|||
|
|
except UnicodeDecodeError:
|
|||
|
|
# 如果无法直接转换为纯文本，则使用 Base64 编码输出结果
|
|||
|
|
t = base64.b64encode(c.bytes)
|
|||
|
|
text += "[Base64]\n" + t.decode("utf-8")
|
|||
|
|
d["text"] = text
|
|||
|
|
data.append(d)
|
|||
|
|
if data:
|
|||
|
|
return {
|
|||
|
|
"code": 100,
|
|||
|
|
"data": data,
|
|||
|
|
}
|
|||
|
|
else:
|
|||
|
|
l = len(codes)
|
|||
|
|
return {
|
|||
|
|
"code": 102,
|
|||
|
|
"data": f"【Error】{l}组二维码解码失败。\nFailed to decode {l} sets of QR codes.",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 图像预处理
|
|||
|
|
def _preprocessing(self, img, argd):
|
|||
|
|
"""
|
|||
|
|
对图像进行预处理，包括中值滤波、锐度增强、对比度增强以及可选的灰度转换和二值化。
|
|||
|
|
|
|||
|
|
img: PIL.Image对象，待处理的图像。
|
|||
|
|
argd: 参数字典，包含以下可选键值对：
|
|||
|
|
- "preprocessing.median_filter_size": 中值滤波器的大小，应为1~9的奇数。默认不进行。
|
|||
|
|
- "preprocessing.sharpness_factor": 锐度增强因子，应为0.1~10。默认不调整锐度。
|
|||
|
|
- "preprocessing.contrast_factor": 对比度增强因子，应为0.1~10。大于1增强对比度，小于1但大于0减少对比度，1保持原样。默认不调整对比度。
|
|||
|
|
- "preprocessing.grayscale": 布尔值，指定是否将图像转换为灰度图像。True为转换，False为不转换。默认为False。
|
|||
|
|
- "preprocessing.threshold": 二值化阈值，用于灰度图像的二值化处理。应为0到255之间的整数。只有当"preprocessing.grayscale"为True时，此参数才生效。默认不进行二值化处理。
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
处理后的PIL.Image对象。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# 中值滤波
|
|||
|
|
s = round(argd.get("preprocessing.median_filter_size", -100))
|
|||
|
|
if s > 0 and s % 2 == 1:
|
|||
|
|
img = img.filter(ImageFilter.MedianFilter(size=s))
|
|||
|
|
# 锐度增强
|
|||
|
|
f = round(argd.get("preprocessing.sharpness_factor", -100))
|
|||
|
|
if f > 0:
|
|||
|
|
img = ImageEnhance.Sharpness(img).enhance(f)
|
|||
|
|
# 对比度增强
|
|||
|
|
c = argd.get("preprocessing.contrast_factor", -100)
|
|||
|
|
if c > 0:
|
|||
|
|
img = ImageEnhance.Contrast(img).enhance(c)
|
|||
|
|
# 转为灰度 & 二值化
|
|||
|
|
if argd.get("preprocessing.grayscale", False):
|
|||
|
|
img = img.convert("L")
|
|||
|
|
t = round(argd.get("preprocessing.threshold", -100))
|
|||
|
|
if t > -1:
|
|||
|
|
img = img.point(lambda p: 255 if p > t else 0)
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
|
|||
|
|
MissionQRCode = _MissionQRCodeClass()
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
zxingcpp 返回值类型
|
|||
|
|
# 字节
|
|||
|
|
bytes <class 'bytes'>: b'testtesttesttest'
|
|||
|
|
# 内容类型
|
|||
|
|
content_type <class 'zxingcpp.ContentType'>: ContentType.Text
|
|||
|
|
name <class 'str'>: Text
|
|||
|
|
value <class 'int'>: 0
|
|||
|
|
Binary <class 'zxingcpp.ContentType'>: ContentType.Binary
|
|||
|
|
GS1 <class 'zxingcpp.ContentType'>: ContentType.GS1
|
|||
|
|
ISO15434 <class 'zxingcpp.ContentType'>: ContentType.ISO15434
|
|||
|
|
Mixed <class 'zxingcpp.ContentType'>: ContentType.Mixed
|
|||
|
|
Text <class 'zxingcpp.ContentType'>: ContentType.Text
|
|||
|
|
UnknownECI <class 'zxingcpp.ContentType'>: ContentType.UnknownECI
|
|||
|
|
ec_level <class 'str'>: M
|
|||
|
|
# 二维码格式
|
|||
|
|
format <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.QRCode
|
|||
|
|
name <class 'str'>: QRCode
|
|||
|
|
value <class 'int'>: 8192
|
|||
|
|
Aztec <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.Aztec
|
|||
|
|
Codabar <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.Codabar
|
|||
|
|
Code128 <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.Code128
|
|||
|
|
Code39 <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.Code39
|
|||
|
|
Code93 <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.Code93
|
|||
|
|
DataBar <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.DataBar
|
|||
|
|
DataBarExpanded <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.DataBarExpanded
|
|||
|
|
DataMatrix <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.DataMatrix
|
|||
|
|
EAN13 <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.EAN13
|
|||
|
|
EAN8 <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.EAN8
|
|||
|
|
ITF <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.ITF
|
|||
|
|
LinearCodes <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.LinearCodes
|
|||
|
|
MatrixCodes <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.MatrixCodes
|
|||
|
|
MaxiCode <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.MaxiCode
|
|||
|
|
MicroQRCode <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.MicroQRCode
|
|||
|
|
NONE <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.NONE
|
|||
|
|
PDF417 <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.PDF417
|
|||
|
|
QRCode <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.QRCode
|
|||
|
|
UPCA <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.UPCA
|
|||
|
|
UPCE <class 'zxingcpp.BarcodeFormat'>: BarcodeFormat.UPCE
|
|||
|
|
# 方向
|
|||
|
|
orientation <class 'int'>: 0
|
|||
|
|
# 位置
|
|||
|
|
position <class 'zxingcpp.Position'>: 16x16 116x16 116x116 16x116
|
|||
|
|
bottom_left <class 'zxingcpp.Point'>: <zxingcpp.Point object at 0x00000222AEE5C370>
|
|||
|
|
bottom_right <class 'zxingcpp.Point'>: <zxingcpp.Point object at 0x00000222C19D07B0>
|
|||
|
|
top_left <class 'zxingcpp.Point'>: <zxingcpp.Point object at 0x00000222AEE5C370>
|
|||
|
|
top_right <class 'zxingcpp.Point'>: <zxingcpp.Point object at 0x00000222C19D0770>
|
|||
|
|
symbology_identifier <class 'str'>: ]Q1
|
|||
|
|
text <class 'str'>: testtesttesttest
|
|||
|
|
valid <class 'bool'>: True
|
|||
|
|
"""
|