pyemail.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 enviar correos electrónicos"
from __future__ import print_function
from __future__ import absolute_import

from future import standard_library

standard_library.install_aliases()
from builtins import input
from builtins import object

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

import os
import sys
import traceback

from email.mime.text import MIMEText
from email.mime.application import MIMEApplication
from email.mime.multipart import MIMEMultipart
import sys, os
import smtplib
from configparser import SafeConfigParser


DEBUG = False


class PyEmail(object):
    "Interfaz para enviar correos de Factura Electrónica"
    _public_methods_ = [
        "Conectar",
        "Crear",
        "Enviar",
        "AgregarDestinatario",
        "Adjuntar",
        "AgregarCC",
        "AgregarBCC",
    ]
    _public_attrs_ = [
        "Motivo",
        "Remitente",
        "Destinatarios",
        "ResponderA",
        "MensajeHTML",
        "MensajeTexto",
        "Version",
        "Excepcion",
        "Traceback",
    ]

    _reg_progid_ = "PyEmail"
    _reg_clsid_ = "{2BEF3037-BF38-41AA-84A3-6F109D543FC9}"

    def __init__(self):
        self.Version = __version__
        self.Excepcion = self.Traceback = ""
        self.Motivo = self.Destinatario = self.ResponderA = ""
        self.MensajeHTML = self.MensajeTexto = None
        self.adjuntos = []
        self.BCC = []
        self.CC = []

    def Conectar(self, servidor, usuario=None, clave=None, puerto=25):
        "Iniciar conexión al servidor de correo electronico"
        try:
            # convertir el nro de puerto a entero porque puede ser string:
            puerto = int(puerto)
            if puerto != 465:
                self.smtp = smtplib.SMTP(servidor, puerto)
            else:
                # creo una conexión segura (SSL, no disponible en Python<2.6):
                self.smtp = smtplib.SMTP_SSL(servidor, puerto)
            if DEBUG:
                self.smtp.set_debuglevel(1)
            self.smtp.ehlo()
            if puerto == 587:
                # inicio una sesión segura (TLS)
                self.smtp.starttls()
            if usuario and clave:
                if sys.version_info[0] < 3:
                    #convertir a string (hmac necesita string "bytes")
                    if isinstance(usuario, str):
                        usuario = usuario.encode("utf8")
                    if isinstance(clave, str):
                        clave = clave.encode("utf8")
                self.smtp.login(usuario, clave)
            return True
        except Exception as e:
            ex = traceback.format_exception(
                sys.exc_info()[0], sys.exc_info()[1], sys.exc_info()[2]
            )
            self.Traceback = "".join(ex)
            self.Excepcion = traceback.format_exception_only(
                sys.exc_info()[0], sys.exc_info()[1]
            )[0]
            return False

    def Crear(self, remitente="", motivo=""):
        "Inicializa un mensaje de correo"
        self.Remitente = remitente
        self.Motivo = motivo
        self.Destinatarios = []
        self.adjuntos = []
        return True

    def AgregarDestinatario(self, destinatario):
        "Agrega una dirección de correo de destino"
        self.Destinatarios.append(destinatario)
        return True

    def AgregarCC(self, destinatario):
        "Agrega una dirección de correo de destino (copia carbónica)"
        self.CC.append(destinatario)
        return True

    def AgregarBCC(self, destinatario):
        "Agrega una dirección de correo de destino (copia carbónica invisible)"
        self.BCC.append(destinatario)
        return True

    def Adjuntar(self, archivo):
        "Agrega un archivo para ser enviado como adjunto"
        self.adjuntos.append(archivo)
        return True

    def Enviar(
        self, remitente="", motivo="", destinatario="", mensaje="", archivo=None
    ):
        "Generar un correo multiparte y enviarlo"
        try:
            to = [destinatario] if destinatario else self.Destinatarios

            msg = MIMEMultipart("related")
            msg["Subject"] = motivo or self.Motivo
            msg["From"] = remitente or self.Remitente
            msg["Reply-to"] = remitente or self.ResponderA
            msg["To"] = ", ".join(to)
            if self.CC:
                msg["CC"] = u", ".join(self.CC)
                to += self.CC
            if self.BCC:
                to += self.BCC

            msg.preamble = "Mensaje de multiples partes.\n"

            if mensaje:
                text = mensaje
                html = None
            else:
                text = self.MensajeTexto
                html = self.MensajeHTML

            if html:
                alt = MIMEMultipart("alternative")
                msg.attach(alt)
                part = MIMEText(text, "text")
                alt.attach(part)
                part = MIMEText(html, "html")
                alt.attach(part)
            else:
                part = MIMEText(text)
                msg.attach(part)

            if archivo:
                self.adjuntos.append(archivo)

            for archivo in self.adjuntos:
                part = MIMEApplication(open(archivo, "rb").read())
                part.add_header(
                    "Content-Disposition",
                    "attachment",
                    filename=os.path.basename(archivo),
                )
                msg.attach(part)

            # print "Enviando email: %s a %s" % (msg['Subject'], msg['To'])
            self.smtp.sendmail(msg["From"], to, msg.as_string())

            return True
        except Exception as e:
            ex = traceback.format_exception(
                sys.exc_info()[0], sys.exc_info()[1], sys.exc_info()[2]
            )
            self.Traceback = "".join(ex)
            self.Excepcion = traceback.format_exception_only(
                sys.exc_info()[0], sys.exc_info()[1]
            )[0]
            return False

    def Salir(self):
        "Termino la conexión al servidor de correo electronico"
        try:
            self.smtp.quit()
            return True
        except Exception as e:
            ex = traceback.format_exception(
                sys.exc_info()[0], sys.exc_info()[1], sys.exc_info()[2]
            )
            self.Traceback = "".join(ex)
            self.Excepcion = traceback.format_exception_only(
                sys.exc_info()[0], sys.exc_info()[1]
            )[0]
            return False


def main():
    global DEBUG
    if "--register" in sys.argv or "--unregister" in sys.argv:
        import win32com.server.register

        win32com.server.register.UseCommandLine(PyEmail)
    elif "py2exe" in sys.argv:
        from distutils.core import setup
        from pyafipws.nsis import build_installer, Target
        import py2exe

        setup(
            name="PyEmail",
            version=__version__,
            description="Interfaz PyAfipWs Email %s",
            long_description=__doc__,
            author="Mariano Reingart",
            author_email="reingart@gmail.com",
            url="http://www.sistemasagiles.com.ar",
            license="GNU GPL v3",
            com_server=["pyemail"],
            console=[],
            options={
                "py2exe": {
                    "includes": [
                        "email.generator",
                        "email.iterators",
                        "email.message",
                        "email.utils",
                    ],
                    "optimize": 2,
                    "excludes": [
                        "pywin",
                        "pywin.dialogs",
                        "pywin.dialogs.list",
                        "win32ui",
                        "distutils.core",
                        "py2exe",
                        "nsis",
                    ],
                    #'skip_archive': True,
                }
            },
            data_files=[
                (".", ["licencia.txt"]),
            ],
            cmdclass={"py2exe": build_installer},
        )
    elif "/Automate" in sys.argv:
        # MS seems to like /automate to run the class factories.
        import win32com.server.localserver

        # win32com.server.localserver.main()
        # start the server.
        win32com.server.localserver.serve([PyEmail._reg_clsid_])
    elif "/prueba" in sys.argv:
        pyemail = PyEmail()
        import getpass
        i = sys.argv.index("/prueba")
        if i + 2 < len(sys.argv):
            usuario = sys.argv[sys.argv.index("/prueba") + 1]
            clave = sys.argv[sys.argv.index("/prueba") + 2]
        else:
            usuario = input("usuario:")
            clave = getpass.getpass("clave:")
        ok = pyemail.Conectar("smtp.gmail.com", usuario=usuario, clave=clave, puerto=587)
        print("login ok?", ok, pyemail.Excepcion)
        print(pyemail.Traceback)
        ok = pyemail.Enviar(usuario, "prueba", usuario, "prueba!", None)
        print("mail enviado?", ok, pyemail.Excepcion)
        ok = pyemail.Salir()
    else:
        config = SafeConfigParser()
        config.read("rece.ini")

        if "/debug" in sys.argv:
            DEBUG = True
            print("VERSION", __version__)
            sys.argv.remove("/debug")

        if len(sys.argv) < 3:
            print("Parámetros: motivo destinatario [mensaje] [archivo]")
            sys.exit(1)

        conf_mail = dict(config.items("MAIL"))
        motivo = sys.argv[1]
        destinatario = sys.argv[2]
        mensaje = len(sys.argv) > 3 and sys.argv[3] or conf_mail["cuerpo"]
        archivo = len(sys.argv) > 4 and sys.argv[4] or None

        print("Motivo: ", motivo)
        print("Destinatario: ", destinatario)
        print("Mensaje: ", mensaje)
        print("Archivo: ", archivo)

        pyemail = PyEmail()
        ok = pyemail.Conectar(
            conf_mail["servidor"],
            conf_mail["usuario"],
            conf_mail["clave"],
            conf_mail.get("puerto", 25),
        )
        if ok:
            pyemail.Enviar(
                conf_mail["remitente"], motivo, destinatario, mensaje, archivo
            )
        else:
            print(pyemail.Traceback)

if __name__ == "__main__":
    main()