35 lines
920 B
Python
35 lines
920 B
Python
|
"""二维码生成服务 — 参考 MOM 系统 label_service.py 的 QR 生成逻辑"""
|
|||
|
|
import io
|
|||
|
|
import qrcode
|
|||
|
|
from qrcode.image.pil import PilImage
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_qrcode_png(content: str, size_px: int = 300) -> io.BytesIO:
|
|||
|
|
"""
|
|||
|
|
生成二维码 PNG 图片，返回 BytesIO 流。
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
content: 二维码内容（如 16 位序列号）
|
|||
|
|
size_px: 输出图片尺寸（像素），默认 300×300
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
io.BytesIO: PNG 格式的图片字节流
|
|||
|
|
"""
|
|||
|
|
qr = qrcode.QRCode(
|
|||
|
|
version=1,
|
|||
|
|
error_correction=qrcode.constants.ERROR_CORRECT_M,
|
|||
|
|
box_size=10,
|
|||
|
|
border=2,
|
|||
|
|
)
|
|||
|
|
qr.add_data(content)
|
|||
|
|
qr.make(fit=True)
|
|||
|
|
|
|||
|
|
img: PilImage = qr.make_image(fill_color="black", back_color="white")
|
|||
|
|
img = img.convert("RGB")
|
|||
|
|
img = img.resize((size_px, size_px))
|
|||
|
|
|
|||
|
|
buf = io.BytesIO()
|
|||
|
|
img.save(buf, format="PNG")
|
|||
|
|
buf.seek(0)
|
|||
|
|
return buf
|