""" Test Email client """ import datetime import getpass import logging import sys from mako.template import Template as MakoTemplate from mylib.scripts.helpers import add_email_opts, get_opts_parser, init_email_client, init_logging log = logging.getLogger("mylib.scripts.email_test") def main(argv=None): # pylint: disable=too-many-locals,too-many-statements """Script main""" if argv is None: argv = sys.argv[1:] # Options parser parser = get_opts_parser(just_try=True) add_email_opts(parser) test_opts = parser.add_argument_group("Test email options") test_opts.add_argument( "-t", "--to", action="store", type=str, dest="test_to", help="Test email recipient", ) test_opts.add_argument( "-m", "--mako", action="store_true", dest="test_mako", help="Test mako templating", ) options = parser.parse_args() if not options.test_to: parser.error("You must specify test email recipient using -t/--to parameter") sys.exit(1) # Initialize logs init_logging(options, "Test EmailClient") if options.email_smtp_user and not options.email_smtp_password: options.email_smtp_password = getpass.getpass("Please enter SMTP password: ") email_client = init_email_client( options, templates=dict( test=dict( subject="Test email", text=( "Just a test email sent at {sent_date}." if not options.test_mako else MakoTemplate("Just a test email sent at ${sent_date}.") ), html=( "Just a test email. (sent at {sent_date})" if not options.test_mako else MakoTemplate( "Just a test email. (sent at ${sent_date})" ) ), ) ), ) log.info("Send a test email to %s", options.test_to) if email_client.send(options.test_to, template="test", sent_date=datetime.datetime.now()): log.info("Test email sent") sys.exit(0) log.error("Fail to send test email") sys.exit(1)