140 lines
4.5 KiB
Python
140 lines
4.5 KiB
Python
|
import os
|
|||
|
|
import tempfile
|
|||
|
|
|
|||
|
|
import cv2
|
|||
|
|
from sqlalchemy import create_engine, Table, Column, Integer, String, MetaData, DateTime,TIMESTAMP
|
|||
|
|
from fpdf import FPDF
|
|||
|
|
import io
|
|||
|
|
import hashlib
|
|||
|
|
import requests
|
|||
|
|
import secrets
|
|||
|
|
from PIL import Image
|
|||
|
|
import matplotlib.pyplot as plt
|
|||
|
|
import matplotlib.image as mpimg
|
|||
|
|
|
|||
|
|
import demo
|
|||
|
|
|
|||
|
|
|
|||
|
|
# from demo import perspective_transform
|
|||
|
|
|
|||
|
|
def init_sqlserver():
|
|||
|
|
# 创建连接引擎
|
|||
|
|
engine = create_engine('postgresql://postgres:!1q2w3E*@8.134.85.73:5432/postgres')
|
|||
|
|
# 连接数据库
|
|||
|
|
connection = engine.connect()
|
|||
|
|
print("Connection to PostgreSQL DB successful")
|
|||
|
|
# 声明表结构
|
|||
|
|
metadata = MetaData()
|
|||
|
|
qrcode_data = Table('qrcode_data', metadata,
|
|||
|
|
Column('key', String),
|
|||
|
|
Column('circles', String),
|
|||
|
|
# createdate 时间类型
|
|||
|
|
Column('createdate', TIMESTAMP)
|
|||
|
|
)
|
|||
|
|
return qrcode_data
|
|||
|
|
|
|||
|
|
def get_images(key, secret, api, count=1):
|
|||
|
|
images = []
|
|||
|
|
urls = []
|
|||
|
|
for i in range(count):
|
|||
|
|
random_number = ''.join([str(secrets.randbelow(9) + 1) for _ in range(16)])
|
|||
|
|
parm = f"api_key={key}&cliD={random_number}&cliT=D1&return_file={secret}"
|
|||
|
|
params = {
|
|||
|
|
"api_key": key,
|
|||
|
|
"cliT": "D2",
|
|||
|
|
"cliD": random_number,
|
|||
|
|
"return_file": ""
|
|||
|
|
}
|
|||
|
|
# 排序
|
|||
|
|
sorted_params = sorted(params.items())
|
|||
|
|
concatenated_params = '&'.join(f"{key}={value}" for key, value in sorted_params)
|
|||
|
|
data = concatenated_params + secret
|
|||
|
|
hash_object = hashlib.md5(data.encode())
|
|||
|
|
md5_hash = hash_object.hexdigest()
|
|||
|
|
url = f"{api}?api_key={key}&cliT=D2&cliD={random_number}&sign={md5_hash}&return_file="
|
|||
|
|
urls.append(url)
|
|||
|
|
for url in urls:
|
|||
|
|
response = requests.get(url)
|
|||
|
|
images.append(response.content)
|
|||
|
|
|
|||
|
|
return images
|
|||
|
|
|
|||
|
|
def ndarray_to_binary(image_array):
|
|||
|
|
# 将numpy数组转换为PIL图像对象
|
|||
|
|
img = Image.fromarray(image_array)
|
|||
|
|
# 创建一个字节流对象
|
|||
|
|
byte_arr = io.BytesIO()
|
|||
|
|
# 将图像保存到字节流中，格式可以自选，如PNG
|
|||
|
|
img.save(byte_arr, format='PNG')
|
|||
|
|
# 获取二进制数据
|
|||
|
|
binary_data = byte_arr.getvalue()
|
|||
|
|
return binary_data
|
|||
|
|
|
|||
|
|
def create_pdf(images, fileName):
|
|||
|
|
# 创建 PDF 类实例
|
|||
|
|
pdf = FPDF('P', 'mm', 'A4')
|
|||
|
|
# 添加一页
|
|||
|
|
pdf.add_page()
|
|||
|
|
# 设置图片的尺寸
|
|||
|
|
image_width = 15 # 图片宽度 30 mm
|
|||
|
|
image_height = 15 # 图片高度 30 mm
|
|||
|
|
margin = 20 # 页边距 10 mm
|
|||
|
|
images_per_row = 5
|
|||
|
|
# 当前行的开始位置
|
|||
|
|
x_offset = margin
|
|||
|
|
y_offset = margin
|
|||
|
|
# 自定义临时文件路径
|
|||
|
|
custom_temp_dir = './temp'
|
|||
|
|
os.makedirs(custom_temp_dir, exist_ok=True)
|
|||
|
|
# 循环处理图片二进制数据
|
|||
|
|
for i, binary_image in enumerate(images):
|
|||
|
|
if i % images_per_row == 0 and i != 0:
|
|||
|
|
# 开始新的一行
|
|||
|
|
x_offset = margin
|
|||
|
|
y_offset += image_height + margin
|
|||
|
|
# # 将binary_image保存为本地文件
|
|||
|
|
# with open(f"{i}.png", "wb") as f:
|
|||
|
|
# f.write(binary_image)
|
|||
|
|
|
|||
|
|
# 创建一个临时文件来保存图像
|
|||
|
|
with tempfile.NamedTemporaryFile(delete=False, suffix=".TIFF", dir=custom_temp_dir) as tmpfile:
|
|||
|
|
tmpfile.write(binary_image)
|
|||
|
|
tmp_filename = tmpfile.name
|
|||
|
|
print(tmp_filename)
|
|||
|
|
if os.access(tmp_filename, os.R_OK):
|
|||
|
|
print("File is readable.")
|
|||
|
|
else:
|
|||
|
|
print("File is not readable.")
|
|||
|
|
|
|||
|
|
img_path = demo.exec_image_result(tmp_filename)
|
|||
|
|
# 将img_path返回来 tiff 格式转换 png 格式
|
|||
|
|
img = Image.open(img_path)
|
|||
|
|
png_path = img_path.replace('.tiff', '.png')
|
|||
|
|
img.save(png_path, format='PNG')
|
|||
|
|
# 添加图片
|
|||
|
|
pdf.image(png_path, x=x_offset, y=y_offset, w=image_width, h=image_height)
|
|||
|
|
# 更新x坐标
|
|||
|
|
x_offset += image_width + margin
|
|||
|
|
# 删除文件
|
|||
|
|
os.remove(tmp_filename)
|
|||
|
|
# 输出 PDF
|
|||
|
|
pdf.output(fileName)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
key = "CL8ca5c629b676d724"
|
|||
|
|
secret = "7a92463b491be3d0aa9704a8980e46ad"
|
|||
|
|
api = "https://open-api.cli.im/cli-open-platform-service/v1/labelStyle/createWithKey"
|
|||
|
|
|
|||
|
|
# 初始化数据库
|
|||
|
|
# qrcode_entity = init_sqlserver()
|
|||
|
|
images = get_images(key, secret, api, count=1)
|
|||
|
|
create_pdf(images, "qr_v2.pdf")
|
|||
|
|
|
|||
|
|
# for i in range(10):
|
|||
|
|
# # 生成16位随机数字
|
|||
|
|
# random_number = ''.join([str(secrets.randbelow(10)) for _ in range(16)])
|
|||
|
|
# # 将随机数字每两位分为一组
|
|||
|
|
# # groups = [random_number[i:i + 2] for i in range(0, len(random_number), 2)]
|
|||
|
|
# # print(groups)
|