netbox-qrgen/netbox_qrgen/utils.py
Ivan Schaller a556518577
init
Signed-off-by: Ivan Schaller <ivan@schaller.sh>
2023-06-21 23:31:47 +02:00

31 lines
800 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) -> PyPNGImage:
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 = qr.make_image(fill_color="black", back_color="white")
return img
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