2. contextlib
    4. 阅读: 31618
    5. 在Python中，读写文件这样的资源要特别注意，必须在使用完毕后正确关闭它们。正确关闭文件资源的一个方法是使用try...finally：
    7. try:
    8.    f = open('/path/to/file', 'r')
    9.    f.read()
    10. finally:
    11.    if f:
    12.        f.close()
    13. 写try...finally非常繁琐。Python的with语句允许我们非常方便地使用资源，而不必担心资源没有关闭，所以上面的代码可以简化为：
    15. with open('/path/to/file', 'r') as f:
    16.    f.read()
    17. 并不是只有open()函数返回的fp对象才能使用with语句。实际上，任何对象，只要正确实现了上下文管理，就可以用于with语句。
    19. 实现上下文管理是通过__enter__和__exit__这两个方法实现的。例如，下面的class实现了这两个方法：
    21. class Query(object):
    23.    def __init__(self, name):
    24.        self.name = name
    26.    def __enter__(self):
    27.        print('Begin')
    28.        return self
    30.    def __exit__(self, exc_type, exc_value, traceback):
    31.        if exc_type:
    32.            print('Error')
    33.        else:
    34.            print('End')
    36.    def query(self):
    37.        print('Query info about %s...' % self.name)
    38. 这样我们就可以把自己写的资源对象用于with语句：
    40. with Query('Bob') as q:
    41.    q.query()
    42. @contextmanager
    44. 编写__enter__和__exit__仍然很繁琐，因此Python的标准库contextlib提供了更简单的写法，上面的代码可以改写如下：
    46. from contextlib import contextmanager
    48. class Query(object):
    50.    def __init__(self, name):
    51.        self.name = name
    53.    def query(self):
    54.        print('Query info about %s...' % self.name)
    56. @contextmanager
    57. def create_query(name):
    58.    print('Begin')
    59.    q = Query(name)
    60.    yield q
    61.    print('End')
    62. @contextmanager这个decorator接受一个generator，用yield语句把with ... as var把变量输出出去，然后，with语句就可以正常地工作了：
    64. with create_query('Bob') as q:
    65.    q.query()
    66. 很多时候，我们希望在某段代码执行前后自动执行特定代码，也可以用@contextmanager实现。例如：
    68. @contextmanager
    69. def tag(name):
    70.    print("<%s>" % name)
    71.    yield
    72.    print("</%s>" % name)
    74. with tag("h1"):
    75.    print("hello")
    76.    print("world")
    77. 上述代码执行结果为：
    79. <h1>
    80. hello
    81. world
    82. </h1>
    83. 代码的执行顺序是：
    85. with语句首先执行yield之前的语句，因此打印出<h1>；
    86. yield调用会执行with语句内部的所有语句，因此打印出hello和world；
    87. 最后执行yield之后的语句，打印出</h1>。
    88. 因此，@contextmanager让我们通过编写generator来简化上下文管理。
    90. @closing
    92. 如果一个对象没有实现上下文，我们就不能把它用于with语句。这个时候，可以用closing()来把该对象变为上下文对象。例如，用with语句使用urlopen()：
    94. from contextlib import closing
    95. from urllib.request import urlopen
    97. with closing(urlopen('https://www.python.org')) as page:
    98.    for line in page:
    99.        print(line)
    100. closing也是一个经过@contextmanager装饰的generator，这个generator编写起来其实非常简单：
    102. @contextmanager
    103. def closing(thing):
    104.    try:
    105.        yield thing
    106.    finally:
    107.        thing.close()
    108. 它的作用就是把任意对象变为上下文对象，并支持with语句。
    110. @contextlib还有一些其他decorator，便于我们编写更简洁的代码。