python上下文管理器细读

原创于 2018-04-13 23:23:51 发布 · 292 阅读

0 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

48 篇文章

订阅专栏

本文通过四个实例介绍了Python中上下文管理器的使用方法，包括如何利用@contextlib.contextmanager装饰器将生成器转换为上下文管理器，如何用其进行异常处理，contextlib.closing的正确使用方式以及如何实现文件的读写操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

test 1

上下文管理器，将生成器转化为上下文管理器

import contextlib
@contextlib.contextmanager
def a():
    print(1)
    yield
    print(3)

with a() as q:
    print(2)

test 2

使用上下文管理器，抽象出异常处理

import contextlib
@contextlib.contextmanager
def b():
    try:
        yield
    except Exception as error:
        print('error:',error)

with b():
    1/0

test 3

contextlib.closing 的使用时，要求方法中必须存在一个close的方法名称

import contextlib
class c:
    def d(self):
        print('start')
    def close(self):
        print('game over!')

with contextlib.closing(c()) as c_obj:
    print('contextlib.close()')

test 4

with 完成多个文件的读写操作

with open('a.jpg', 'rb') as from_file, open('b.jpg', 'wb') as to_file:
    to_file.write(from_file.read())