如何使用上下文管理器例外

Question

我需要从电子邮件中读取文件附件。 我想使用上下文管理器进行资源管理（连接应该关闭）

@contextmanager
def some_f():
    conn = None
    try:
        for _ in xrange(CONNECTION_ERRORS_MAX):
            try:
                conn = POP3(self.pop3_host)
                conn.user(self.pop3_user)
                conn.pass_(self.pop3_password)
                logger.debug('connected {}', conn.getwelcome())
                _, msg_ids, _ = conn.uidl()
                yield (
                    (msg_id, self.files_from_msg(conn, file_batch_id, msg_id))
                    for msg_id, file_batch_id in
                    (ul.split(' ') for ul in msg_ids)
                    if not self.is_ignored(file_batch_id, msg_id)
                )
            except socket.error as e:
                if e.errno == errno.ECONNREFUSED:
                    pass
                else:
                    raise e
    finally:
        if conn:
            conn.quit()

因此，此功能应该执行的操作是尝试连接到电子邮件服务器，如果发生另一个套接字错误，则如果拒绝连接，则重试n次，然后重新发送。 如果重试尝试已用尽，则重新引发连接拒绝错误。 使用者使用完所有文件后将关闭连接。 现在，它使生成器没有产生错误。

Answer 1

重组try-except子句，使它包围最少的事物。 在except子句中，必要时可以continue ，如果未触发，则应从for循环中break 。