Question

以下功能允许使用with-statment和pop连接。但是如果没有建立连接，则finally中的quit（）将引发异常。如何解决这个问题？

@contextmanager
def pop_connect(server, user, password, timeout, use_SSL=False):
    try:
        pop = poplib.POP3_SSL if use_SSL else poplib.POP3
        pop_conn = pop(server, timeout=timeout)
        pop_conn.pass_(password)
        yield pop_conn
    except poplib.error_proto as pop_error:
        print('Authentication for receiving emails failed:{}'.format(pop_error))
    except OSError as os_error:
        print('Name resolution or connection failed:{}'.format(os_error))
    finally:
            pop_conn.quit()

Answer 1

我想您可以将pop_conn.quit()放在try: pass作为相应的except行动：

finally:
    try:
        pop_conn.quit()
    except <WhaterverException>:
        pass

Answer 2

解决方案是重新抛出处理程序中的异常。上下文管理者除了收益率之外没有：

@contextmanager
def pop_connect(server, user, password, timeout, use_SSL=False):
    try:
        pop_conn = poplib.POP3_SSL(server,timeout=timeout) if use_SSL else poplib.POP3_SSL(server,timeout=timeout)
        pop_conn.user(user)
        pop_conn.pass_(password)
        yield pop_conn
    except poplib.error_proto as pop_error:
        print('Receiving mail failed:{}'.format(pop_error))
        raise
    except OSError as os_error:
        print('Name resolution or connection failed:{}'.format(os_error))
        raise
    finally:
        try:
            pop_conn.quit()
        except UnboundLocalError:
            pass

Unestablished连接和内容管理器

2 个答案: