Update qr.py

qr.py

@@ -4,7 +4,13 @@ import cv2
 import os
 from PIL import Image
 
-def make_qr(txt=None,data=None,im_size=None):
+def make_qr(txt=None,data=None,im_size=None,color_f=None,color_b=None):
+        
+    qrm = qr.QRCode(box_size=10,error_correction=qr.constants.ERROR_CORRECT_H)
+    if color_f == None:
+        color_f = "#000"
+    if color_b == None:
+        color_b = "#fff"
     if txt != None and txt != "" and data != None:
         f = Image.open(f'{data}')
         f.thumbnail((im_size,im_size))
@@ -12,7 +18,9 @@ def make_qr(txt=None,data=None,im_size=None):
         imr = open(f'tmp.jpg','rb')
         out = f'{txt}+++{base64.b64encode(imr.read())}'
         print (f'txt+data {out}')
-        img1 = qr.make(out,box_size=10,error_correction=qr.constants.ERROR_CORRECT_H)
+        qrm.add_data(out)
+        qrm.make(fit=True)
+        img1 = qrm.make_image(fill_color=color_f, back_color=color_b)
         img1.save("im.png")
         return "im.png"
     if txt == None or txt == "" and data != None:
@@ -22,14 +30,18 @@ def make_qr(txt=None,data=None,im_size=None):
         imr = open(f'tmp1.jpg','rb')
         out = f'+++{base64.b64encode(imr.read())}'
         print (f'data {out}')
-        img1 = qr.make(out,box_size=10,error_correction=qr.constants.ERROR_CORRECT_H)
+        qrm.add_data(out)
+        qrm.make(fit=True)
+        img1 = qrm.make_image(fill_color=color_f, back_color=color_b)
         img1.save("im1.png")
         return "im1.png"
         
     if txt != None and txt != "" and data == None:
         out = f'{txt}'
         print (f'txt {out}')
-        img1 = qr.make(out,box_size=10,error_correction=qr.constants.ERROR_CORRECT_H)
+        qrm.add_data(out)
+        qrm.make(fit=True)
+        img1 = qrm.make_image(fill_color=color_f, back_color=color_b)
         img1.save("im2.png")
         return "im2.png"