Handlers¶
Handlers are classes which can implement hook methods that get
called at various points in the SMTP dialog. Handlers can also be named on
the command line, but if the class’s constructor takes arguments,
you must define a @classmethod that converts the positional arguments and
returns a handler instance:
from_cli(cls, parser, *args)- Convert the positional arguments, as strings passed in on the command
line, into a handler instance.
parseris the ArgumentParser instance in use.
If from_cli() is not defined, the handler can still be used on the command
line, but its constructor cannot accept arguments.
Handler hooks¶
Handlers can implement hooks that get called during the SMTP dialog, or in
exceptional cases. These handler hooks are all called asynchronously
(i.e. they are coroutines) and they must return a status string, such as
'250 OK'. All handler hooks are optional and default behaviors are
carried out by the SMTP class when a hook is omitted, so you only need to
implement the ones you care about. When a handler hook is defined, it may
have additional responsibilities as described below.
All handler hooks take at least three arguments, the SMTP server instance,
a session instance, and an envelope instance.
Some methods take additional arguments.
The following hooks are currently defined:
handle_HELO(server, session, envelope, hostname)- Called during
HELO. Thehostnameargument is the host name given by the client in theHELOcommand. If implemented, this hook must also set thesession.host_nameattribute before returning'250 {}'.format(server.hostname)as the status. handle_EHLO(server, session, envelope, hostname)- Called during
EHLO. Thehostnameargument is the host name given by the client in theEHLOcommand. If implemented, this hook must also set thesession.host_nameattribute. This hook may push additional250-<command>responses to the client by yielding fromserver.push(status)before returning250 HELPas the final response. handle_NOOP(server, session, envelope, arg)- Called during
NOOP. handle_QUIT(server, session, envelope)- Called during
QUIT. handle_VRFY(server, session, envelope, address)- Called during
VRFY. Theaddressargument is the parsed email address given by the client in theVRFYcommand. handle_MAIL(server, session, envelope, address, mail_options)- Called during
MAIL FROM. Theaddressargument is the parsed email address given by the client in theMAIL FROMcommand, andmail_optionsare any additional ESMTP mail options providing by the client. If implemented, this hook must also set theenvelope.mail_fromattribute and it may extendenvelope.mail_options(which is always a Python list). handle_RCPT(server, session, envelope, address, rcpt_options)- Called during
RCPT TO. Theaddressargument is the parsed email address given by the client in theRCPT TOcommand, andrcpt_optionsare any additional ESMTP recipient options providing by the client. If implemented, this hook should append the address toenvelope.rcpt_tosand may extendenvelope.rcpt_options(both of which are always Python lists). handle_RSET(server, session, envelope)- Called during
RSET. handle_DATA(server, session, envelope)- Called during
DATAafter the entire message (“SMTP content” as described in RFC 5321) has been received. The content is available on theenvelopeobject, but the values are dependent on whether theSMTPclass was instantiated withdecode_data=False(the default) ordecode_data=True. In the former case, bothenvelope.contentandenvelope.original_contentwill be the content bytes (normalized according to the transparency rules in RFC 5321, §4.5.2). In the latter case,envelope.original_contentwill be the normalized bytes, butenvelope.contentwill be the UTF-8 decoded string of the original content.
In addition to the SMTP command hooks, the following hooks can also be implemented by handlers. These have different APIs, and are called synchronously (i.e. they are not coroutines).
handle_STARTTLS(server, session, envelope)- If implemented, and if SSL is supported, this method gets called
during the TLS handshake phase of
connection_made(). It should return True if the handshake succeeded, and False otherwise. handle_exception(error)- If implemented, this method is called when any error occurs during the
handling of a connection (e.g. if an
smtp_<command>()method raises an exception). The exception object is passed in. This method must return a status string, such as'542 Internal server error'. If the method returns None or raises an exception, an exception will be logged, and a 500 code will be returned to the client.
Built-in handlers¶
The following built-in handlers can be imported from aiosmtpd.handlers:
Debugging- this class prints the contents of the received messages to a given output stream. Programmatically, you can pass the stream to print to into the constructor. When specified on the command line, the positional argument must either be the stringstdoutorstderrindicating which stream to use.Proxy- this class is a relatively simple SMTP proxy; it forwards messages to a remote host and port. The constructor takes the host name and port as positional arguments. This class cannot be used on the command line.Sink- this class just consumes and discards messages. It’s essentially the “no op” handler. It can be used on the command line, but accepts no positional arguments.Message- this class is a base class (it must be subclassed) which converts the message content into a message instance. The class used to create these instances can be passed to the constructor, and defaults to email.message.Message.This message instance gains a few additional headers (e.g.
X-Peer,X-MailFrom, andX-RcptTo). You can override this behavior by overriding theprepare_message()method, which takes a session and an envelope. The message instance is then passed to the handler’shandle_message()method. It is this method that must be implemented in the subclass.prepare_message()andhandle_message()are both called synchronously. This handler cannot be used on the command line.AsyncMessage- a subclass of theMessagehandler, with the only difference being thathandle_message()is called asynchronously. This handler cannot be used on the command line.Mailbox- a subclass of theMessagehandler which adds the messages to a Maildir. See below for details.
The Mailbox handler¶
A convenient handler is the Mailbox handler, which stores incoming
messages into a maildir:
>>> import os
>>> from aiosmtpd.controller import Controller
>>> from aiosmtpd.handlers import Mailbox
>>> from tempfile import TemporaryDirectory
>>> # Clean up the temporary directory at the end of this doctest.
>>> tempdir = resources.enter_context(TemporaryDirectory())
>>> maildir_path = os.path.join(tempdir, 'maildir')
>>> controller = Controller(Mailbox(maildir_path))
>>> controller.start()
>>> # Arrange for the controller to be stopped at the end of this doctest.
>>> ignore = resources.callback(controller.stop)
Now we can connect to the server and send it a message…
>>> from smtplib import SMTP
>>> client = SMTP(controller.hostname, controller.port)
>>> client.sendmail('aperson@example.com', ['bperson@example.com'], """\
... From: Anne Person <anne@example.com>
... To: Bart Person <bart@example.com>
... Subject: A test
... Message-ID: <ant>
...
... Hi Bart, this is Anne.
... """)
{}
…and a second message…
>>> client.sendmail('cperson@example.com', ['dperson@example.com'], """\
... From: Cate Person <cate@example.com>
... To: Dave Person <dave@example.com>
... Subject: A test
... Message-ID: <bee>
...
... Hi Dave, this is Cate.
... """)
{}
…and a third message.
>>> client.sendmail('eperson@example.com', ['fperson@example.com'], """\
... From: Elle Person <elle@example.com>
... To: Fred Person <fred@example.com>
... Subject: A test
... Message-ID: <cat>
...
... Hi Fred, this is Elle.
... """)
{}
We open up the mailbox again, and all three messages are waiting for us.
>>> from mailbox import Maildir
>>> from operator import itemgetter
>>> mailbox = Maildir(maildir_path)
>>> messages = sorted(mailbox, key=itemgetter('message-id'))
>>> for message in messages:
... print(message['Message-ID'], message['From'], message['To'])
<ant> Anne Person <anne@example.com> Bart Person <bart@example.com>
<bee> Cate Person <cate@example.com> Dave Person <dave@example.com>
<cat> Elle Person <elle@example.com> Fred Person <fred@example.com>