qr

#!/usr/bin/python
# coding: utf-8
"""
qr - Convert stdin (or the first argument) to a QR Code.

When stdout is a tty the QR Code is printed to the terminal and when stdout is
a pipe to a file an image is written. The default image format is PNG.
"""
import sys
import optparse
import qrcode

default_factories = {
    'pil': 'qrcode.image.pil.PilImage',
    'pymaging': 'qrcode.image.pure.PymagingImage',
    'svg': 'qrcode.image.svg.SvgImage',
    'svg-fragment': 'qrcode.image.svg.SvgFragmentImage',
    'svg-path': 'qrcode.image.svg.SvgPathImage',
}


def print_tty(qr, out=None):
    """
    Output the QR Code to a TTY (potentially useful for debugging).

    If the data has not been compiled yet, make it first.
    """
    if out is None:
        import sys
        out = sys.stdout

    if qr.data_cache is None:
        qr.make()

    modcount = qr.modules_count
    out.write(("█" * (modcount * 2 + 4)) + "\n")
    for r in range(modcount):
        out.write("██")
        for c in range(modcount):
            if qr.modules[r][c]:
                out.write("  ")
            else:
                out.write("██")
        out.write("██\n")
    out.write(("█" * (modcount * 2 + 4)) + "\n")
    out.flush()


def main(*args):
    qr = qrcode.QRCode()

    parser = optparse.OptionParser(usage=__doc__.strip())
    parser.add_option(
        "--factory", help="Full python path to the image factory class to "
        "create the image with. You can use the following shortcuts to the "
        "built-in image factory classes: {0}.".format(
            ", ".join(sorted(default_factories.keys()))))
    parser.add_option(
        "--optimize", type=int, help="Optimize the data by looking for chunks "
        "of at least this many characters that could use a more efficient "
        "encoding method. Use 0 to turn off chunk optimization.")
    opts, args = parser.parse_args(list(args))

    if opts.factory:
        module = default_factories.get(opts.factory, opts.factory)
        if '.' not in module:
            parser.error("The image factory is not a full python path")
        module, name = module.rsplit('.', 1)
        imp = __import__(module, {}, [], [name])
        image_factory = getattr(imp, name)
    else:
        image_factory = None

    if args:
        data = args[0]
    else:
        data = sys.stdin.read()
    if opts.optimize is None:
        qr.add_data(data)
    else:
        qr.add_data(data, optimize=opts.optimize)

    print_tty(qr)

    #img = qr.make_image(image_factory=image_factory)
    #img.save(sys.stdout)


if __name__ == "__main__":
    main(*sys.argv[1:])