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
|
||
"""
|