python的上下文管理器

最新推荐文章于 2025-09-29 15:54:10 发布

原创最新推荐文章于 2025-09-29 15:54:10 发布 · 95 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

10 篇文章

订阅专栏

本文探讨了Python中上下文管理器的实现方式，包括使用class定义资源管理器和利用contextlib模块创建上下文管理器。通过示例展示了如何在文件操作中应用上下文管理器，以及如何优雅地处理异常。

class Resource():

    def __enter__(self):
        print('===connect to resource===')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('===close resource connection===')
        return True #告诉python解释器异常已处理
        return False #告诉python解释器抛出异常

    def operate(self):
        1/0

import contextlib

@contextlib.contextmanager
def open_func(file_name):
    # __enter__方法
    print('open file:', file_name, 'in __enter__')
    file_handler = open(file_name, 'r')

    try:
        yield file_handler # yield后相当于__exit__
    except Exception as exc:
        # deal with exception
        raise Exception(str(exc))
        print('the exception was thrown: {}'.format(str(exc)))
    finally:
        print('close file:', file_name, 'in __exit__')
        file_handler.close()

        return

import os
from functools import partial
file_path = os.path.dirname(os.path.abspath(__file__))
get_file_path = partial(os.path.join, file_path)
print(get_file_path("deploy.sh"))

if __name__ == "__main__":
    with open_func(get_file_path("deploy.sh")) as file_in:
        for line in file_in:
            # 1/0
            print(line)


    # with Resource() as res:
    #     res.operate()