fix save_qrcode

diff --git a/lib/bmp.py b/lib/bmp.py
index b4bd410..66a1007 100644 (file)
--- a/lib/bmp.py
+++ b/lib/bmp.py
@@ -194,14 +194,13 @@ class BitMap(object):
   \r
 \r
 def save_qrcode(qr, filename):\r
-    bitmap = BitMap( 35*8, 35*8 )\r
-    #print len(bitmap.bitarray)\r
+    k = qr.moduleCount\r
+    bitmap = BitMap( (k+2)*8, (k+2)*8 )\r
     bitmap.bitarray = []\r
-    k = 33\r
-    for r in range(35):\r
-        tmparray = [ 0 ] * 35*8\r
+    for r in range(k+2):\r
+        tmparray = [ 0 ] * (k+2)*8\r
 \r
-        if 0 < r < 34:\r
+        if 0 < r < k+1:\r
             for c in range(k):\r
                 if qr.isDark(r-1, c):\r
                     tmparray[ (1+c)*8:(2+c)*8] = [1]*8\r