f = open('my_file.txt', 'w')
f.write('hello world')
# 如果在 write 和 close 之間發(fā)生錯(cuò)誤，close() 將永遠(yuǎn)不會(huì)被執(zhí)行！
f.close()

f = None # 在 try 外面初始化，確保 finally 中可以訪問
try:
    f = open('my_file.txt', 'w')
    f.write('hello world')
    # ... 其他可能出錯(cuò)的操作 ...
finally:
    if f:
        f.close()

with expression as variable:
    # 在這個(gè)代碼塊中，資源是可用的
    # ... do something with variable ...

# 離開 with 代碼塊后，資源會(huì)自動(dòng)被清理

with open('my_file.txt', 'w') as f:
    f.write('hello world')
    # 在這里可以進(jìn)行各種文件操作
    # 比如 f.read(), f.writelines() 等

# 當(dāng)代碼執(zhí)行離開這個(gè) with 塊時(shí)（無論是正常結(jié)束還是發(fā)生異常），
# Python 會(huì)自動(dòng)調(diào)用 f.close()，我們完全不需要操心。

# 1. 創(chuàng)建上下文管理器對(duì)象
manager = open('my_file.txt', 'w')

# 2. 調(diào)用 __enter__ 方法，返回值賦給 f
f = manager.__enter__()

# 3. 執(zhí)行 with 塊中的代碼
try:
    f.write('hello world')
finally:
    # 4. 無論如何，都調(diào)用 __exit__ 方法進(jìn)行清理
    # (這里簡單展示，實(shí)際會(huì)傳遞異常信息)
    manager.__exit__(None, None, None)

import time

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

    def __enter__(self):
        print(f"計(jì)時(shí)器 '{self.name}' 開始...")
        self.start_time = time.time()
        # 這個(gè)類本身就是資源，所以返回 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"計(jì)時(shí)器 '{self.name}' 結(jié)束，耗時(shí): {duration:.4f} 秒")
        # 如果有異常，這里可以記錄日志
        if exc_type:
            print(f"在 '{self.name}' 中發(fā)生了異常: {exc_value}")
        # 返回 False 或 None，讓異常正常拋出
        return False

# 使用自定義的 Timer
with Timer("數(shù)據(jù)處理") as t:
    print("正在處理數(shù)據(jù)...")
    time.sleep(2)
    print("數(shù)據(jù)處理完成。")

print("-" * 20)

with Timer("有問題的操作") as t:
    print("準(zhǔn)備執(zhí)行一個(gè)會(huì)出錯(cuò)的操作...")
    time.sleep(1)
    result = 1 / 0  # 這里會(huì)產(chǎn)生一個(gè) ZeroDivisionError
    print("這行代碼不會(huì)被執(zhí)行")

計(jì)時(shí)器 '數(shù)據(jù)處理' 開始...
正在處理數(shù)據(jù)...
數(shù)據(jù)處理完成。
計(jì)時(shí)器 '數(shù)據(jù)處理' 結(jié)束，耗時(shí): 2.0021 秒
--------------------
計(jì)時(shí)器 '有問題的操作' 開始...
準(zhǔn)備執(zhí)行一個(gè)會(huì)出錯(cuò)的操作...
計(jì)時(shí)器 '有問題的操作' 結(jié)束，耗時(shí): 1.0011 秒
在 '有問題的操作' 中發(fā)生了異常: division by zero
Traceback (most recent call last):
  File "...", line 36, in <module>
    result = 1 / 0  # 這里會(huì)產(chǎn)生一個(gè) ZeroDivisionError
ZeroDivisionError: division by zero

import time
from contextlib import contextmanager

@contextmanager
def timer(name):
    print(f"計(jì)時(shí)器 '{name}' 開始...")
    start_time = time.time()
    
    # yield 之前的部分，相當(dāng)于 __enter__
    # yield 的值會(huì)成為 as 后面的變量（如果沒有 yield 值，則為 None）
    try:
        yield
    finally:
        # yield 之后的部分，相當(dāng)于 __exit__
        end_time = time.time()
        duration = end_time - start_time
        print(f"計(jì)時(shí)器 '{name}' 結(jié)束，耗時(shí): {duration:.4f} 秒")

# 使用方法完全一樣
with timer("數(shù)據(jù)處理_v2"):
    print("正在處理數(shù)據(jù)...")
    time.sleep(2)
    print("數(shù)據(jù)處理完成。")