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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
|
# This file is part of Tryton. The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
import logging
import smtplib
import ssl
import time
from email.message import Message
from email.mime.text import MIMEText
from email.utils import formatdate
from urllib.parse import parse_qs, unquote_plus
from trytond.config import config, parse_uri
from trytond.transaction import Transaction
__all__ = ['sendmail_transactional', 'sendmail', 'SMTPDataManager']
logger = logging.getLogger(__name__)
retry = config.getint('email', 'retry', default=5)
def sendmail_transactional(
from_addr, to_addrs, msg, transaction=None, datamanager=None,
strict=False):
if transaction is None:
transaction = Transaction()
assert isinstance(transaction, Transaction), transaction
if datamanager is None:
datamanager = SMTPDataManager(strict=strict)
datamanager = transaction.join(datamanager)
datamanager.put(from_addr, to_addrs, msg)
def sendmail(from_addr, to_addrs, msg, server=None, strict=False):
if server is None:
server = get_smtp_server(strict=strict)
if not server:
return
quit = True
else:
assert server.uri
quit = False
if 'Date' not in msg:
msg['Date'] = formatdate()
for count in range(retry, -1, -1):
if count != retry:
time.sleep(0.02 * (retry - count))
try:
senderrs = server.sendmail(from_addr, to_addrs, msg.as_string())
except smtplib.SMTPServerDisconnected:
if count:
server = get_smtp_server(strict=strict)
if server:
continue
if strict:
raise
logger.error('fail to send email', exc_info=True)
except smtplib.SMTPResponseException as e:
if count and 400 <= e.smtp_code <= 499 and hasattr(server, 'uri'):
if e.smtp_code != 421:
server.quit()
server = get_smtp_server(server.uri, strict=strict)
if server:
continue
if strict:
raise
logger.error('fail to send email', exc_info=True)
except Exception:
if strict:
raise
logger.error('fail to send email', exc_info=True)
else:
if senderrs:
logger.warning('fail to send email to %s', senderrs)
break
if quit:
server.quit()
else:
return server
def send_test_email(to_addrs, server=None):
from_ = config.get('email', 'from')
msg = MIMEText('Success!\nYour email settings work correctly.')
msg['From'] = from_
msg['To'] = to_addrs
msg['Subject'] = 'Tryton test email'
sendmail(
config.get('email', 'from'), to_addrs, msg, server=server, strict=True)
def get_smtp_server(uri=None, strict=False):
if uri is None:
uri = config.get('email', 'uri')
ini_uri = uri
uri = parse_uri(uri)
extra = {}
if uri.query:
cast = {'timeout': int}
for key, value in parse_qs(uri.query, strict_parsing=True).items():
extra[key] = cast.get(key, lambda a: a)(value[0])
if uri.scheme.startswith('smtps'):
connector = smtplib.SMTP_SSL
extra['context'] = ssl.create_default_context()
else:
connector = smtplib.SMTP
try:
server = connector(uri.hostname, uri.port, **extra)
except Exception:
if strict:
raise
logger.error('fail to connect to %s', uri, exc_info=True)
return
if 'tls' in uri.scheme:
server.starttls(context=ssl.create_default_context())
if uri.username and uri.password:
server.login(
unquote_plus(uri.username),
unquote_plus(uri.password))
server.uri = ini_uri
return server
class SMTPDataManager(object):
def __init__(self, uri=None, strict=False):
self.uri = uri
self.strict = strict
self.queue = []
self._server = None
def put(self, from_addr, to_addrs, msg):
assert isinstance(msg, Message), msg
self.queue.append((from_addr, to_addrs, msg))
def __eq__(self, other):
if not isinstance(other, SMTPDataManager):
return NotImplemented
return (self.uri == other.uri) and (self.strict == other.strict)
def abort(self, trans):
self._finish()
def tpc_begin(self, trans):
pass
def commit(self, trans):
pass
def tpc_vote(self, trans):
if self._server is None:
self._server = get_smtp_server(self.uri, strict=self.strict)
def tpc_finish(self, trans):
if self._server is not None:
for from_addr, to_addrs, msg in self.queue:
new_server = sendmail(
from_addr, to_addrs, msg, server=self._server)
if new_server:
self._server = new_server
self._server.quit()
self._finish()
def tpc_abort(self, trans):
if self._server:
self._server.close()
self._finish()
def _finish(self):
self._server = None
self.queue = []
|
