Ivan Schaller
3249de4609
All checks were successful
ci/woodpecker/push/tests Pipeline was successful
Signed-off-by: Ivan Schaller <ivan@schaller.sh>
32 lines
968 B
Python
32 lines
968 B
Python
import base64
|
|
from io import BytesIO
|
|
from typing import Union
|
|
|
|
import qrcode
|
|
from qrcode.image.pure import PyPNGImage
|
|
from qrcode.image.svg import SvgPathFillImage
|
|
|
|
|
|
def generate_qrcode(url: str, **kwargs) -> tuple[PyPNGImage, SvgPathFillImage]:
|
|
qr = qrcode.QRCode(
|
|
version=kwargs["qr_version"],
|
|
error_correction=kwargs["qr_error_correction"],
|
|
box_size=kwargs["qr_box_size"],
|
|
border=kwargs["qr_border_size"],
|
|
image_factory=PyPNGImage,
|
|
)
|
|
qr.add_data(url)
|
|
qr.make(fit=True)
|
|
|
|
img_png = qr.make_image(image_factory=PyPNGImage, fill_color="black", back_color="white")
|
|
img_svg = qr.make_image(image_factory=SvgPathFillImage, fill_color="black", back_color="white")
|
|
|
|
return img_png, img_svg
|
|
|
|
|
|
def get_base64(img: Union[SvgPathFillImage, PyPNGImage]) -> str:
|
|
stream = BytesIO()
|
|
img.save(stream)
|
|
base64_value = base64.b64encode(stream.getvalue()).decode(encoding="ascii")
|
|
|
|
return base64_value
|