57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
|
""" QR Code related utilities. """
|
||
|
|
|
||
|
|
from os import remove
|
||
|
|
from pyqrcode import create
|
||
|
|
from pyzbar.pyzbar import decode
|
||
|
|
from PIL import Image
|
||
|
|
from pagermaid import log
|
||
|
|
from pagermaid.listener import listener
|
||
|
|
from pagermaid.utils import obtain_message, upload_attachment
|
||
|
|
|
||
|
|
|
||
|
|
@listener(outgoing=True, command="genqr",
|
||
|
|
description="生成包含文字内容的 QR Code 。",
|
||
|
|
parameters="<string>")
|
||
|
|
async def genqr(context):
|
||
|
|
""" Generate QR codes. """
|
||
|
|
reply_id = context.reply_to_msg_id
|
||
|
|
try:
|
||
|
|
message = await obtain_message(context)
|
||
|
|
except ValueError:
|
||
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
||
|
|
return
|
||
|
|
await context.edit("生成QR码。")
|
||
|
|
try:
|
||
|
|
create(message, error='L', mode='binary').png('qr.webp', scale=6)
|
||
|
|
except UnicodeEncodeError:
|
||
|
|
await context.edit("出错了呜呜呜 ~ 目标消息中的字符无效。")
|
||
|
|
return
|
||
|
|
await upload_attachment("qr.webp", context.chat_id, reply_id)
|
||
|
|
remove("qr.webp")
|
||
|
|
await context.delete()
|
||
|
|
await log(f"为 `{message}` 生成了一张 QR 码。")
|
||
|
|
|
||
|
|
|
||
|
|
@listener(outgoing=True, command="parseqr",
|
||
|
|
description="将回复的消息文件解析为 QR码 并输出结果。")
|
||
|
|
async def parseqr(context):
|
||
|
|
""" Parse attachment of replied message as a QR Code and output results. """
|
||
|
|
success = False
|
||
|
|
target_file_path = await context.client.download_media(
|
||
|
|
await context.get_reply_message()
|
||
|
|
)
|
||
|
|
if not target_file_path:
|
||
|
|
await context.edit("出错了呜呜呜 ~ 回复的消息中没有附件。")
|
||
|
|
return
|
||
|
|
try:
|
||
|
|
message = str(decode(Image.open(target_file_path))[0].data)[2:][:-1]
|
||
|
|
success = True
|
||
|
|
await context.edit(f"**内容: **\n"
|
||
|
|
f"`{message}`")
|
||
|
|
except IndexError:
|
||
|
|
await context.edit("出错了呜呜呜 ~ 回复的附件不是 QR 码。")
|
||
|
|
message = None
|
||
|
|
if success:
|
||
|
|
await log(f"已解析一张带有 QR 码的消息，内容： `{message}`.")
|
||
|
|
remove(target_file_path)
|