使用上下文管理器处理与imaplib的连接

Question

我正在尝试使用python的imaplib连接到邮件服务器。 这是我的代码当前的样子：

@contextlib.contextmanager
def get_connection(username: str, password: str, host=GMAIL_IMAP_HOST, port=GMAIL_IMAP_PORT):
    '''
    ARGS:
        username: User username
        password: User Password
        host: IMAP server address
        post: IMAP port

    Connect to IMAP server and close connection
    '''

    conn = None

    try:
        conn = imaplib.IMAP4_SSL(host, port)
        conn.login(username, password)
        response, _ = conn.select('INBOX')

        if response != 'OK':
            raise MyException('An error occurred authenticating - {}'.format(response))

        yield conn
    finally:
        if conn is not None:
            conn.close()   # <- Check if mailbox is selected
            conn.logout()  # <- Check if logged in

如何检查用户是否已通过身份验证以及是否选择了邮箱？