← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/smtp_connection.py

  • Committer: Ian Clatworthy
  • Date: 2009-09-09 11:43:10 UTC
  • mto: (4634.37.2 prepare-2.0)
  • mto: This revision was merged to the branch mainline in revision 4689.
  • Revision ID: ian.clatworthy@canonical.com-20090909114310-glw7tv76i5gnx9pt
put rules back in Makefile supporting plain-style docs

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/smtp_connection.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""A convenience class around smtplib."""
18
18
 
21
21
import smtplib
22
22
import socket
23
23
 
24
 
from bzrlib import ui
 
24
from bzrlib import (
 
25
    config,
 
26
    osutils,
 
27
    )
25
28
from bzrlib.errors import (
26
29
    NoDestinationAddress,
27
30
    SMTPError,
61
64
            return
62
65
 
63
66
        self._create_connection()
 
67
        # FIXME: _authenticate() should only be called when the server has
 
68
        # refused unauthenticated access, so it can safely try to authenticate 
 
69
        # with the default username. JRV20090407
64
70
        self._authenticate()
65
71
 
66
72
    def _create_connection(self):
79
85
            else:
80
86
                raise
81
87
 
82
 
        # If this fails, it just returns an error, but it shouldn't raise an
83
 
        # exception unless something goes really wrong (in which case we want
84
 
        # to fail anyway).
85
 
        self._connection.starttls()
 
88
        # Say EHLO (falling back to HELO) to query the server's features.
 
89
        code, resp = self._connection.ehlo()
 
90
        if not (200 <= code <= 299):
 
91
            code, resp = self._connection.helo()
 
92
            if not (200 <= code <= 299):
 
93
                raise SMTPError("server refused HELO: %d %s" % (code, resp))
 
94
 
 
95
        # Use TLS if the server advertised it:
 
96
        if self._connection.has_extn("starttls"):
 
97
            code, resp = self._connection.starttls()
 
98
            if not (200 <= code <= 299):
 
99
                raise SMTPError("server refused STARTTLS: %d %s" % (code, resp))
 
100
            # Say EHLO again, to check for newly revealed features
 
101
            code, resp = self._connection.ehlo()
 
102
            if not (200 <= code <= 299):
 
103
                raise SMTPError("server refused EHLO: %d %s" % (code, resp))
86
104
 
87
105
    def _authenticate(self):
88
106
        """If necessary authenticate yourself to the server."""
 
107
        auth = config.AuthenticationConfig()
89
108
        if self._smtp_username is None:
90
 
            return
 
109
            # FIXME: Since _authenticate gets called even when no authentication
 
110
            # is necessary, it's not possible to use the default username 
 
111
            # here yet.
 
112
            self._smtp_username = auth.get_user('smtp', self._smtp_server)
 
113
            if self._smtp_username is None:
 
114
                return
91
115
 
92
116
        if self._smtp_password is None:
93
 
            self._smtp_password = ui.ui_factory.get_password(
94
 
                'Please enter the SMTP password: %(user)s@%(host)s',
95
 
                user=self._smtp_username,
96
 
                host=self._smtp_server)
97
 
 
98
 
        self._connection.login(self._smtp_username, self._smtp_password)
 
117
            self._smtp_password = auth.get_password(
 
118
                'smtp', self._smtp_server, self._smtp_username)
 
119
 
 
120
        # smtplib requires that the username and password be byte
 
121
        # strings.  The CRAM-MD5 spec doesn't give any guidance on
 
122
        # encodings, but the SASL PLAIN spec says UTF-8, so that's
 
123
        # what we'll use.
 
124
        username = osutils.safe_utf8(self._smtp_username)
 
125
        password = osutils.safe_utf8(self._smtp_password)
 
126
 
 
127
        self._connection.login(username, password)
99
128
 
100
129
    @staticmethod
101
130
    def get_message_addresses(message):