##// END OF EJS Templates
bookmarks: work around missing posixfile_nt.readline()
bookmarks: work around missing posixfile_nt.readline()

File last commit:

r7195:9fabcb1f default
r7666:2ad81e9b default
Show More
mail.py
170 lines | 6.3 KiB | text/x-python | PythonLexer
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889 # mail.py - mail sending bits for mercurial
#
# Copyright 2006 Matt Mackall <mpm@selenic.com>
#
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.
Matt Mackall
Simplify i18n imports
r3891 from i18n import _
Dirkjan Ochtman
replace usage of os.popen() with util.popen()...
r6548 import os, smtplib, socket
Christian Ebert
mail: add methods to handle non-ascii chars...
r7114 import email.Header, email.MIMEText, email.Utils
Dirkjan Ochtman
replace usage of os.popen() with util.popen()...
r6548 import util
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889
def _smtp(ui):
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 '''build an smtp connection and return a function to send mail'''
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889 local_hostname = ui.config('smtp', 'local_hostname')
s = smtplib.SMTP(local_hostname=local_hostname)
mailhost = ui.config('smtp', 'host')
if not mailhost:
raise util.Abort(_('no [smtp]host in hgrc - cannot send mail'))
mailport = int(ui.config('smtp', 'port', 25))
Alexis S. L. Carvalho
fix typo in mail.py
r2964 ui.note(_('sending mail: smtp host %s, port %s\n') %
(mailhost, mailport))
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889 s.connect(host=mailhost, port=mailport)
if ui.configbool('smtp', 'tls'):
Alexis S. L. Carvalho
mail.py: don't try to use TLS if python doesn't have SSL support...
r4093 if not hasattr(socket, 'ssl'):
raise util.Abort(_("can't use TLS: Python SSL support "
"not installed"))
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889 ui.note(_('(using tls)\n'))
s.ehlo()
s.starttls()
s.ehlo()
username = ui.config('smtp', 'username')
password = ui.config('smtp', 'password')
Arun Thomas
Patchbomb: Prompt password when using SMTP/TLS and no password in .hgrc....
r5749 if username and not password:
password = ui.getpass()
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889 if username and password:
ui.note(_('(authenticating to mail server as %s)\n') %
(username))
s.login(username, password)
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 def send(sender, recipients, msg):
try:
return s.sendmail(sender, recipients, msg)
except smtplib.SMTPRecipientsRefused, inst:
recipients = [r[1] for r in inst.recipients.values()]
raise util.Abort('\n' + '\n'.join(recipients))
except smtplib.SMTPException, inst:
raise util.Abort(inst)
Bryan O'Sullivan
Backed out changeset dc6ed2736c81
r5947
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 return send
Bryan O'Sullivan
Backed out changeset dc6ed2736c81
r5947
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 def _sendmail(ui, sender, recipients, msg):
'''send mail using sendmail.'''
program = ui.config('email', 'method')
Matt Mackall
templater: move email function to util
r5975 cmdline = '%s -f %s %s' % (program, util.email(sender),
' '.join(map(util.email, recipients)))
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 ui.note(_('sending mail: %s\n') % cmdline)
Dirkjan Ochtman
replace usage of os.popen() with util.popen()...
r6548 fp = util.popen(cmdline, 'w')
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 fp.write(msg)
ret = fp.close()
if ret:
raise util.Abort('%s %s' % (
os.path.basename(program.split(None, 1)[0]),
util.explain_exit(ret)[0]))
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889
def connect(ui):
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 '''make a mail connection. return a function to send mail.
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889 call as sendmail(sender, list-of-recipients, msg).'''
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 if ui.config('email', 'method', 'smtp') == 'smtp':
Bryan O'Sullivan
Backed out changeset dc6ed2736c81
r5947 return _smtp(ui)
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 return lambda s, r, m: _sendmail(ui, s, r, m)
Matt Mackall
Move ui.sendmail to mail.connect/sendmail
r2889
def sendmail(ui, sender, recipients, msg):
Matt Mackall
patchbomb: undo backout and fix bugs in the earlier patch
r5973 send = connect(ui)
return send(sender, recipients, msg)
Bryan O'Sullivan
patchbomb: Validate email config before we start prompting for info.
r4489
def validateconfig(ui):
'''determine if we have enough config data to try sending email.'''
method = ui.config('email', 'method', 'smtp')
if method == 'smtp':
if not ui.config('smtp', 'host'):
raise util.Abort(_('smtp specified as email transport, '
'but no smtp host configured'))
else:
if not util.find_exe(method):
raise util.Abort(_('%r specified as email transport, '
'but not in PATH') % method)
Christian Ebert
mail: add methods to handle non-ascii chars...
r7114
Christian Ebert
mail: mime-encode patches that are utf-8...
r7191 def mimetextpatch(s, subtype='plain', display=False):
'''If patch in utf-8 transfer-encode it.'''
if not display:
for cs in ('us-ascii', 'utf-8'):
try:
s.decode(cs)
return email.MIMEText.MIMEText(s, subtype, cs)
except UnicodeDecodeError:
pass
return email.MIMEText.MIMEText(s, subtype)
Christian Ebert
mail: add methods to handle non-ascii chars...
r7114 def _charsets(ui):
'''Obtains charsets to send mail parts not containing patches.'''
charsets = [cs.lower() for cs in ui.configlist('email', 'charsets')]
fallbacks = [util._fallbackencoding.lower(),
util._encoding.lower(), 'utf-8']
for cs in fallbacks: # util.unique does not keep order
if cs not in charsets:
charsets.append(cs)
return [cs for cs in charsets if not cs.endswith('ascii')]
def _encode(ui, s, charsets):
'''Returns (converted) string, charset tuple.
Finds out best charset by cycling through sendcharsets in descending
order. Tries both _encoding and _fallbackencoding for input. Only as
last resort send as is in fake ascii.
Caveat: Do not use for mail parts containing patches!'''
try:
s.decode('ascii')
except UnicodeDecodeError:
sendcharsets = charsets or _charsets(ui)
for ics in (util._encoding, util._fallbackencoding):
try:
u = s.decode(ics)
except UnicodeDecodeError:
continue
for ocs in sendcharsets:
try:
return u.encode(ocs), ocs
except UnicodeEncodeError:
pass
except LookupError:
Christian Ebert
mail: correct typo in variable name
r7195 ui.warn(_('ignoring invalid sendcharset: %s\n') % ocs)
Christian Ebert
mail: add methods to handle non-ascii chars...
r7114 # if ascii, or all conversion attempts fail, send (broken) ascii
return s, 'us-ascii'
def headencode(ui, s, charsets=None, display=False):
'''Returns RFC-2047 compliant header from given string.'''
if not display:
# split into words?
s, cs = _encode(ui, s, charsets)
return str(email.Header.Header(s, cs))
return s
def addressencode(ui, address, charsets=None, display=False):
'''Turns address into RFC-2047 compliant header.'''
if display or not address:
return address or ''
name, addr = email.Utils.parseaddr(address)
name = headencode(ui, name, charsets)
try:
acc, dom = addr.split('@')
acc = acc.encode('ascii')
dom = dom.encode('idna')
addr = '%s@%s' % (acc, dom)
except UnicodeDecodeError:
raise util.Abort(_('invalid email address: %s') % addr)
except ValueError:
try:
# too strict?
addr = addr.encode('ascii')
except UnicodeDecodeError:
raise util.Abort(_('invalid local address: %s') % addr)
return email.Utils.formataddr((name, addr))
def mimeencode(ui, s, charsets=None, display=False):
'''creates mime text object, encodes it if needed, and sets
charset and transfer-encoding accordingly.'''
cs = 'us-ascii'
if not display:
s, cs = _encode(ui, s, charsets)
return email.MIMEText.MIMEText(s, 'plain', cs)