17
17
"""A convenience class around smtplib."""
19
from email import Utils
19
from email.utils import getaddresses, parseaddr
28
from bzrlib.errors import (
31
DefaultSMTPConnectionRefused,
32
SMTPConnectionRefused,
35
smtp_password = config.Option('smtp_password', default=None,
37
Password to use for authentication to SMTP server.
39
smtp_server = config.Option('smtp_server', default=None,
41
Hostname of the SMTP server to use for sending email.
43
smtp_username = config.Option('smtp_username', default=None,
45
Username to use for authentication to SMTP server.
49
class SMTPError(BzrError):
51
_fmt = "SMTP error: %(error)s"
53
def __init__(self, error):
57
class SMTPConnectionRefused(SMTPError):
59
_fmt = "SMTP connection to %(host)s refused"
61
def __init__(self, error, host):
66
class DefaultSMTPConnectionRefused(SMTPConnectionRefused):
68
_fmt = "Please specify smtp_server. No server at default %(host)s."
71
class NoDestinationAddress(InternalBzrError):
73
_fmt = "Message does not have a destination address."
36
76
class SMTPConnection(object):
37
77
"""Connect to an SMTP server and send an email.
39
This is a gateway between bzrlib.config.Config and smtplib.SMTP. It
79
This is a gateway between breezy.config.Config and smtplib.SMTP. It
40
80
understands the basic bzr SMTP configuration information: smtp_server,
41
81
smtp_username, and smtp_password.
48
88
if self._smtp_factory is None:
49
89
self._smtp_factory = smtplib.SMTP
50
90
self._config = config
51
self._config_smtp_server = config.get_user_option('smtp_server')
91
self._config_smtp_server = config.get('smtp_server')
52
92
self._smtp_server = self._config_smtp_server
53
93
if self._smtp_server is None:
54
94
self._smtp_server = self._default_smtp_server
56
self._smtp_username = config.get_user_option('smtp_username')
57
self._smtp_password = config.get_user_option('smtp_password')
96
self._smtp_username = config.get('smtp_username')
97
self._smtp_password = config.get('smtp_password')
59
99
self._connection = None
66
106
self._create_connection()
67
107
# FIXME: _authenticate() should only be called when the server has
68
# refused unauthenticated access, so it can safely try to authenticate
108
# refused unauthenticated access, so it can safely try to authenticate
69
109
# with the default username. JRV20090407
70
110
self._authenticate()
74
114
self._connection = self._smtp_factory()
76
116
self._connection.connect(self._smtp_server)
77
except socket.error, e:
117
except socket.error as e:
78
118
if e.args[0] == errno.ECONNREFUSED:
79
119
if self._config_smtp_server is None:
80
120
raise DefaultSMTPConnectionRefused(socket.error,
96
136
if self._connection.has_extn("starttls"):
97
137
code, resp = self._connection.starttls()
98
138
if not (200 <= code <= 299):
99
raise SMTPError("server refused STARTTLS: %d %s" % (code, resp))
139
raise SMTPError("server refused STARTTLS: %d %s" %
100
141
# Say EHLO again, to check for newly revealed features
101
142
code, resp = self._connection.ehlo()
102
143
if not (200 <= code <= 299):
107
148
auth = config.AuthenticationConfig()
108
149
if self._smtp_username is None:
109
150
# FIXME: Since _authenticate gets called even when no authentication
110
# is necessary, it's not possible to use the default username
151
# is necessary, it's not possible to use the default username
112
153
self._smtp_username = auth.get_user('smtp', self._smtp_server)
113
154
if self._smtp_username is None:
131
172
"""Get the origin and destination addresses of a message.
133
174
:param message: A message object supporting get() to access its
134
headers, like email.Message or bzrlib.email_message.EmailMessage.
175
headers, like email.message.Message or
176
breezy.email_message.EmailMessage.
135
177
:return: A pair (from_email, to_emails), where from_email is the email
136
178
address in the From header, and to_emails a list of all the
137
179
addresses in the To, Cc, and Bcc headers.
139
from_email = Utils.parseaddr(message.get('From', None))[1]
181
from_email = parseaddr(message.get('From', None))[1]
140
182
to_full_addresses = []
141
183
for header in ['To', 'Cc', 'Bcc']:
142
184
value = message.get(header, None)
144
186
to_full_addresses.append(value)
145
to_emails = [ pair[1] for pair in
146
Utils.getaddresses(to_full_addresses) ]
187
to_emails = [pair[1] for pair in
188
getaddresses(to_full_addresses)]
148
190
return from_email, to_emails
153
195
The message will be sent to all addresses in the To, Cc and Bcc
156
:param message: An email.Message or email.MIMEMultipart object.
198
:param message: An email.message.Message or
199
email.mime.multipart.MIMEMultipart object.
159
202
from_email, to_emails = self.get_message_addresses(message)
166
209
self._connection.sendmail(from_email, to_emails,
167
210
message.as_string())
168
except smtplib.SMTPRecipientsRefused, e:
211
except smtplib.SMTPRecipientsRefused as e:
169
212
raise SMTPError('server refused recipient: %d %s' %
170
e.recipients.values()[0])
171
except smtplib.SMTPResponseException, e:
213
next(iter(e.recipients.values())))
214
except smtplib.SMTPResponseException as e:
172
215
raise SMTPError('%d %s' % (e.smtp_code, e.smtp_error))
173
except smtplib.SMTPException, e:
216
except smtplib.SMTPException as e:
174
217
raise SMTPError(str(e))