pyqr.py

#!/usr/bin/python
# -*- coding: latin-1 -*-
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by the
# Free Software Foundation; either version 3, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTIBILITY
# or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
# for more details.

"Módulo para generar códigos QR"

__author__ = "Mariano Reingart <reingart@gmail.com>"
__copyright__ = "Copyright (C) 2011 Mariano Reingart"
__license__ = "LGPL 3.0"
__version__ = "1.04b"

import base64
import json
import os
import sys
import tempfile
import traceback

import qrcode


TEST_QR_DATA = """
eyJ2ZXIiOjEsImZlY2hhIjoiMjAyMC0xMC0xMyIsImN1aXQiOjMwMDAwMDAwMDA3LCJwdG9WdGEiOj
EwLCJ0aXBvQ21wIjoxLCJucm9DbXAiOjk0LCJpbXBvcnRlIjoxMjEwMCwibW9uZWRhIjoiRE9MIiwi
Y3R6Ijo2NSwidGlwb0RvY1JlYyI6ODAsIm5yb0RvY1JlYyI6MjAwMDAwMDAwMDEsInRpcG9Db2RBdX
QiOiJFIiwiY29kQXV0Ijo3MDQxNzA1NDM2NzQ3Nn0=""".replace("\n", "")


class PyQR:
    "Interfaz para generar Codigo QR de Factura Electr�nica"
    _public_methods_ = ['GenerarImagen', 'CrearArchivo',
                        ]
    _public_attrs_ = ['Version', 'Excepcion', 'Traceback', "URL",
                      "Archivo", "Extension",
                      'qr_ver', 'box_size', 'border', 'error_correction',
                     ]

    _reg_progid_ = "PyQR"
    _reg_clsid_ = "{0868A2B6-2DC7-478D-8884-A10E92C588DE}"

    URL = "https://www.afip.gob.ar/fe/qr/?p=%s"
    Archivo = "qr.png"
    Extension = "PNG"

    # qrencode default parameters:
    qr_ver = 1
    box_size = 10
    border = 4
    error_correction = qrcode.constants.ERROR_CORRECT_L

    def __init__(self):
        self.Version = __version__
        self.Exception = self.Traceback = ""

    def CrearArchivo(self):
        """Crea un nombre de archivo temporal"""
        # para evitar errores de permisos y poder generar varios qr simultaneos
        tmp = tempfile.NamedTemporaryFile(prefix="qr_afip_",
                                          suffix=".%s" % self.Extension.lower(),
                                          delete=False)
        self.Archivo = tmp.name
        return self.Archivo

    def GenerarImagen(self, ver=1,
                      fecha="2020-10-13",
                      cuit=30000000007,
                      pto_vta=10, tipo_cmp=1, nro_cmp=94,
                      importe=12100, moneda="PES", ctz=1.000,
                      tipo_doc_rec=80, nro_doc_rec=20000000001,
                      tipo_cod_aut="E", cod_aut=70417054367476,
                      ):
        "Generar una imágen con el código QR"
        # basado en: https://www.afip.gob.ar/fe/qr/especificaciones.asp
        datos_cmp = {
            "ver": int(ver),
            "fecha": fecha,
            "cuit": int(cuit),
            "ptoVta": int(pto_vta),
            "tipoCmp": int(tipo_cmp),
            "nroCmp": int(nro_cmp),
            "importe": float(importe),
            "moneda": moneda,
            "ctz": float(ctz),
            "tipoDocRec": int(tipo_doc_rec),
            "nroDocRec": int(nro_doc_rec),
            "tipoCodAut": tipo_cod_aut,
            "codAut": int(cod_aut),
            }

        # convertir a representación json y codificar en base64:
        datos_cmp_json = json.dumps(datos_cmp)
        url = self.URL % (base64.b64encode(datos_cmp_json))

        qr = qrcode.QRCode(
            version=self.qr_ver,
            error_correction=self.error_correction,
            box_size=self.box_size,
            border=self.border,
        )
        qr.add_data(url)
        qr.make(fit=True)

        img = qr.make_image(fill_color="black", back_color="white")

        img.save(self.Archivo, self.Extension.upper())

        return url


if __name__ == '__main__':

    if "--register" in sys.argv or "--unregister" in sys.argv:
        import win32com.server.register
        win32com.server.register.UseCommandLine(PyQR)
    elif "/Automate" in sys.argv:
        try:
            # MS seems to like /automate to run the class factories.
            import win32com.server.localserver
            win32com.server.localserver.serve([PyQR._reg_clsid_])
        except Exception:
            raise
    else:

        pyqr = PyQR()

        if '--datos' in sys.argv:
            args = sys.argv[sys.argv.index("--datos")+1:]
            (ver, fecha, cuit, pto_vta, tipo_cmp, nro_cmp, importe, moneda, ctz,
            tipo_doc_rec, nro_doc_rec, tipo_cod_aut, cod_aut) = args
        else:
            ver = 1
            fecha = "2020-10-13"
            cuit = 30000000007
            pto_vta = 10
            tipo_cmp = 1
            nro_cmp = 94
            importe = 12100
            moneda = "DOL"
            ctz = 65.000
            tipo_doc_rec = 80
            nro_doc_rec = 20000000001
            tipo_cod_aut = "E"
            cod_aut = 70417054367476

        if '--archivo' in sys.argv:
            pyqr.Archivo = sys.argv[sys.argv.index("--archivo")+1]
            ext = os.path.splitext(pyqr.Archivo)[1][1:].upper()
            if ext == "JPG":
                ext = "JPEG"
            PyQR.Extension = ext
        else:
            pyqr.CrearArchivo()

        if '--size' in sys.argv:
            pyqr.box_size = int(sys.argv[sys.argv.index("--size")+1])

        if '--border' in sys.argv:
            pyqr.border = int(sys.argv[sys.argv.index("--border")+1])

        if '--url' in sys.argv:
            pyqr.URL = sys.argv[sys.argv.index("--url")+1]

        print "datos:", (ver, fecha, cuit, pto_vta, tipo_cmp, nro_cmp,
                         importe, moneda, ctz, tipo_doc_rec, nro_doc_rec,
                         tipo_cod_aut, cod_aut)
        print "archivo", pyqr.Archivo
        print "extension", pyqr.Extension

        url = pyqr.GenerarImagen(ver, fecha, cuit, pto_vta, tipo_cmp, nro_cmp,
                                importe, moneda, ctz, tipo_doc_rec, nro_doc_rec,
                                tipo_cod_aut, cod_aut)

        print "url generada:", url

        if "--prueba" in sys.argv:
            qr_data_test = json.loads(base64.b64decode(TEST_QR_DATA))
            qr_data_gen = json.loads(base64.b64decode(url[33:]))
            assert url.startswith("https://www.afip.gob.ar/fe/qr/?p=")
            assert qr_data_test == qr_data_gen, "Diff: %r != %r" % (qr_data_test, qr_data_gen)
            print "QR data ok:", qr_data_gen

        if not '--mostrar' in sys.argv:
            pass
        elif sys.platform=="linux2":
            os.system("eog ""%s""" % pyqr.Archivo)
        else:
            os.startfile(pyqr.archivo)