Python中with的作用和使用解读

f = open('my_file.txt', 'w')
f.write('hello world')
# 如果在 write 和 close 之间发生错误，close() 将永远不会被执行！
f.close()

f = None # 在 try 外面初始化，确保 finally 中可以访问
try:
    f = open('my_file.txt', 'w')
    f.write('hello world')
    # ... 其他可能出错的操作 ...
finally:
    if f:
        f.close()

with expression as variable:
    # 在这个代码块中，资源是可用的
    # ... do something with variable ...

# 离开 with 代码块后，资源会自动被清理

with open('my_file.txt', 'w') as f:
    f.write('hello world')
    # 在这里可以进行各种文件操作
    # 比如 f.read(), f.writelines() 等

# 当代码执行离开这个 with 块时（无论是正常结束还是发生异常），
# Python 会自动调用 f.close()，我们完全不需要操心。

# 1. 创建上下文管理器对象
manager = open('my_file.txt', 'w')

# 2. 调用 __enter__ 方法，返回值赋给 f
f = manager.__enter__()

# 3. 执行 with 块中的代码
try:
    f.write('hello world')
finally:
    # 4. 无论如何，都调用 __exit__ 方法进行清理
    # (这里简单展示，实际会传递异常信息)
    manager.__exit__(None, None, None)

import time

class Timer:
    def __init__(self, name):
        self.name = name

    def __enter__(self):
        print(f"计时器 '{self.name}' 开始...")
        self.start_time = time.time()
        # 这个类本身就是资源，所以返回 self
        return self 

    def __exit__(self, exc_type, exc_value, traceback):
        self.end_time = time.time()
        duration = self.end_time - self.start_time
        print(f"计时器 '{self.name}' 结束，耗时: {duration:.4f} 秒")
        # 如果有异常，这里可以记录日志
        if exc_type:
            print(f"在 '{self.name}' 中发生了异常: {exc_value}")
        # 返回 False 或 None，让异常正常抛出
        return False

# 使用自定义的 Timer
with Timer("数据处理") as t:
    print("正在处理数据...")
    time.sleep(2)
    print("数据处理完成。")

print("-" * 20)

with Timer("有问题的操作") as t:
    print("准备执行一个会出错的操作...")
    time.sleep(1)
    result = 1 / 0  # 这里会产生一个 ZeroDivisionError
    print("这行代码不会被执行")

计时器 '数据处理' 开始...
正在处理数据...
数据处理完成。
计时器 '数据处理' 结束，耗时: 2.0021 秒
--------------------
计时器 '有问题的操作' 开始...
准备执行一个会出错的操作...
计时器 '有问题的操作' 结束，耗时: 1.0011 秒
在 '有问题的操作' 中发生了异常: division by zero
Traceback (most recent call last):
  File "...", line 36, in <module>
    result = 1 / 0  # 这里会产生一个 ZeroDivisionError
ZeroDivisionError: division by zero

import time
from contextlib import contextmanager

@contextmanager
def timer(name):
    print(f"计时器 '{name}' 开始...")
    start_time = time.time()
    
    # yield 之前的部分，相当于 __enter__
    # yield 的值会成为 as 后面的变量（如果没有 yield 值，则为 None）
    try:
        yield
    finally:
        # yield 之后的部分，相当于 __exit__
        end_time = time.time()
        duration = end_time - start_time
        print(f"计时器 '{name}' 结束，耗时: {duration:.4f} 秒")

# 使用方法完全一样
with timer("数据处理_v2"):
    print("正在处理数据...")
    time.sleep(2)
    print("数据处理完成。")