← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Doc/includes/email-dir.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
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()