Python3的上下文管理器

写在前面

本篇文章主要演示在实际场景中，上下文管理器的使用方法，直接用代码的方式进行演示。

如果需要对上下文管理器进行了解入门，可以参考下API文档。

上下文管理器
Python 的 with 语句支持通过上下文管理器所定义的运行时上下文这一概念。 此对象的实现使用了一对专门方法，允许用户自定义类来定义运行时上下文，在语句体被执行前进…

STEP.1 代码

from contextlib import contextmanager


class Connection:
    def conn(self):
        print("conn")
        return "conn"

    def close(self):
        print("close")

    def commit(self):
        print("commit")

    def rollback(self):
        print("rollback")

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type:
            self.rollback()
        else:
            self.commit()
        self.close()
        return True

    def __enter__(self):
        return self.conn()


@contextmanager
def conn():
    try:
        print("conn")
        yield "conn"
        print("commit")
    except Exception:
        print("rollback")
    finally:
        print("close")


connection = Connet = Connection


if __name__ == "__main__":
    """
    模拟数据库操作场景，代码执行成功，则提交事务，否则回滚事务。
    不管执行是否成功，最后关闭连接。
    """

    # 传统方式，自己维护上下文
    c = Connection()
    try:
        print("GET:", c.conn())
        raise Exception
        c.commit()
    except Exception:
        c.rollback()
    finally:
        c.close()
    print("=" * 60)

    # 定义Class进入和退出时的动作
    with connection() as c:
        print("GET:", c)
        raise Exception
    print("=" * 60)

    # 使用contextlib包中封装好的@contextmanager进行上下文管理
    with conn() as c:
        print("GET:", c)
        raise Exception

输出

conn
GET: conn
rollback
close
============================================================
conn
GET: conn
rollback
close
============================================================
conn
GET: conn
rollback
close

结束

本文主要使用代码的形式对Python上下文管理器进行演示，代码地址：ContextManager.py