qr_handler.py

import qrcode
from PIL import Image
from io import BytesIO
import os
import json
import numpy
import cv2
from dotenv import load_dotenv
from utils import get_ip
class QrHandler():
    def __init__(self):
        self.port = os.getenv('PORT')
    def generate_qrcode(self,text):
        qr = qrcode.QRCode(
            version=1,
            error_correction=qrcode.constants.ERROR_CORRECT_L,
            box_size=10,
            border=1,
        )
        qr.add_data(text)
        qr.make(fit=True)
        img = qr.make_image(fill_color="black", back_color="white")
        return img

    def save_qrcode_to_bytes(self,text):
        img = self.generate_qrcode(text)
        bytes_io = BytesIO()
        img.save(bytes_io, format="PNG")
        bytes_io.seek(0)
        return bytes_io

    def generate_qr_endpoint(self,prefix,classId,className,lessonId,token="token"):
        address = f"{get_ip(prefix)}:{self.port}"
        json_data = {"address": address, "token": token,"classId":classId, "className": className,"lessonId":lessonId}
        json_data = json.dumps(json_data)
        img = numpy.array(self.generate_qrcode(json_data).convert("RGB"))
        return img

    def test_gen(self):
        image = self.generate_qr_endpoint("192.168.1",classId=2,className="Physics",lessonId="2")
        cv2.imshow("test",image)
        cv2.waitKey(0)
        # closing all open windows
        cv2.destroyAllWindows()

if __name__ == "__main__":
    load_dotenv()
    test = QrHandler()
    test.test_gen()