Initial commit

Author: Arachnid

qr_encode.c

@@ -0,0 +1,40 @@
+#include <Python.h>
+#include <stdlib.h>
+#include <qrencode.h>
+
+static PyObject *qr_encode(PyObject *self, PyObject *args)
+{
+    char *str;
+    int i, version, level, hint, case_sensitive, num_pixels;
+    QRcode *code;
+    PyObject *ret;
+    
+    if(!PyArg_ParseTuple(args, "siiii", &str, &version, &level, &hint,
+                         &case_sensitive))
+        return NULL;
+    
+    code = QRcode_encodeString(str, version, level, hint, case_sensitive);
+    
+    num_pixels = code->width * code->width;
+    for(i = 0; i < num_pixels; i++)
+      code->data[i] = 255 - (code->data[i] & 0x1) * 0xFF;
+    
+    ret = Py_BuildValue("(iis#)", code->version, code->width,
+                        code->data, num_pixels);
+    QRcode_free(code);
+    return ret;
+};
+
+static PyMethodDef qr_encode_methods[] =
+{
+    {"encode", qr_encode, METH_VARARGS, "Encodes a string as a QR-code. Returns a tuple of (version, width, data)"},
+    {NULL, NULL, 0, NULL}
+};
+
+PyMODINIT_FUNC
+initqr_encode(void)
+{
+    PyObject *m = Py_InitModule("qr_encode", qr_encode_methods);
+    if(m == NULL)
+      return;
+}

qrencode/__init__.py

@@ -0,0 +1,78 @@
+from qr_encode import encode as _encode
+import Image
+
+
+QR_ECLEVEL_L = 0
+QR_ECLEVEL_M = 1
+QR_ECLEVEL_Q = 2
+QR_ECLEVEL_H = 3
+levels = [QR_ECLEVEL_L, QR_ECLEVEL_M, QR_ECLEVEL_Q, QR_ECLEVEL_H]
+
+
+QR_MODE_8 = 2
+QR_MODE_KANJI = 3
+hints = [QR_MODE_8, QR_MODE_KANJI]
+
+
+def encode(data, version=0, level=QR_ECLEVEL_L, hint=QR_MODE_8,
+           case_sensitive=True):
+    """Creates a QR-Code from string data.
+    
+    Args:
+      data: string: The data to encode in a QR-code. If a unicode string is
+          supplied, it will be encoded in UTF-8.
+      version: int: The minimum version to use. If set to 0, the library picks
+          the smallest version that the data fits in.
+      level: int: Error correction level. Defaults to 'L'.
+      hint: int: The type of data to encode. Either QR_MODE_8 or QR_MODE_KANJI.
+      case_sensitive: bool: Should string data be encoded case-preserving?
+    Returns:
+      A (version, size, image) tuple, where image is a size*size PIL image of
+      the QR-code.
+    """
+    if isinstance(data, unicode):
+        data = data.encode('utf8')
+    elif not isinstance(data, basestring):
+        raise ValueError('data argument must be a string.')
+    version = int(version)
+    if level not in levels:
+        raise ValueError('Invalid error-correction level.')
+    if hint not in hints:
+        raise ValueError('Invalid encoding mode.')
+    if case_sensitive:
+        version, size, data = _encode(data, version, level, hint, True)
+    else:
+        version, size, data = _encode(data, version, level, hint, False)
+    
+    im = Image.fromstring('L', (size, size), data)
+    return (version, size, im)
+
+def encode_scaled(data, size, version=0, level=QR_ECLEVEL_L, hint=QR_MODE_8,
+                  case_sensitive=True):
+    """Creates a QR-code from string data, resized to the specified dimensions.
+
+    Args:
+      data: string: The data to encode in a QR-code. If a unicode string is
+          supplied, it will be encoded in UTF-8.
+      size: int: Output size. If this is not an exact multiple of the QR-code's
+          dimensions, padding will be added. If this is smaller than the
+          QR-code's dimensions, it is ignored.
+      version: int: The minimum version to use. If set to 0, the library picks
+          the smallest version that the data fits in.
+      level: int: Error correction level. Defaults to 'L'.
+      hint: int: The type of data to encode. Either QR_MODE_8 or QR_MODE_KANJI.
+      case_sensitive: bool: Should string data be encoded case-preserving?
+    Returns:
+      A (version, size, image) tuple, where image is a size*size PIL image of
+      the QR-code.
+    """
+    version, src_size, im = encode(data, version, level, hint, case_sensitive)
+    if size < src_size:
+      size = src_size
+    qr_size = (size / src_size) * src_size
+    im = im.resize((qr_size, qr_size), Image.NEAREST)
+    pad = (size - qr_size) / 2
+    ret = Image.new("L", (size, size), 255)
+    ret.paste(im, (pad, pad))
+    
+    return (version, size, ret)

setup.py

@@ -0,0 +1,14 @@
+from distutils.core import setup, Extension
+
+encode = Extension('qr_encode', sources=['qr_encode.c'], libraries=['qrencode'])
+
+setup(name='qrencode',
+      version='1.0',
+      description='Encodes QR-codes.',
+      author='Nick Johnson',
+      author_email='[email protected]',
+      url='http://github.com/Arachnid/pyqrencode/tree/master',
+      long_description='''A simple wrapper for the C qrencode library.''',
+      packages=['qrencode'],
+      ext_modules=[encode],
+      requires=['PIL'])