Python上下文管理器with块及@contextmanager的使用

import sys


class LookingGlass:
    def __enter__(self):
        # 将原始的 sys.stdout.write 函数保存起来
        self.original_wirte = sys.stdout.write
        # 将 sys.stdout.write 函数替换为 reverse_write 方法
        sys.stdout.write = self.reverse_wirte
        # 返回一个字符串作为上下文管理器的上下文对象
        return 'QWERTYUIOP'

    def reverse_wirte(self, text):
        # 对传入的文本进行反转后输出
        return self.original_wirte(text[::-1])

    def __exit__(self, exc_type, exc_val, exc_tb):  # 如果一切正常，那么Python 调⽤__exit__ ⽅法时传⼊的参数是None, None, None
        # 在退出上下文管理器时，恢复原始的 sys.stdout.write 函数
        sys.stdout.write = self.original_wirte
        # 如果发生的异常是 ZeroDivisionError，输出相应提示信息
        if exc_type is ZeroDivisionError:
            print('Please DO NOT divide by zero!')
            return True  # 返回True，告诉解释器异常已处理


if __name__ == '__main__':
    with LookingGlass() as what:
        # print(obj)实质就是调用sys.stdout.write(obj+'\n')
        print("床前明月光")  # 光月明前床
        print("疑是地上霜")  # 霜上地是疑
        print(what)  # POIUYTREWQ
    print(what)  # QWERTYUIOP 处理上下文管理器，输出的文本正常，不再翻转

import sys
import contextlib


@contextlib.contextmanager
def looking_glass():
    original_text = sys.stdout.write

    def reverse_write(text):
        original_text(text[::-1])

    sys.stdout.write = reverse_write
    msg = ''
    # 这里处理异常是因为如果with块有异常抛出，Python会将其捕获，然后再looking_glass函数的yield表达式再次抛出
    # 但是那里又没有处理错误的代码，因此looking_glass函数将终止执行，永远无法恢复成原来的sys.stdout.write方法，导致系统处于无效状态。
    try:
        yield 'QWERTYUIOP'
    except ZeroDivisionError:
        msg = 'Please DO NOT divide by zero!'
    finally:
        sys.stdout.write = original_text
        if msg:
            print(msg)


if __name__ == '__main__':
    with looking_glass() as what:
        print("床前明月光")  # 光月明前床
        print("疑是地上霜")  # 霜上地是疑
        print(what)  # POIUYTREWQ
    print(what)  # QWERTYUIOP

@looking_glass()
def varse():
    print("举头望明月低头思故乡")

if __name__ == '__main__':
    varse() # 乡故思头低月明望头举