76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
""" Test Email client using mylib.config.Config for configuration """
|
|
import datetime
|
|
import logging
|
|
import sys
|
|
|
|
from mako.template import Template as MakoTemplate
|
|
|
|
from mylib.config import Config
|
|
from mylib.email import EmailClient
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def main(argv=None): # pylint: disable=too-many-locals,too-many-statements
|
|
"""Script main"""
|
|
if argv is None:
|
|
argv = sys.argv[1:]
|
|
|
|
config = Config(__doc__, __name__.replace(".", "_"))
|
|
|
|
email_client = EmailClient(config=config)
|
|
email_client.configure()
|
|
|
|
# Options parser
|
|
parser = config.get_arguments_parser(description=__doc__)
|
|
|
|
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 = config.parse_arguments_options()
|
|
|
|
if not options.test_to:
|
|
parser.error("You must specify test email recipient using -t/--to parameter")
|
|
sys.exit(1)
|
|
|
|
email_client.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=(
|
|
"<strong>Just a test email.</strong> <small>(sent at {sent_date})</small>"
|
|
if not options.test_mako
|
|
else MakoTemplate(
|
|
"<strong>Just a test email.</strong> <small>(sent at ${sent_date})</small>"
|
|
)
|
|
),
|
|
)
|
|
)
|
|
|
|
logging.info("Send a test email to %s", options.test_to)
|
|
if email_client.send(options.test_to, template="test", sent_date=datetime.datetime.now()):
|
|
logging.info("Test email sent")
|
|
sys.exit(0)
|
|
logging.error("Fail to send test email")
|
|
sys.exit(1)
|