二维码

arknights_mower/utils/qrcode.py

@@ -0,0 +1,50 @@
+import json
+from typing import Dict, List
+from zlib import compress, decompress
+
+from base45 import b45decode, b45encode
+from PIL import Image, ImageChops
+from pyzbar import pyzbar
+from qrcode import make
+
+QRCODE_SIZE = 680
+GAP_SIZE = 44
+
+
+def encode(data: str, n: int = 4) -> List[Image.Image]:
+    data = b45encode(compress(data.encode("utf-8"), level=9))
+    length = len(data)
+    split: List[bytes] = []
+    for i in range(n):
+        start = length // n * i
+        end = length if i == n - 1 else length // n * (i + 1)
+        split.append(data[start:end])
+    return [trim(make(i).get_image()) for i in split]
+
+
+def trim(img: Image.Image) -> Image.Image:
+    bg = Image.new(img.mode, img.size, img.getpixel((0, 0)))
+    diff = ImageChops.difference(img, bg)
+    img = img.crop(diff.getbbox())
+    img = img.resize((QRCODE_SIZE, QRCODE_SIZE))
+    return img
+
+
+def export(plan: Dict, upper: Image.Image) -> Image.Image:
+    img = Image.new(
+        upper.mode,
+        (upper.width, upper.height + GAP_SIZE + QRCODE_SIZE),
+        (255, 255, 255),
+    )
+    img.paste(upper)
+    qrcode_list = encode(json.dumps(plan))
+    for idx, i in enumerate(qrcode_list):
+        img.paste(i, (GAP_SIZE + idx * (GAP_SIZE + QRCODE_SIZE), upper.height))
+    return img
+
+
+def decode(img: Image.Image) -> Dict:
+    data = pyzbar.decode(img)
+    data.sort(key=lambda i: i.rect.left)
+    data = b45decode(b"".join([i.data for i in data]))
+    return json.loads(decompress(data).decode("utf-8"))

requirements.txt

@@ -28,3 +28,6 @@ rapidocr-onnxruntime==1.3.7
 beautifulsoup4==4.12.2
 lxml==4.9.3
 evalidate==2.0.2
+qrcode==7.4.2
+pyzbar==0.1.9
+base45==0.4.4