pyqr.py

#!/usr/bin/python
# -*- coding: utf8 -*-
# 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 Lesser General Public License
# for more details.

"Módulo para generar códigos QR"
from __future__ import print_function
from __future__ import unicode_literals
from builtins import object

__author__ = "Mariano Reingart <reingart@gmail.com>"
__copyright__ = "Copyright (C) 2020-2021 Mariano Reingart"
__license__ = "LGPL-3.0-or-later"
__version__ = "3.05b"

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

import qrcode


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

TYPELIB = False


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

    _reg_progid_ = "PyQR"
    _reg_clsid_ = "{B176B1CE-E7B5-4BB2-ADEC-9EB9F249DF07}"

    if TYPELIB:
        _typelib_guid_ = '{418C11BF-1051-4B51-95CE-638DC3686634}'
        _typelib_version_ = 1, 5
        _com_interfaces_ = ['IPyQR']

    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,
        color_relleno="black",
        color_fondo="white",
    ):
        "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.encode('ascii')).decode('ascii'))

        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=color_relleno, back_color=color_fondo)

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

        return url


from pyafipws.utils import get_install_dir
INSTALL_DIR = PyQR.InstallDir = get_install_dir()


def main():
    if "--register" in sys.argv or "--unregister" in sys.argv:
        import pythoncom
        if TYPELIB:
            if '--register' in sys.argv:
                tlb = os.path.abspath(os.path.join(INSTALL_DIR, "typelib", "pyqr.tlb"))
                print("Registering %s" % (tlb, ))
                tli=pythoncom.LoadTypeLib(tlb)
                pythoncom.RegisterTypeLib(tli, tlb)
            elif '--unregister' in sys.argv:
                k = PyQR
                pythoncom.UnRegisterTypeLib(k._typelib_guid_,
                                            k._typelib_version_[0],
                                            k._typelib_version_[1],
                                            0,
                                            pythoncom.SYS_WIN32)
                print("Unregistered typelib")
        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" or sys.platform == "linux":
            os.system("eog " "%s" "" % pyqr.Archivo)
        else:
            os.startfile(pyqr.Archivo)

    return url

if __name__ == "__main__":
    main()