source: rtems-tools/rtemstoolkit/mailer.py @ 5cb66b4

4.105
Last change on this file since 5cb66b4 was 50fdf12, checked in by Chris Johns <chrisj@…>, on 02/14/14 at 19:30:06

rt: Add the rtems-tester.

  • Property mode set to 100644
File size: 4.3 KB
Line 
1#
2# RTEMS Tools Project (http://www.rtems.org/)
3# Copyright 2013-2014 Chris Johns (chrisj@rtems.org)
4# All rights reserved.
5#
6# This file is part of the RTEMS Tools package in 'rtems-tools'.
7#
8# Redistribution and use in source and binary forms, with or without
9# modification, are permitted provided that the following conditions are met:
10#
11# 1. Redistributions of source code must retain the above copyright notice,
12# this list of conditions and the following disclaimer.
13#
14# 2. Redistributions in binary form must reproduce the above copyright notice,
15# this list of conditions and the following disclaimer in the documentation
16# and/or other materials provided with the distribution.
17#
18# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
22# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28# POSSIBILITY OF SUCH DAMAGE.
29#
30
31#
32# Manage emailing results or reports.
33#
34
35import os
36import smtplib
37import socket
38
39import error
40import options
41import path
42
43def append_options(opts):
44    opts['--mail'] = 'Send email report or results.'
45    opts['--smtp-host'] = 'SMTP host to send via.'
46    opts['--mail-to'] = 'Email address to send the email too.'
47    opts['--mail-from'] = 'Email address the report is from.'
48
49class mail:
50    def __init__(self, opts):
51        self.opts = opts
52
53    def from_address(self):
54
55        def _clean(l):
56            if '#' in l:
57                l = l[:l.index('#')]
58            if '\r' in l:
59                l = l[:l.index('r')]
60            if '\n' in l:
61                l = l[:l.index('\n')]
62            return l.strip()
63
64        addr = self.opts.get_arg('--mail-from')
65        if addr is not None:
66            return addr[1]
67        mailrc = None
68        if 'MAILRC' in os.environ:
69            mailrc = os.environ['MAILRC']
70        if mailrc is None and 'HOME' in os.environ:
71            mailrc = path.join(os.environ['HOME'], '.mailrc')
72        if mailrc is not None and path.exists(mailrc):
73            # set from="Joe Blow <joe@blow.org>"
74            try:
75                mrc = open(mailrc, 'r')
76                lines = mrc.readlines()
77                mrc.close()
78            except IOError, err:
79                raise error.general('error reading: %s' % (mailrc))
80            for l in lines:
81                l = _clean(l)
82                if 'from' in l:
83                    fa = l[l.index('from') + len('from'):]
84                    if '=' in fa:
85                        addr = fa[fa.index('=') + 1:].replace('"', ' ').strip()
86            if addr is not None:
87                return addr
88        addr = self.opts.defaults.get_value('%{_sbgit_mail}')
89        return addr
90
91    def smtp_host(self):
92        host = self.opts.get_arg('--smtp-host')
93        if host is not None:
94            return host[1]
95        host = self.opts.defaults.get_value('%{_mail_smtp_host}')
96        if host is not None:
97            return host
98        return 'localhost'
99
100    def send(self, to_addr, subject, body):
101        from_addr = self.from_address()
102        msg = "From: %s\r\nTo: %s\r\nSubject: %s\r\n\r\n" % \
103            (from_addr, to_addr, subject) + body
104        try:
105            s = smtplib.SMTP(self.smtp_host())
106            s.sendmail(from_addr, [to_addr], msg)
107        except smtplib.SMTPException, se:
108            raise error.general('sending mail: %s' % (str(se)))
109        except socket.error, se:
110            raise error.general('sending mail: %s' % (str(se)))
111
112if __name__ == '__main__':
113    import sys
114    optargs = {}
115    append_options(optargs)
116    opts = options.load(sys.argv, optargs = optargs, defaults = 'defaults.mc')
117    m = mail(opts)
118    print 'From: %s' % (m.from_address())
119    print 'SMTP Host: %s' % (m.smtp_host())
120    m.send(m.from_address(), 'Test mailer.py', 'This is a test')
Note: See TracBrowser for help on using the repository browser.