第15章 上下文管理器和else块#

1
def get_result():
2
    a = 3
3
    b = 0
4
    try:
5
        b = 3 / 0
6
    except Exception as e:
7
        print("ERROR=", e)
8
    finally:
9
        print("a=", a)
10

11

12
if __name__ == "__main__":
13
    get_result()

1
ERROR= division by zero
2
a= 3

1
# 第一种方法：
2
file1 = open("test.txt")
3
file2 = open("output.txt", "w")
4
while True:
5
    line = file1.readline()
6
    # 这里可以进行逻辑处理
7
    file2.write(line)
8
    if not line:
9
        break
10

11
# 记住文件处理完，关闭是个好习惯
12
file1.close()
13
file2.close()
14

15
# 读文件有3种方法：
16
# - read()将文本文件所有行读到一个字符串中。
17
# - readline()是一行一行的读
18
# - readlines()是将文本文件中所有行读到一个list中，文本文件每一行是list的一个元素。
19

20
# 优点：readline()可以在读行过程中跳过特定行。
21

22
# 第二种方法：文件迭代器，用for循环的方法
23
file2 = open("output.txt", "w")
24
for line in open("test.txt"):
25
    # 这里可以进行逻辑处理
26
    file2.write(line)
27

28
# 第三种方法： 推荐使用这个方法文件上下文管理器
29
with open('somefile.txt', 'r') as f:
30
    data = f.read()
31

32
with open('somefile.txt', 'r') as f:
33
    for line in f:
34
        print(line)
35

36
with open('somefile.txt', 'w') as f:
37
    f.write("hello")

1
class T(object):
2
    def __enter__(self):
3
        print('T.__enter__')
4
        return '我是__enter__的返回值'
5

6
    def __exit__(self, exc_type, exc_val, exc_tb):
7
        print('T.__exit__')
8

9

10
with T() as t:
11
    print(t)

1
T.__enter__
2
我是__enter__的返回值
3
T.__exit__

1
1. yield 语句前面的所有代码在 with 块开始时（即解释器调用` __enter__` 方法时）执行
2
2. yield 语句后面的代码在with 块结束时（即调用 `__exit__` 方法时）执行。

1
import sys
2
import contextlib
3

4

5
@contextlib.contextmanager
6
def WoHa(n):
7
    original_write = sys.stdout.write
8

9
    def reverse_write(text):
10
        original_write(text[::-1])
11

12
    sys.stdout.write = reverse_write
13
    yield n
14
    sys.stdout.write = original_write
15
    return True
16

17

18
obj1 = WoHa('你手机拿反了')
19
with obj1 as content:
20
    print('哈哈镜花缘')
21
    print(content)
22

23
print('#### with 执行完毕后，在输出content: ####')
24
print(content)

1
缘花镜哈哈
2
了反拿机手你
3
#### with 执行完毕后，在输出content: ####
4
你手机拿反了