Merge pull request #1 from OUDON/feat/change-image-module-size

OUDON · web-flow · commit 1657f97c6718 · 2022-06-10T11:06:56.000+09:00
feat: Add module_size parameter into QRImage()
diff --git a/example.py b/example.py
@@ -22,7 +22,7 @@ def main():
     # print(qr)
 
     # Save as png
-    image = QRImage(qr)
+    image = QRImage(qr, module_size=8)
     image.show()
     image.save("my_qr.png")
 
diff --git a/src/rmqrcode/qr_image.py b/src/rmqrcode/qr_image.py
@@ -1,11 +1,17 @@
 from .enums.color import Color
 
 from PIL import Image
+from PIL import ImageDraw
 
 
 class QRImage:
-    def __init__(self, qr):
-        self._img = Image.new('RGB', (qr.width() + 4, qr.height() + 4), (255, 255, 255))
+    def __init__(self, qr, module_size=10):
+        self._module_size = module_size
+        self._img = Image.new(
+            'RGB',
+            ((qr.width() + 4) * module_size, (qr.height() + 4) * module_size),
+            (255, 255, 255)
+        )
         self._make_image(qr)
 
 
@@ -19,11 +25,15 @@ def save(self, name):
 
 
     def _make_image(self, qr):
+        draw = ImageDraw.Draw(self._img)
         for y in range(qr.height()):
             for x in range(qr.width()):
                 r, g, b = 125, 125, 125
                 if qr.value_at(x, y) == Color.BLACK:
                     r, g, b = 0, 0, 0
                 elif qr.value_at(x, y) == Color.WHITE:
                     r, g, b, = 255, 255, 255
-                self._img.putpixel((x+2, y+2), (r, g, b))
+                draw.rectangle(
+                    xy=((x + 2) * self._module_size, (y + 2) * self._module_size, (x + 1 + 2) * self._module_size, (y + 1 + 2) * self._module_size),
+                    fill=(r, g, b)
+                )
