First commit

mailex.py

@@ -0,0 +1,76 @@
+#!/usr/bin/env python
+import smtplib, os
+from email.MIMEMultipart import MIMEMultipart
+from email.MIMEBase import MIMEBase
+from email.MIMEText import MIMEText
+from email.Utils import COMMASPACE, formatdate
+from email import Encoders
+import argparse, getpass, socket, sys
+
+
+def send_mail(send_from, send_to, subject, text, files=[], server="localhost"):
+    assert type(send_to) == list
+    assert type(files) == list
+
+    try:
+        msg = MIMEMultipart()
+        msg['From'] = send_from
+        msg['To'] = COMMASPACE.join(send_to)
+        msg['Date'] = formatdate(localtime=True)
+        msg['Subject'] = subject
+
+        msg.attach(MIMEText(text))
+
+        for f in files:
+            try:
+                part = MIMEBase('application', "octet-stream")
+                part.set_payload(open(f, "rb").read())
+                Encoders.encode_base64(part)
+                part.add_header('Content-Disposition', 'attachment; filename="%s"' % os.path.basename(f))
+                msg.attach(part)
+            except TypeError:
+                print "Failed to attach file: %s" % f
+
+        smtp = smtplib.SMTP(server)
+        smtp.sendmail(send_from, send_to, msg.as_string())
+        smtp.close()
+    except smtplib.SMTPRecipientsRefused, smtpE:
+        print smtpE.message
+
+def _get_default_from():
+    return "%s@%s" % (getpass.getuser(), socket.gethostname())
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    #commands
+    parser.add_argument("-v", "--version", action="version", version="0.1")
+
+    #required
+    parser.add_argument("-t", "--to-address", help="To address", required=True)
+
+    #optional
+    parser.add_argument("-f", "--from-address", nargs="?", default=_get_default_from(), help="From address")
+    parser.add_argument("-a", "--attachments", help="File(s) to attach")
+    parser.add_argument("-s", "--subject", nargs="?", help="Mail subject", default="")
+    parser.add_argument("-b", "--mail-body", nargs="?", help="Mail body", default="")
+    args = parser.parse_args()
+
+    #read any piped text
+    if sys.stdin.isatty():
+        if args.mail_body:
+            body = args.mail_body
+        else:
+            body = raw_input('Enter message\n')
+    else:
+        body = sys.stdin.read()
+
+    files = []
+    if args.attachments:
+        files = [args.attachments]
+
+    send_mail(
+        send_from=args.from_address,
+        send_to=[args.to_address],
+        subject=args.subject,
+        text=body,
+        files=files)