352 lines
15 KiB
Python
352 lines
15 KiB
Python
|
"""QR code rendered with alphanumeric characters."""
|
|||
|
|
import qrcode
|
|||
|
|
from PIL import Image, ImageDraw, ImageFont
|
|||
|
|
import svgwrite
|
|||
|
|
from typing import List
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
DEFAULT_CHARSET = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|||
|
|
|
|||
|
|
def is_finder_pattern(x: int, y: int, matrix_size: int) -> bool:
|
|||
|
|
"""判断给定坐标是否在QR码的三个定位点区域内（包含分隔符的8x8模块）"""
|
|||
|
|
# 边框已被裁剪，定位点从(0,0)开始
|
|||
|
|
|
|||
|
|
# 左上角定位点区域 (0,0) 到 (7,7) - 包含7x7定位点和1像素分隔符
|
|||
|
|
if 0 <= x <= 7 and 0 <= y <= 7:
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 右上角定位点区域 (matrix_size-8,0) 到 (matrix_size-1,7)
|
|||
|
|
if (matrix_size - 8) <= x <= (matrix_size - 1) and 0 <= y <= 7:
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 左下角定位点区域 (0,matrix_size-8) 到 (7,matrix_size-1)
|
|||
|
|
if 0 <= x <= 7 and (matrix_size - 8) <= y <= (matrix_size - 1):
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def generate_matrix(data: str, ec_level: str = "H") -> List[List[bool]]:
|
|||
|
|
"""Return boolean matrix of the QR code."""
|
|||
|
|
level_map = {
|
|||
|
|
"L": qrcode.constants.ERROR_CORRECT_L,
|
|||
|
|
"M": qrcode.constants.ERROR_CORRECT_M,
|
|||
|
|
"Q": qrcode.constants.ERROR_CORRECT_Q,
|
|||
|
|
"H": qrcode.constants.ERROR_CORRECT_H,
|
|||
|
|
}
|
|||
|
|
qr = qrcode.QRCode(error_correction=level_map.get(ec_level.upper(), qrcode.constants.ERROR_CORRECT_H))
|
|||
|
|
qr.add_data(data)
|
|||
|
|
qr.make()
|
|||
|
|
return qr.get_matrix()
|
|||
|
|
|
|||
|
|
def render_to_png(
|
|||
|
|
matrix,
|
|||
|
|
out_path: str,
|
|||
|
|
cell_size: int = 20,
|
|||
|
|
charset: str = DEFAULT_CHARSET,
|
|||
|
|
font_path: str | None = None,
|
|||
|
|
logo_path: str | None = None,
|
|||
|
|
render_mode: str = "combined", # 新增参数：'solid', 'text', 'combined'
|
|||
|
|
):
|
|||
|
|
# 去掉QR码的4像素边框
|
|||
|
|
border = 4
|
|||
|
|
original_size = len(matrix)
|
|||
|
|
|
|||
|
|
# 裁剪矩阵，去掉边框
|
|||
|
|
cropped_matrix = []
|
|||
|
|
for y in range(border, original_size - border):
|
|||
|
|
row = []
|
|||
|
|
for x in range(border, original_size - border):
|
|||
|
|
row.append(matrix[y][x])
|
|||
|
|
cropped_matrix.append(row)
|
|||
|
|
|
|||
|
|
matrix = cropped_matrix
|
|||
|
|
N = len(matrix)
|
|||
|
|
img_size = N * cell_size
|
|||
|
|
img = Image.new("RGBA", (img_size, img_size), (255, 255, 255, 0))
|
|||
|
|
draw = ImageDraw.Draw(img)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 设置字体，优先使用指定字体或项目中的字体
|
|||
|
|
font_size = int(cell_size * 0.7) # 基础字体大小
|
|||
|
|
font_size_large = int(cell_size * 0.9) # 第一个和最后一个字符的更大字体
|
|||
|
|
|
|||
|
|
if font_path:
|
|||
|
|
try:
|
|||
|
|
font = ImageFont.truetype(font_path, font_size)
|
|||
|
|
font_large = ImageFont.truetype(font_path, font_size_large)
|
|||
|
|
except:
|
|||
|
|
font = ImageFont.load_default()
|
|||
|
|
font_large = font
|
|||
|
|
else:
|
|||
|
|
# 尝试使用项目目录中的字体
|
|||
|
|
try:
|
|||
|
|
# 检查当前目录和上级目录中的字体文件
|
|||
|
|
possible_fonts = [
|
|||
|
|
"SIMHEI.TTF",
|
|||
|
|
"../SIMHEI.TTF",
|
|||
|
|
"../../SIMHEI.TTF",
|
|||
|
|
os.path.join(os.path.dirname(__file__), "..", "SIMHEI.TTF")
|
|||
|
|
]
|
|||
|
|
font_loaded = False
|
|||
|
|
for font_file in possible_fonts:
|
|||
|
|
if os.path.exists(font_file):
|
|||
|
|
font = ImageFont.truetype(font_file, font_size)
|
|||
|
|
font_large = ImageFont.truetype(font_file, font_size_large)
|
|||
|
|
font_loaded = True
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
if not font_loaded:
|
|||
|
|
# 尝试系统字体
|
|||
|
|
try:
|
|||
|
|
font = ImageFont.truetype("arial.ttf", font_size)
|
|||
|
|
font_large = ImageFont.truetype("arial.ttf", font_size_large)
|
|||
|
|
except:
|
|||
|
|
font = ImageFont.load_default()
|
|||
|
|
font_large = font
|
|||
|
|
except:
|
|||
|
|
font = ImageFont.load_default()
|
|||
|
|
font_large = font
|
|||
|
|
|
|||
|
|
# 绘制二维码
|
|||
|
|
# 计算需要绘制的总位置数，确保至少包含一次完整字符串
|
|||
|
|
total_positions = 0
|
|||
|
|
for y, row in enumerate(matrix):
|
|||
|
|
for x, bit in enumerate(row):
|
|||
|
|
if bit:
|
|||
|
|
total_positions += 1
|
|||
|
|
|
|||
|
|
# 如果字符串长度大于总位置数，调整字符串以确保覆盖
|
|||
|
|
effective_charset = charset
|
|||
|
|
if len(charset) > total_positions and total_positions > 0:
|
|||
|
|
# 如果字符串太长，截取到能完整显示的长度
|
|||
|
|
effective_charset = charset[:total_positions]
|
|||
|
|
|
|||
|
|
# 开始绘制，使用优化后的字符分配
|
|||
|
|
position_counter = 0
|
|||
|
|
for y, row in enumerate(matrix):
|
|||
|
|
for x, bit in enumerate(row):
|
|||
|
|
|
|||
|
|
if bit:
|
|||
|
|
x_pos = x * cell_size
|
|||
|
|
y_pos = y * cell_size
|
|||
|
|
|
|||
|
|
# 使用位置计数器来确保字符串的完整性
|
|||
|
|
ch = effective_charset[position_counter % len(effective_charset)]
|
|||
|
|
char_index = position_counter % len(effective_charset)
|
|||
|
|
position_counter += 1
|
|||
|
|
|
|||
|
|
# 根据渲染模式绘制不同内容
|
|||
|
|
if render_mode == "solid":
|
|||
|
|
# 只绘制黑色方块
|
|||
|
|
draw.rectangle(
|
|||
|
|
[x_pos, y_pos, x_pos + cell_size - 1, y_pos + cell_size - 1],
|
|||
|
|
fill="black"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
elif render_mode == "text":
|
|||
|
|
# 检查是否为定位点区域
|
|||
|
|
if is_finder_pattern(x, y, N):
|
|||
|
|
# 定位点区域绘制黑色方块
|
|||
|
|
draw.rectangle(
|
|||
|
|
[x_pos, y_pos, x_pos + cell_size - 1, y_pos + cell_size - 1],
|
|||
|
|
fill="black"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# 非定位点区域绘制文字（黑色文字在白色背景上）
|
|||
|
|
# 判断字符类型
|
|||
|
|
is_chinese = '\u4e00' <= ch <= '\u9fff'
|
|||
|
|
is_uppercase = ch.isupper()
|
|||
|
|
is_lowercase = ch.islower()
|
|||
|
|
is_digit = ch.isdigit()
|
|||
|
|
|
|||
|
|
# 判断是否是第一个或最后一个字符，选择对应字体
|
|||
|
|
# 只有第一次循环的首尾字符才放大
|
|||
|
|
is_first_cycle = position_counter <= len(effective_charset)
|
|||
|
|
is_first_or_last_in_cycle = char_index == 0 or char_index == len(effective_charset) - 1
|
|||
|
|
is_first_or_last = is_first_cycle and is_first_or_last_in_cycle
|
|||
|
|
current_font = font_large if is_first_or_last else font
|
|||
|
|
|
|||
|
|
# 使用方块的中心点作为文字的中心
|
|||
|
|
center_x = x_pos + cell_size // 2
|
|||
|
|
center_y = y_pos + cell_size // 2
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 根据字符类型调整垂直偏移
|
|||
|
|
if is_chinese:
|
|||
|
|
# 中文字符直接居中
|
|||
|
|
draw.text(
|
|||
|
|
(center_x, center_y),
|
|||
|
|
ch,
|
|||
|
|
font=current_font,
|
|||
|
|
fill="black",
|
|||
|
|
anchor="mm"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# 英文和数字需要特殊处理
|
|||
|
|
# 获取文字边界框以进行精确调整
|
|||
|
|
bbox = draw.textbbox((0, 0), ch, font=current_font)
|
|||
|
|
text_width = bbox[2] - bbox[0]
|
|||
|
|
text_height = bbox[3] - bbox[1]
|
|||
|
|
|
|||
|
|
# 计算基础位置
|
|||
|
|
text_x = center_x - text_width // 2 - bbox[0]
|
|||
|
|
text_y = center_y - text_height // 2 - bbox[1]
|
|||
|
|
|
|||
|
|
# 根据字符类型进行垂直调整
|
|||
|
|
if is_uppercase or is_digit:
|
|||
|
|
# 大写字母和数字向上偏移
|
|||
|
|
text_y -= int(cell_size * 0.05)
|
|||
|
|
elif is_lowercase:
|
|||
|
|
# 小写字母可能需要不同的调整
|
|||
|
|
text_y -= int(cell_size * 0.03)
|
|||
|
|
|
|||
|
|
draw.text((text_x, text_y), ch, font=current_font, fill="black")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
# 如果出现任何错误，使用简单的中心定位
|
|||
|
|
bbox = draw.textbbox((0, 0), ch, font=current_font)
|
|||
|
|
text_width = bbox[2] - bbox[0]
|
|||
|
|
text_height = bbox[3] - bbox[1]
|
|||
|
|
text_x = center_x - text_width // 2
|
|||
|
|
text_y = center_y - text_height // 2
|
|||
|
|
draw.text((text_x, text_y), ch, font=current_font, fill="black")
|
|||
|
|
|
|||
|
|
elif render_mode == "combined":
|
|||
|
|
# 先绘制黑色方块
|
|||
|
|
draw.rectangle(
|
|||
|
|
[x_pos, y_pos, x_pos + cell_size - 1, y_pos + cell_size - 1],
|
|||
|
|
fill="black"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 在黑色方块上绘制灰色文字 RGB(102, 102, 102)
|
|||
|
|
# 判断字符类型
|
|||
|
|
is_chinese = '\u4e00' <= ch <= '\u9fff'
|
|||
|
|
is_uppercase = ch.isupper()
|
|||
|
|
is_lowercase = ch.islower()
|
|||
|
|
is_digit = ch.isdigit()
|
|||
|
|
|
|||
|
|
# 使用方块的中心点作为文字的中心
|
|||
|
|
center_x = x_pos + cell_size // 2
|
|||
|
|
center_y = y_pos + cell_size // 2
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 根据字符类型调整垂直偏移
|
|||
|
|
if is_chinese:
|
|||
|
|
# 中文字符直接居中
|
|||
|
|
draw.text(
|
|||
|
|
(center_x, center_y),
|
|||
|
|
ch,
|
|||
|
|
font=font,
|
|||
|
|
fill="black",
|
|||
|
|
anchor="mm"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# 英文和数字需要特殊处理
|
|||
|
|
# 获取文字边界框以进行精确调整
|
|||
|
|
bbox = draw.textbbox((0, 0), ch, font=font)
|
|||
|
|
text_width = bbox[2] - bbox[0]
|
|||
|
|
text_height = bbox[3] - bbox[1]
|
|||
|
|
|
|||
|
|
# 计算基础位置
|
|||
|
|
text_x = center_x - text_width // 2 - bbox[0]
|
|||
|
|
text_y = center_y - text_height // 2 - bbox[1]
|
|||
|
|
|
|||
|
|
# 根据字符类型进行垂直调整
|
|||
|
|
if is_uppercase or is_digit:
|
|||
|
|
# 大写字母和数字向上偏移
|
|||
|
|
text_y -= int(cell_size * 0.05)
|
|||
|
|
elif is_lowercase:
|
|||
|
|
# 小写字母可能需要不同的调整
|
|||
|
|
text_y -= int(cell_size * 0.03)
|
|||
|
|
|
|||
|
|
draw.text((text_x, text_y), ch, font=font, fill="black")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
# 如果出现任何错误，使用简单的中心定位
|
|||
|
|
bbox = draw.textbbox((0, 0), ch, font=font)
|
|||
|
|
text_width = bbox[2] - bbox[0]
|
|||
|
|
text_height = bbox[3] - bbox[1]
|
|||
|
|
text_x = center_x - text_width // 2
|
|||
|
|
text_y = center_y - text_height // 2
|
|||
|
|
draw.text((text_x, text_y), ch, font=font, fill="black")
|
|||
|
|
|
|||
|
|
img.save(out_path)
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
def render_to_svg(
|
|||
|
|
matrix,
|
|||
|
|
out_path: str,
|
|||
|
|
cell_size: int = 20,
|
|||
|
|
charset: str = DEFAULT_CHARSET,
|
|||
|
|
font_family: str = "Courier New",
|
|||
|
|
):
|
|||
|
|
# 去掉QR码的4像素边框
|
|||
|
|
border = 4
|
|||
|
|
original_size = len(matrix)
|
|||
|
|
|
|||
|
|
# 裁剪矩阵，去掉边框
|
|||
|
|
cropped_matrix = []
|
|||
|
|
for y in range(border, original_size - border):
|
|||
|
|
row = []
|
|||
|
|
for x in range(border, original_size - border):
|
|||
|
|
row.append(matrix[y][x])
|
|||
|
|
cropped_matrix.append(row)
|
|||
|
|
|
|||
|
|
matrix = cropped_matrix
|
|||
|
|
N = len(matrix)
|
|||
|
|
size = N * cell_size
|
|||
|
|
dwg = svgwrite.Drawing(out_path, size=(size, size), profile="tiny")
|
|||
|
|
|
|||
|
|
# 计算字符分配
|
|||
|
|
total_positions = sum(sum(row) for row in matrix)
|
|||
|
|
effective_charset = charset
|
|||
|
|
if len(charset) > total_positions and total_positions > 0:
|
|||
|
|
effective_charset = charset[:total_positions]
|
|||
|
|
|
|||
|
|
position_counter = 0
|
|||
|
|
for y, row in enumerate(matrix):
|
|||
|
|
for x, bit in enumerate(row):
|
|||
|
|
if bit:
|
|||
|
|
x_pos = x * cell_size
|
|||
|
|
y_pos = y * cell_size
|
|||
|
|
|
|||
|
|
# 检查是否为定位点区域
|
|||
|
|
if is_finder_pattern(x, y, N):
|
|||
|
|
# 定位点区域只绘制黑色方块
|
|||
|
|
dwg.add(
|
|||
|
|
dwg.rect(
|
|||
|
|
insert=(x_pos, y_pos),
|
|||
|
|
size=(cell_size, cell_size),
|
|||
|
|
fill="black"
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# 非定位点区域绘制字符
|
|||
|
|
ch = effective_charset[position_counter % len(effective_charset)]
|
|||
|
|
position_counter += 1
|
|||
|
|
|
|||
|
|
# 调整 y 位置以实现垂直居中效果
|
|||
|
|
text_y = y_pos + cell_size * 0.65
|
|||
|
|
dwg.add(
|
|||
|
|
dwg.text(
|
|||
|
|
ch,
|
|||
|
|
insert=(x_pos + cell_size * 0.5, text_y),
|
|||
|
|
font_size=cell_size * 0.6,
|
|||
|
|
font_family=font_family,
|
|||
|
|
fill="black",
|
|||
|
|
text_anchor="middle"
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
dwg.save()
|
|||
|
|
return out_path
|
|||
|
|
|
|||
|
|
# Convenience wrappers
|
|||
|
|
def generate_qr_image(data: str, **kwargs):
|
|||
|
|
matrix = generate_matrix(data, kwargs.pop("ec_level", "H"))
|
|||
|
|
return render_to_png(matrix, kwargs.get("out_path", "qr.png"), **kwargs)
|
|||
|
|
|
|||
|
|
def generate_qr_svg(data: str, **kwargs):
|
|||
|
|
matrix = generate_matrix(data, kwargs.pop("ec_level", "H"))
|
|||
|
|
return render_to_svg(matrix, kwargs.get("out_path", "qr.svg"), **kwargs)
|