| #! /usr/bin/env python |
| |
| '''SMTP/ESMTP client class. |
| |
| Author: The Dragon De Monsyne <dragondm@integral.org> |
| ESMTP support, test code and doc fixes added by |
| Eric S. Raymond <esr@thyrsus.com> |
| Better RFC 821 compliance (MAIL and RCPT, and CRLF in data) |
| by Carey Evans <c.evans@clear.net.nz>, for picky mail servers. |
| |
| This was modified from the Python 1.5 library HTTP lib. |
| |
| This should follow RFC 821 (SMTP) and RFC 1869 (ESMTP). |
| |
| Notes: |
| |
| Please remember, when doing ESMTP, that the names of the SMTP service |
| extensions are NOT the same thing as the option keywords for the RCPT |
| and MAIL commands! |
| |
| Example: |
| |
| >>> import smtplib |
| >>> s=smtplib.SMTP("localhost") |
| >>> print s.help() |
| This is Sendmail version 8.8.4 |
| Topics: |
| HELO EHLO MAIL RCPT DATA |
| RSET NOOP QUIT HELP VRFY |
| EXPN VERB ETRN DSN |
| For more info use "HELP <topic>". |
| To report bugs in the implementation send email to |
| sendmail-bugs@sendmail.org. |
| For local information send email to Postmaster at your site. |
| End of HELP info |
| >>> s.putcmd("vrfy","someone@here") |
| >>> s.getreply() |
| (250, "Somebody OverHere <somebody@here.my.org>") |
| >>> s.quit() |
| ''' |
| |
| import socket |
| import string |
| import re |
| import rfc822 |
| import types |
| |
| SMTP_PORT = 25 |
| CRLF="\r\n" |
| |
| # used for exceptions |
| class SMTPException(Exception): pass |
| class SMTPServerDisconnected(SMTPException): pass |
| class SMTPSenderRefused(SMTPException): pass |
| class SMTPRecipientsRefused(SMTPException): pass |
| class SMTPDataError(SMTPException): pass |
| |
| def quoteaddr(addr): |
| """Quote a subset of the email addresses defined by RFC 821. |
| |
| Should be able to handle anything rfc822.parseaddr can handle. |
| """ |
| m=None |
| try: |
| m=rfc822.parseaddr(addr)[1] |
| except AttributeError: |
| pass |
| if not m: |
| #something weird here.. punt -ddm |
| return addr |
| else: |
| return "<%s>" % m |
| |
| def quotedata(data): |
| """Quote data for email. |
| |
| Double leading '.', and change Unix newline '\n', or Mac '\r' into |
| Internet CRLF end-of-line. |
| """ |
| return re.sub(r'(?m)^\.', '..', |
| re.sub(r'(?:\r\n|\n|\r(?!\n))', CRLF, data)) |
| |
| class SMTP: |
| """This class manages a connection to an SMTP or ESMTP server. |
| SMTP Objects: |
| SMTP objects have the following attributes: |
| helo_resp |
| This is the message given by the server in responce to the |
| most recent HELO command. |
| |
| ehlo_resp |
| This is the message given by the server in responce to the |
| most recent EHLO command. This is usually multiline. |
| |
| does_esmtp |
| This is a True value _after you do an EHLO command_, if the |
| server supports ESMTP. |
| |
| esmtp_features |
| This is a dictionary, which, if the server supports ESMTP, |
| will _after you do an EHLO command_, contain the names of the |
| SMTP service extentions this server supports, and their |
| parameters (if any). |
| Note, all extention names are mapped to lower case in the |
| dictionary. |
| |
| For method docs, see each method's docstrings. In general, there is |
| a method of the same name to perform each SMTP command, and there |
| is a method called 'sendmail' that will do an entire mail |
| transaction. |
| """ |
| debuglevel = 0 |
| file = None |
| helo_resp = None |
| ehlo_resp = None |
| does_esmtp = 0 |
| |
| def __init__(self, host = '', port = 0): |
| """Initialize a new instance. |
| |
| If specified, `host' is the name of the remote host to which to |
| connect. If specified, `port' specifies the port to which to connect. |
| By default, smtplib.SMTP_PORT is used. |
| |
| """ |
| self.esmtp_features = {} |
| if host: self.connect(host, port) |
| |
| def set_debuglevel(self, debuglevel): |
| """Set the debug output level. |
| |
| A non-false value results in debug messages for connection and for all |
| messages sent to and received from the server. |
| |
| """ |
| self.debuglevel = debuglevel |
| |
| def connect(self, host='localhost', port = 0): |
| """Connect to a host on a given port. |
| |
| If the hostname ends with a colon (`:') followed by a number, and |
| there is no port specified, that suffix will be stripped off and the |
| number interpreted as the port number to use. |
| |
| Note: This method is automatically invoked by __init__, if a host is |
| specified during instantiation. |
| |
| """ |
| if not port: |
| i = string.find(host, ':') |
| if i >= 0: |
| host, port = host[:i], host[i+1:] |
| try: port = string.atoi(port) |
| except string.atoi_error: |
| raise socket.error, "nonnumeric port" |
| if not port: port = SMTP_PORT |
| self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| if self.debuglevel > 0: print 'connect:', (host, port) |
| self.sock.connect(host, port) |
| (code,msg)=self.getreply() |
| if self.debuglevel >0 : print "connect:", msg |
| return msg |
| |
| def send(self, str): |
| """Send `str' to the server.""" |
| if self.debuglevel > 0: print 'send:', `str` |
| if self.sock: |
| try: |
| self.sock.send(str) |
| except socket.error: |
| raise SMTPServerDisconnected('Server not connected') |
| else: |
| raise SMTPServerDisconnected('please run connect() first') |
| |
| def putcmd(self, cmd, args=""): |
| """Send a command to the server.""" |
| str = '%s %s%s' % (cmd, args, CRLF) |
| self.send(str) |
| |
| def getreply(self): |
| """Get a reply from the server. |
| |
| Returns a tuple consisting of: |
| |
| - server response code (e.g. '250', or such, if all goes well) |
| Note: returns -1 if it can't read response code. |
| |
| - server response string corresponding to response code (multiline |
| responses are converted to a single, multiline string). |
| """ |
| resp=[] |
| self.file = self.sock.makefile('rb') |
| while 1: |
| line = self.file.readline() |
| if self.debuglevel > 0: print 'reply:', `line` |
| resp.append(string.strip(line[4:])) |
| code=line[:3] |
| #check if multiline resp |
| if line[3:4]!="-": |
| break |
| try: |
| errcode = string.atoi(code) |
| except(ValueError): |
| errcode = -1 |
| |
| errmsg = string.join(resp,"\n") |
| if self.debuglevel > 0: |
| print 'reply: retcode (%s); Msg: %s' % (errcode,errmsg) |
| return errcode, errmsg |
| |
| def docmd(self, cmd, args=""): |
| """Send a command, and return its response code.""" |
| self.putcmd(cmd,args) |
| (code,msg)=self.getreply() |
| return code |
| |
| # std smtp commands |
| def helo(self, name=''): |
| """SMTP 'helo' command. |
| Hostname to send for this command defaults to the FQDN of the local |
| host. |
| """ |
| name=string.strip(name) |
| if len(name)==0: |
| name=socket.gethostbyaddr(socket.gethostname())[0] |
| self.putcmd("helo",name) |
| (code,msg)=self.getreply() |
| self.helo_resp=msg |
| return code |
| |
| def ehlo(self, name=''): |
| """ SMTP 'ehlo' command. |
| Hostname to send for this command defaults to the FQDN of the local |
| host. |
| """ |
| name=string.strip(name) |
| if len(name)==0: |
| name=socket.gethostbyaddr(socket.gethostname())[0] |
| self.putcmd("ehlo",name) |
| (code,msg)=self.getreply() |
| # According to RFC1869 some (badly written) |
| # MTA's will disconnect on an ehlo. Toss an exception if |
| # that happens -ddm |
| if code == -1 and len(msg) == 0: |
| raise SMTPServerDisconnected("Server not connected") |
| self.ehlo_resp=msg |
| if code<>250: |
| return code |
| self.does_esmtp=1 |
| #parse the ehlo responce -ddm |
| resp=string.split(self.ehlo_resp,'\n') |
| del resp[0] |
| for each in resp: |
| m=re.match(r'(?P<feature>[A-Za-z0-9][A-Za-z0-9\-]*)',each) |
| if m: |
| feature=string.lower(m.group("feature")) |
| params=string.strip(m.string[m.end("feature"):]) |
| self.esmtp_features[feature]=params |
| return code |
| |
| def has_extn(self, opt): |
| """Does the server support a given SMTP service extension?""" |
| return self.esmtp_features.has_key(string.lower(opt)) |
| |
| def help(self, args=''): |
| """SMTP 'help' command. |
| Returns help text from server.""" |
| self.putcmd("help", args) |
| (code,msg)=self.getreply() |
| return msg |
| |
| def rset(self): |
| """SMTP 'rset' command -- resets session.""" |
| code=self.docmd("rset") |
| return code |
| |
| def noop(self): |
| """SMTP 'noop' command -- doesn't do anything :>""" |
| code=self.docmd("noop") |
| return code |
| |
| def mail(self,sender,options=[]): |
| """SMTP 'mail' command -- begins mail xfer session.""" |
| optionlist = '' |
| if options and self.does_esmtp: |
| optionlist = string.join(options, ' ') |
| self.putcmd("mail", "FROM:%s %s" % (quoteaddr(sender) ,optionlist)) |
| return self.getreply() |
| |
| def rcpt(self,recip,options=[]): |
| """SMTP 'rcpt' command -- indicates 1 recipient for this mail.""" |
| optionlist = '' |
| if options and self.does_esmtp: |
| optionlist = ' ' + string.join(options, ' ') |
| self.putcmd("rcpt","TO:%s%s" % (quoteaddr(recip),optionlist)) |
| return self.getreply() |
| |
| def data(self,msg): |
| """SMTP 'DATA' command -- sends message data to server. |
| Automatically quotes lines beginning with a period per rfc821. |
| """ |
| self.putcmd("data") |
| (code,repl)=self.getreply() |
| if self.debuglevel >0 : print "data:", (code,repl) |
| if code <> 354: |
| return -1 |
| else: |
| self.send(quotedata(msg)) |
| self.send("%s.%s" % (CRLF, CRLF)) |
| (code,msg)=self.getreply() |
| if self.debuglevel >0 : print "data:", (code,msg) |
| return code |
| |
| def verify(self, address): |
| """SMTP 'verify' command -- checks for address validity.""" |
| self.putcmd("vrfy", quoteaddr(address)) |
| return self.getreply() |
| # a.k.a. |
| vrfy=verify |
| |
| def expn(self, address): |
| """SMTP 'verify' command -- checks for address validity.""" |
| self.putcmd("expn", quoteaddr(address)) |
| return self.getreply() |
| |
| # some useful methods |
| def sendmail(self, from_addr, to_addrs, msg, mail_options=[], |
| rcpt_options=[]): |
| """This command performs an entire mail transaction. |
| |
| The arguments are: |
| - from_addr : The address sending this mail. |
| - to_addrs : A list of addresses to send this mail to. A bare |
| string will be treated as a list with 1 address. |
| - msg : The message to send. |
| - mail_options : List of ESMTP options (such as 8bitmime) for the |
| mail command. |
| - rcpt_options : List of ESMTP options (such as DSN commands) for |
| all the rcpt commands. |
| |
| If there has been no previous EHLO or HELO command this session, this |
| method tries ESMTP EHLO first. If the server does ESMTP, message size |
| and each of the specified options will be passed to it. If EHLO |
| fails, HELO will be tried and ESMTP options suppressed. |
| |
| This method will return normally if the mail is accepted for at least |
| one recipient. Otherwise it will throw an exception (either |
| SMTPSenderRefused, SMTPRecipientsRefused, or SMTPDataError) That is, |
| if this method does not throw an exception, then someone should get |
| your mail. If this method does not throw an exception, it returns a |
| dictionary, with one entry for each recipient that was refused. |
| |
| Example: |
| |
| >>> import smtplib |
| >>> s=smtplib.SMTP("localhost") |
| >>> tolist=["one@one.org","two@two.org","three@three.org","four@four.org"] |
| >>> msg = ''' |
| ... From: Me@my.org |
| ... Subject: testin'... |
| ... |
| ... This is a test ''' |
| >>> s.sendmail("me@my.org",tolist,msg) |
| { "three@three.org" : ( 550 ,"User unknown" ) } |
| >>> s.quit() |
| |
| In the above example, the message was accepted for delivery to three |
| of the four addresses, and one was rejected, with the error code |
| 550. If all addresses are accepted, then the method will return an |
| empty dictionary. |
| |
| """ |
| if not self.helo_resp and not self.ehlo_resp: |
| if self.ehlo() >= 400: |
| self.helo() |
| esmtp_opts = [] |
| if self.does_esmtp: |
| # Hmmm? what's this? -ddm |
| # self.esmtp_features['7bit']="" |
| if self.has_extn('size'): |
| esmtp_opts.append("size=" + `len(msg)`) |
| for option in mail_options: |
| esmtp_opts.append(option) |
| |
| (code,resp) = self.mail(from_addr, esmtp_opts) |
| if code <> 250: |
| self.rset() |
| raise SMTPSenderRefused('%s: %s' % (from_addr, resp)) |
| senderrs={} |
| if type(to_addrs) == types.StringType: |
| to_addrs = [to_addrs] |
| for each in to_addrs: |
| (code,resp)=self.rcpt(each, rcpt_options) |
| if (code <> 250) and (code <> 251): |
| senderrs[each]=(code,resp) |
| if len(senderrs)==len(to_addrs): |
| # the server refused all our recipients |
| self.rset() |
| raise SMTPRecipientsRefused(string.join( |
| map(lambda x:"%s: %s" % (x[0], x[1][1]), senderrs.items()), |
| '; ')) |
| code=self.data(msg) |
| if code <>250 : |
| self.rset() |
| raise SMTPDataError('data transmission error: %s' % code) |
| #if we got here then somebody got our mail |
| return senderrs |
| |
| |
| def close(self): |
| """Close the connection to the SMTP server.""" |
| if self.file: |
| self.file.close() |
| self.file = None |
| if self.sock: |
| self.sock.close() |
| self.sock = None |
| |
| |
| def quit(self): |
| """Terminate the SMTP session.""" |
| self.docmd("quit") |
| self.close() |
| |
| |
| # Test the sendmail method, which tests most of the others. |
| # Note: This always sends to localhost. |
| if __name__ == '__main__': |
| import sys, rfc822 |
| |
| def prompt(prompt): |
| sys.stdout.write(prompt + ": ") |
| return string.strip(sys.stdin.readline()) |
| |
| fromaddr = prompt("From") |
| toaddrs = string.splitfields(prompt("To"), ',') |
| print "Enter message, end with ^D:" |
| msg = '' |
| while 1: |
| line = sys.stdin.readline() |
| if not line: |
| break |
| msg = msg + line |
| print "Message length is " + `len(msg)` |
| |
| server = SMTP('localhost') |
| server.set_debuglevel(1) |
| server.sendmail(fromaddr, toaddrs, msg) |
| server.quit() |