使用Twisted Python的SMTP模块清理资源

时间:2012-08-30 14:38:24

标签: python twisted smtpd

这与之前回答的问题有关:Logging SMTP connections with Twisted。我有一个数据库资源,我在ConsoleMessageDelivery的每个实例中创建,我需要确保在关闭套接字时清理它。我有一个名为DenyFactory的WrappingFactory,当套接字关闭时会调用DenyFactory.unregisterProtocol方法,但我无法(我可以弄清楚)如何访问在被销毁的ConsoleMessageDelivery实例中创建的资源。我在ConsoleMessageDelivery中尝试了一个 del ()方法,但从未调用过。在这种情况下清理资源的最佳方法是什么?

class ConsoleMessageDelivery:
    implements(smtp.IMessageDelivery)

    def receivedHeader(self, helo, origin, recipients):
        myHostname, clientIP = helo
        headerValue = "by %s from %s with ESMTP ; %s" % (myHostname, clientIP, smtp.rfc822date())
        # email.Header.Header used for automatic wrapping of long lines
        return "Received: %s" % Header(headerValue)

    def validateFrom(self, helo, origin):
        # All addresses are accepted
        return origin

    def validateTo(self, user):
        if user.dest.local == "console":
            return lambda: ConsoleMessage()
        raise smtp.SMTPBadRcpt(user)

class ConsoleMessage:
    implements(smtp.IMessage)

    def __init__(self):
        self.lines = []

    def lineReceived(self, line):
        self.lines.append(line)

    def eomReceived(self):
        return defer.succeed(None)

    def connectionLost(self):
        # There was an error, throw away the stored lines
        self.lines = None

class ConsoleSMTPFactory(smtp.SMTPFactory):
    protocol = smtp.ESMTP

    def __init__(self, *a, **kw):
        smtp.SMTPFactory.__init__(self, *a, **kw)
        self.delivery = ConsoleMessageDelivery()

    def buildProtocol(self, addr):
        p = smtp.SMTPFactory.buildProtocol(self, addr)
        p.delivery = self.delivery
        return p

class DenyFactory(WrappingFactory):

    def buildProtocol(self, clientAddress):
        if clientAddress.host == '1.3.3.7':
            # Reject it
            return None
        # Accept everything else
        return WrappingFactory.buildProtocol(self, clientAddress)

    def unregisterProtocol(self, p):
        print "Unregister called"

1 个答案:

答案 0 :(得分:3)

首先,不要使用__del__,特别是如果您想要清理一些资源。 __del__可防止参考周期中对象的垃圾回收。 (或者,切换到PyPy,它可以通过对循环中的对象集合强制执行任意排序来收集此类对象。)

接下来,考虑在邮件传递工厂中打开数据库连接(或启动连接池),并在所有邮件传递对象之间共享它。这样你就不需要清理连接,因为你将它们重新用于将来的消息,并且你没有为每条消息分配一个新的消息,所以没有泄漏。

最后,如果您确实需要任何每个事务对象,则可以在eomReceived对象的connectionLostIMessage实现中清除它们。一旦SMTP事务的DATA部分完成,将调用其中一种方法(因为所有数据都已收到或因为连接丢失)。请注意,因为SMTP支持在单个事务中向多个收件人传递邮件,所以即使只有一个IMessage对象,也可能有多个IMessageDelivery对象参与。因此,您可能希望将消息传递对象上成功validateTo调用的调用次数与消息对象上的eomReceived / connectionLost调用次数进行计数器匹配。当每次调用的次数相同时,交易就完成了。