patchbomb: prompt only once for SMTP password
authorMatt Mackall <mpm@selenic.com>
Thu, 17 Jan 2008 13:51:59 -0600
changeset 5866 dc6ed2736c81
parent 5865 e7127f669edb
child 5867 b6d8972ce339
child 5947 528c986f0162
patchbomb: prompt only once for SMTP password - simplify mail._sendmail to be a function rather than a class - simplify connect to return a function rather than a class - move exception handling from mail.sendmail to mail.connect - use a single connection for all messages in patchbomb
hgext/patchbomb.py
mercurial/mail.py
--- a/hgext/patchbomb.py	Wed Jan 16 11:32:25 2008 -0600
+++ b/hgext/patchbomb.py	Thu Jan 17 13:51:59 2008 -0600
@@ -381,6 +381,7 @@
     parent = None
 
     sender_addr = email.Utils.parseaddr(sender)[1]
+    sendmail = None
     for m in msgs:
         try:
             m['Message-Id'] = genmsgid(m['X-Mercurial-Node'])
@@ -426,10 +427,12 @@
             fp.write('\n\n')
             fp.close()
         else:
+            if not sendmail:
+                sendmail = mail.connect(ui)
             ui.status('Sending ', m['Subject'], ' ...\n')
             # Exim does not remove the Bcc field
             del m['Bcc']
-            mail.sendmail(ui, sender, to + bcc + cc, m.as_string(0))
+            sendmail(ui, sender, to + bcc + cc, m.as_string(0))
 
 cmdtable = {
     "email":
--- a/mercurial/mail.py	Wed Jan 16 11:32:25 2008 -0600
+++ b/mercurial/mail.py	Thu Jan 17 13:51:59 2008 -0600
@@ -38,39 +38,43 @@
         s.login(username, password)
     return s
 
-class _sendmail(object):
+def _sendmail(ui, sender, recipients, msg):
     '''send mail using sendmail.'''
-
-    def __init__(self, ui, program):
-        self.ui = ui
-        self.program = program
-
-    def sendmail(self, sender, recipients, msg):
-        cmdline = '%s -f %s %s' % (
-            self.program, templater.email(sender),
-            ' '.join(map(templater.email, recipients)))
-        self.ui.note(_('sending mail: %s\n') % cmdline)
-        fp = os.popen(cmdline, 'w')
-        fp.write(msg)
-        ret = fp.close()
-        if ret:
-            raise util.Abort('%s %s' % (
-                os.path.basename(self.program.split(None, 1)[0]),
-                util.explain_exit(ret)[0]))
+    program = ui.config('email', 'method')
+    cmdline = '%s -f %s %s' % (program, templater.email(sender),
+                               ' '.join(map(templater.email, recipients)))
+    ui.note(_('sending mail: %s\n') % cmdline)
+    fp = os.popen(cmdline, 'w')
+    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]))
 
 def connect(ui):
-    '''make a mail connection. object returned has one method, sendmail.
+    '''make a mail connection. return a function to send mail.
     call as sendmail(sender, list-of-recipients, msg).'''
 
-    method = ui.config('email', 'method', 'smtp')
-    if method == 'smtp':
-        return _smtp(ui)
+    func =  _sendmail
+    if ui.config('email', 'method', 'smtp') == 'smtp':
+        func = _smtp(ui)
 
-    return _sendmail(ui, method)
+    def send(ui, sender, recipients, msg):
+        try:
+            return func.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)
+
+    return send
 
 def sendmail(ui, sender, recipients, msg):
     try:
-        return connect(ui).sendmail(sender, recipients, msg)
+        send = connect(ui)
+        return send(sender, recipients, msg)
     except smtplib.SMTPRecipientsRefused, inst:
         recipients = [r[1] for r in inst.recipients.values()]
         raise util.Abort('\n' + '\n'.join(recipients))