mirror of
https://github.com/9001/copyparty.git
synced 2025-10-31 04:32:20 -06:00
08ebb0b4c9
move copyparty-original code to qrkode.py stolen/qrcodegen.py can be deleted and replaced with system lib this is safe and has minimal affect on functionality; performance will be a tiny bit slower without the vendored copy
113 lines
2.9 KiB
Python
113 lines
2.9 KiB
Python
# coding: utf-8
|
|
from __future__ import print_function, unicode_literals
|
|
|
|
import os
|
|
|
|
try:
|
|
if os.getenv("PRTY_SYS_ALL") or os.getenv("PRTY_SYS_QRCG"):
|
|
raise ImportError()
|
|
from .stolen.qrcodegen import QrCode
|
|
|
|
qrgen = QrCode.encode_binary
|
|
VENDORED = True
|
|
except ImportError:
|
|
VENDORED = False
|
|
from qrcodegen import QrCode
|
|
|
|
if os.getenv("PRTY_MODSPEC"):
|
|
from inspect import getsourcefile
|
|
|
|
print("PRTY_MODSPEC: qrcode:", getsourcefile(QrCode))
|
|
|
|
if True: # pylint: disable=using-constant-test
|
|
import typing
|
|
from typing import Any, Optional, Sequence, Union
|
|
|
|
|
|
if not VENDORED:
|
|
|
|
def _qrgen(data: Union[bytes, Sequence[int]]) -> "QrCode":
|
|
ret = None
|
|
V = QrCode.Ecc
|
|
for e in [V.HIGH, V.QUARTILE, V.MEDIUM, V.LOW]:
|
|
qr = QrCode.encode_binary(data, e)
|
|
qr.size = qr._size
|
|
qr.modules = qr._modules
|
|
if not ret or ret.size > qr.size:
|
|
ret = qr
|
|
return ret
|
|
|
|
qrgen = _qrgen
|
|
|
|
|
|
def qr2txt(qr: QrCode, zoom: int = 1, pad: int = 4) -> str:
|
|
tab = qr.modules
|
|
sz = qr.size
|
|
if sz % 2 and zoom == 1:
|
|
tab.append([False] * sz)
|
|
|
|
tab = [[False] * sz] * pad + tab + [[False] * sz] * pad
|
|
tab = [[False] * pad + x + [False] * pad for x in tab]
|
|
|
|
rows: list[str] = []
|
|
if zoom == 1:
|
|
for y in range(0, len(tab), 2):
|
|
row = ""
|
|
for x in range(len(tab[y])):
|
|
v = 2 if tab[y][x] else 0
|
|
v += 1 if tab[y + 1][x] else 0
|
|
row += " ▄▀█"[v]
|
|
rows.append(row)
|
|
else:
|
|
for tr in tab:
|
|
row = ""
|
|
for zb in tr:
|
|
row += " █"[int(zb)] * 2
|
|
rows.append(row)
|
|
|
|
return "\n".join(rows)
|
|
|
|
|
|
def qr2png(
|
|
qr: QrCode,
|
|
zoom: int,
|
|
pad: int,
|
|
bg: Optional[tuple[int, int, int]],
|
|
fg: Optional[tuple[int, int, int]],
|
|
ap: str,
|
|
) -> None:
|
|
from PIL import Image
|
|
|
|
tab = qr.modules
|
|
sz = qr.size
|
|
psz = sz + pad * 2
|
|
if bg:
|
|
img = Image.new("RGB", (psz, psz), bg)
|
|
else:
|
|
img = Image.new("RGBA", (psz, psz), (0, 0, 0, 0))
|
|
fg = (fg[0], fg[1], fg[2], 255)
|
|
for y in range(sz):
|
|
for x in range(sz):
|
|
if tab[y][x]:
|
|
img.putpixel((x + pad, y + pad), fg)
|
|
if zoom != 1:
|
|
img = img.resize((sz * zoom, sz * zoom), Image.Resampling.NEAREST)
|
|
img.save(ap)
|
|
|
|
|
|
def qr2svg(qr: QrCode, border: int) -> str:
|
|
parts: list[str] = []
|
|
for y in range(qr.size):
|
|
sy = border + y
|
|
for x in range(qr.size):
|
|
if qr.modules[y][x]:
|
|
parts.append("M%d,%dh1v1h-1z" % (border + x, sy))
|
|
t = """\
|
|
<?xml version="1.0" encoding="UTF-8"?>
|
|
<svg xmlns="http://www.w3.org/2000/svg" version="1.1" viewBox="0 0 {0} {0}" stroke="none">
|
|
<rect width="100%" height="100%" fill="#F7F7F7"/>
|
|
<path d="{1}" fill="#111111"/>
|
|
</svg>
|
|
"""
|
|
return t.format(qr.size + border * 2, " ".join(parts))
|