test email script (using python)

test_email.py

#!/usr/bin/env python
# simple script to test that email (internal relay) is working.

import smtplib
import argparse
from uuid import uuid4
from email.mime.text import MIMEText

def smtp_mail(subject, message, addr_from, addr_to, smtp_host):
    msg = MIMEText(message.encode('utf-8'), 'plain', 'utf-8')
    msg['Subject'] = subject.encode('utf-8')
    msg['From'] = addr_from
    if isinstance(addr_to, (list, tuple)):
        msg['To'] = ",".join(addr_to)
    else:
        msg['To'] = addr_to
        addr_to = [addr_to]
    s = smtplib.SMTP(smtp_host)
    s.sendmail(addr_from, addr_to, msg.as_string())
    s.quit()

def test_email(addr_from, addr_to, smtp_host):
    subject = 'test email'
    message = 'test email %s' % uuid4()
    smtp_mail(subject, message, addr_from, addr_to, smtp_host)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('smtp_host')
    parser.add_argument('addr_from')
    parser.add_argument('addr_to')
    args = parser.parse_args()
    test_email(args.addr_from, args.addr_to, args.smtp_host)

if __name__ == '__main__':
    main()