Python中with的运行机制

一、读取文件数据

    读取一个文件的数据，需要获取文件句柄，然后读取数据，最后关闭文件。

example01.py是一个最简单的例子，但是这段代码有两个问题。一是可能忘记关闭文件句柄，二是文件读取数据发生异常时没有进行任何处理。

example02.py中加入了异常保护，但是代码比较冗长。

example03.py中使用了with...as... 的方法来实现，代码简洁、安全。

# example01.py
file = open("/tmp/foo.txt")
data = file.read()
file.cloae()

# example02.py
file = open("/tmp/foo.txt")
try:
    data = file.read()
finally:
    file.cloae()

# example03.py
with open("/tmp/foo.txt") as file:
    data = file.read()

二、with的运行机制

with statement：用上下文管理器(with statement context managers)定义的方法来封装块的执行.
with上下文管理器：
object.__enter__(self)    // 返回的值作为as的值
object.__exit__(self, exc_type, exc_value, traceback)  // 这些参数描述了context的异常退出。如果没有异常，三个参数都为None.

处理流程：
(1) 上下文表达式求值以获得一个上下文管理器(context manager)
(2) 加载context manager的__exit__()，以便后边使用
(3) 调用context manager的__enter__()
(4) 如果with表达式存在一个目标，把__enter__()的返回值给目标
(5) 执行内容
(6) 调用context manager的__exit__()，如果异常发生，与之相关联的参数exc_type, exc_value, traceback传给__exit__()方法。否则，参数赋值为None.
如果代码块由于异常退出，__exit__()返回false，异常发生；如果__exit__()返回的是true，异常就不存在，继续执行with的表达式。
在with后面的代码块抛出任何异常时，__exit__()方法被执行。开发库时，清理资源，关闭文件等操作都可以放在__exit__()方法中。

# with_example01.py
class Sample:
    ''' Sample'''
    def __enter__(self):
        print("In __enter__()")
        return "Foo"
    def __exit__(self, type, value, trace):
        print("In __exit__()")

def get_sample():
    return Sample()

with get_sample() as sample:
    print("sample:", sample)

with_example01.py 演示了正常的处理流程，结果如下

In __enter__()

sample: Foo

In __exit__()

# with_example02.py
class Sample:
    ''' Sample'''
    def __enter__(self):
        return self    # return a new Sample object
    def __exit__(self, type, value, trace):
        print("type:", type)
        print("value:", value)
        print("trace:", trace)
    def do_something(self):
        bar = 1/0
        return bar + 10

with Sample() as sample:
    sample.do_something()

with_example02.py 演示了异常情况，运行结果如下：

type: <class 'ZeroDivisionError'>

Traceback (most recent call last):

  File "D:\SelfStudy\PythonStudy\grammar\basic usage\with_example01.py", line 32, in <module>

    sample.do_something()

  File "D:\SelfStudy\PythonStudy\grammar\basic usage\with_example01.py", line 28, in do_something

    bar = 1/0

ZeroDivisionError: division by zero

value: division by zero

trace: <traceback object at 0x0065C1E8>