Doc/includes/email-dir.py

   1 #!/usr/bin/env python
   2
   3 """Send the contents of a directory as a MIME message."""
   4
   5 import os
   6 import sys
   7 import smtplib
   8 # For guessing MIME type based on file name extension
   9 import mimetypes
  10
  11 from optparse import OptionParser
  12
  13 from email import encoders
  14 from email.message import Message
  15 from email.mime.audio import MIMEAudio
  16 from email.mime.base import MIMEBase
  17 from email.mime.image import MIMEImage
  18 from email.mime.multipart import MIMEMultipart
  19 from email.mime.text import MIMEText
  20
  21 COMMASPACE = ', '
  22
  23
  24 def main():
  25     parser = OptionParser(usage="""\
  26 Send the contents of a directory as a MIME message.
  27
  28 Usage: %prog [options]
  29
  30 Unless the -o option is given, the email is sent by forwarding to your local
  31 SMTP server, which then does the normal delivery process.  Your local machine
  32 must 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()
 109         s.connect()
 110         s.sendmail(opts.sender, opts.recipients, composed)
 111         s.close()
 112
 113
 114 if __name__ == '__main__':
 115     main()