1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#
# Copyright (c) 2014 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
from StringIO import StringIO
from email.utils import parseaddr
from leap.mail.outgoing.service import OutgoingMail
from twisted.internet.defer import Deferred, fail
from twisted.mail.smtp import SMTPSenderFactory
from twisted.internet import reactor, defer
from pixelated.support.functional import flatten
class SMTPDownException(Exception):
def __init__(self):
Exception.__init__(self, "Couldn't send mail now, try again later.")
class MailSender(object):
def __init__(self, smtp_config, keymanager):
self._smtp_config = smtp_config
self._keymanager = keymanager
def sendmail(self, mail):
recipients = flatten([mail.to, mail.cc, mail.bcc])
outgoing_mail = self._create_outgoing_mail()
deferreds = []
for recipient in recipients:
deferreds.append(outgoing_mail.send_message(mail.to_smtp_format(), recipient))
return defer.gatherResults(deferreds)
def _create_outgoing_mail(self):
return OutgoingMail(str(self._smtp_config.account_email),
self._keymanager,
self._smtp_config.cert_path,
self._smtp_config.cert_path,
str(self._smtp_config.remote_smtp_host),
int(self._smtp_config.remote_smtp_port))
class LocalSmtpMailSender(object):
def __init__(self, account_email_address, smtp):
self.smtp = smtp
self.account_email_address = account_email_address
def sendmail(self, mail):
if self.smtp.ensure_running():
recipients = flatten([mail.to, mail.cc, mail.bcc])
result_deferred = Deferred()
sender_factory = SMTPSenderFactory(
fromEmail=self.account_email_address,
toEmail=set([parseaddr(recipient)[1] for recipient in recipients]),
file=StringIO(mail.to_smtp_format()),
deferred=result_deferred)
reactor.connectTCP('localhost', self.smtp.local_smtp_port_number,
sender_factory)
return result_deferred
return fail(SMTPDownException())
|
