Python函数输出捕获：理解print与return及标准输出重定向

from GmailChecker import GmailChecker

def Checker():
    Email = 'example@gmail.com'
    Checker = GmailChecker.verify(Email, 1) # 这一行会打印输出，但Checker变量实际接收到的是None
    try:
        if 'Unregistered' in Checker: # 错误发生在这里
            print ('False')
        else:
            print ('Alive')
    except:
        print('Error occurred')

Checker()

import sys
import io
# 假设 GmailChecker 库已正确安装
# from GmailChecker import GmailChecker 

# 模拟 GmailChecker.verify 的行为
# 实际上，你需要安装并导入真实的 GmailChecker 库
class MockGmailChecker:
    @staticmethod
    def verify(email, option):
        # 模拟库的打印行为
        if email == 'unregistered@gmail.com':
            print(f"{email} Unregistered")
        elif email == 'alive@gmail.com':
            print(f"{email} Alive")
        else:
            print(f"{email} Unknown Status")
        return None # 模拟库返回 None

# 在实际应用中，请使用：
# from GmailChecker import GmailChecker
# 替换 MockGmailChecker
GmailChecker = MockGmailChecker # 仅为示例，实际使用时请删除此行

def capture_function_output(func, *args, **kwargs):
    """
    捕获指定函数在执行期间打印到标准输出的内容。

    Args:
        func (callable): 要执行并捕获其输出的函数。
        *args: 传递给 func 的位置参数。
        **kwargs: 传递给 func 的关键字参数。

    Returns:
        str: 函数执行期间捕获到的标准输出内容。
    """
    # 保存原始的sys.stdout
    old_stdout = sys.stdout
    # 创建一个StringIO对象来捕获输出
    redirected_output = io.StringIO()
    # 将sys.stdout重定向到StringIO对象
    sys.stdout = redirected_output

    try:
        # 执行目标函数
        func(*args, **kwargs)
    finally:
        # 恢复sys.stdout到其原始状态，确保后续打印正常
        # 这一步至关重要，无论函数执行是否出错，都必须恢复
        sys.stdout = old_stdout

    # 获取捕获到的字符串
    captured_string = redirected_output.getvalue()
    return captured_string

# 示例使用
email_to_check_unregistered = 'unregistered@gmail.com'
email_to_check_alive = 'alive@gmail.com'
email_to_check_unknown = 'unknown@gmail.com'

# 捕获未注册邮箱的输出
output_unregistered = capture_function_output(GmailChecker.verify, email_to_check_unregistered, 1)
print(f"捕获到的输出 (未注册):\n'{output_unregistered.strip()}'") # .strip()去除可能的空白符

if 'Unregistered' in output_unregistered:
    print(f"邮箱 {email_to_check_unregistered} 状态: 未注册")
elif 'Alive' in output_unregistered:
    print(f"邮箱 {email_to_check_unregistered} 状态: 活跃")
else:
    print(f"邮箱 {email_to_check_unregistered} 状态: 未知或错误")

print("-" * 30)

# 捕获活跃邮箱的输出
output_alive = capture_function_output(GmailChecker.verify, email_to_check_alive, 1)
print(f"捕获到的输出 (活跃):\n'{output_alive.strip()}'")

if 'Unregistered' in output_alive:
    print(f"邮箱 {email_to_check_alive} 状态: 未注册")
elif 'Alive' in output_alive:
    print(f"邮箱 {email_to_check_alive} 状态: 活跃")
else:
    print(f"邮箱 {email_to_check_alive} 状态: 未知或错误")

print("-" * 30)

# 捕获未知状态邮箱的输出
output_unknown = capture_function_output(GmailChecker.verify, email_to_check_unknown, 1)
print(f"捕获到的输出 (未知):\n'{output_unknown.strip()}'")

if 'Unregistered' in output_unknown:
    print(f"邮箱 {email_to_check_unknown} 状态: 未注册")
elif 'Alive' in output_unknown:
    print(f"邮箱 {email_to_check_unknown} 状态: 活跃")
else:
    print(f"邮箱 {email_to_check_unknown} 状态: 未知或错误")