1. #!/usr/bin/env python
    2. # -*- coding: utf-8 -*-
    4. # 你想自己去实现一个新的上下文管理器，以便使用with语句。
    6. # 实现一个新的上下文管理器的最简单的方法就是使用 contexlib 模块中的 @contextmanager 装饰器。 下面是一个实现了代码块计时功能的上下文管理器例子：
    8. import time
    9. from contextlib import contextmanager
    12. @contextmanager
    13. def timethis(label):
    14.     start = time.time()
    15.     try:
    16.         yield
    17.     finally:
    18.         end = time.time()
    19.         print('{}: {}'.format(label, end - start))
    22. with timethis('counting'):
    23.     n = 10000000
    24.     while n > 0:
    25.         n -= 1
    28. # counting: 1.2312605381011963
    30. # 在函数 timethis() 中，yield 之前的代码会在上下文管理器中作为 __enter__() 方法执行， 所有在 yield 之后的代码会作为 __exit__() 方法执行。 如果出现了异常，异常会在yield语句那里抛出。
    32. # 下面是一个更加高级一点的上下文管理器，实现了列表对象上的某种事务：
    34. @contextmanager
    35. def list_transaction(orig_list):
    36.     working = list(orig_list)
    37.     yield working
    38.     orig_list[:] = working
    41. items = [1, 2, 3]
    42. with list_transaction(items) as working:
    43.     working.append(4)
    44.     working.append(5)
    46. print(items)  # [1, 2, 3, 4, 5]
    48. with list_transaction(items) as working:
    49.     working.append(6)
    50.     working.append(7)
    51.     raise RuntimeError('oops')
    53. # items
    54. # [1, 2, 3, 4, 5]
    1. # 通常情况下，如果要写一个上下文管理器，你需要定义一个类，里面包含一个 __enter__() 和一个 __exit__() 方法，如下所示：
    3. import time
    6. class timethis:
    7.     def __init__(self, label):
    8.         self.label = label
    10.     def __enter__(self):
    11.         self.start = time.time()
    13.     def __exit__(self, exc_type, exc_val, exc_tb):
    14.         end = time.time()
    15.         print('{}: {}'.format(self.label, end - self.start))
    18. with timethis('耗时') as t:
    19.     n = 10000000
    20.     while n > 0:
    21.         n -= 1
    22. # 耗时: 1.266099452972412
    24. # @contextmanager 应该仅仅用来写自包含的上下文管理函数。 如果你有一些对象(比如一个文件、网络连接或锁)，需要支持 with 语句，那么你就需要单独实现 __enter__() 方法和 __exit__() 方法。