1#!/usr/bin/env python 2 3"""Send the contents of a directory as a MIME message.""" 4 5import os 6import sys 7import smtplib 8# For guessing MIME type based on file name extension 9import mimetypes 10 11from optparse import OptionParser 12 13from email import encoders 14from email.message import Message 15from email.mime.audio import MIMEAudio 16from email.mime.base import MIMEBase 17from email.mime.image import MIMEImage 18from email.mime.multipart import MIMEMultipart 19from email.mime.text import MIMEText 20 21COMMASPACE = ', ' 22 23 24def main(): 25 parser = OptionParser(usage="""\ 26Send the contents of a directory as a MIME message. 27 28Usage: %prog [options] 29 30Unless the -o option is given, the email is sent by forwarding to your local 31SMTP server, which then does the normal delivery process. Your local machine 32must be running an SMTP server. 33""") 34 parser.add_option('-d', '--directory', 35 type='string', action='store', 36 help="""Mail the contents of the specified directory, 37 otherwise use the current directory. Only the regular 38 files in the directory are sent, and we don't recurse to 39 subdirectories.""") 40 parser.add_option('-o', '--output', 41 type='string', action='store', metavar='FILE', 42 help="""Print the composed message to FILE instead of 43 sending the message to the SMTP server.""") 44 parser.add_option('-s', '--sender', 45 type='string', action='store', metavar='SENDER', 46 help='The value of the From: header (required)') 47 parser.add_option('-r', '--recipient', 48 type='string', action='append', metavar='RECIPIENT', 49 default=[], dest='recipients', 50 help='A To: header value (at least one required)') 51 opts, args = parser.parse_args() 52 if not opts.sender or not opts.recipients: 53 parser.print_help() 54 sys.exit(1) 55 directory = opts.directory 56 if not directory: 57 directory = '.' 58 # Create the enclosing (outer) message 59 outer = MIMEMultipart() 60 outer['Subject'] = 'Contents of directory %s' % os.path.abspath(directory) 61 outer['To'] = COMMASPACE.join(opts.recipients) 62 outer['From'] = opts.sender 63 outer.preamble = 'You will not see this in a MIME-aware mail reader.\n' 64 65 for filename in os.listdir(directory): 66 path = os.path.join(directory, filename) 67 if not os.path.isfile(path): 68 continue 69 # Guess the content type based on the file's extension. Encoding 70 # will be ignored, although we should check for simple things like 71 # gzip'd or compressed files. 72 ctype, encoding = mimetypes.guess_type(path) 73 if ctype is None or encoding is not None: 74 # No guess could be made, or the file is encoded (compressed), so 75 # use a generic bag-of-bits type. 76 ctype = 'application/octet-stream' 77 maintype, subtype = ctype.split('/', 1) 78 if maintype == 'text': 79 fp = open(path) 80 # Note: we should handle calculating the charset 81 msg = MIMEText(fp.read(), _subtype=subtype) 82 fp.close() 83 elif maintype == 'image': 84 fp = open(path, 'rb') 85 msg = MIMEImage(fp.read(), _subtype=subtype) 86 fp.close() 87 elif maintype == 'audio': 88 fp = open(path, 'rb') 89 msg = MIMEAudio(fp.read(), _subtype=subtype) 90 fp.close() 91 else: 92 fp = open(path, 'rb') 93 msg = MIMEBase(maintype, subtype) 94 msg.set_payload(fp.read()) 95 fp.close() 96 # Encode the payload using Base64 97 encoders.encode_base64(msg) 98 # Set the filename parameter 99 msg.add_header('Content-Disposition', 'attachment', filename=filename) 100 outer.attach(msg) 101 # Now send or store the message 102 composed = outer.as_string() 103 if opts.output: 104 fp = open(opts.output, 'w') 105 fp.write(composed) 106 fp.close() 107 else: 108 s = smtplib.SMTP('localhost') 109 s.sendmail(opts.sender, opts.recipients, composed) 110 s.quit() 111 112 113if __name__ == '__main__': 114 main() 115