6-contextlib

栏目: 数据库 · 发布时间: 6年前

内容简介:在python 中只要正确实现了上下文管理管理,就可以使用 with 语句。而 contextlib 就是一个帮助实现上下文管理的模块。所以什么是上下文管理? 一般来说,只要实现了 __enter__() 和 __exit__() 两个方法,在python 中就算是实现了上下文管理一个简单的例子

contextlib

python 中只要正确实现了上下文管理管理,就可以使用 with 语句。而 contextlib 就是一个帮助实现上下文管理的模块。

所以什么是上下文管理? 一般来说,只要实现了 __enter__() 和 __exit__() 两个方法,在python 中就算是实现了上下文管理

一个简单的例子

class Context():
    def __init__(self):
        print("__init__")
    
    def __enter__(self):
        print("__enter__")
    
    def __exit__(self,exc_type, exc_val, exc_tb):
        print("__exit__")

with Context():
    print("doing some works")

contextmanager装饰器

contextmanager在使用中应该算是最常见的。contextmanager是一个装饰器,用于实现上下文管理。 但首先,要明确的一点是能够使用 with 语句不一定就是说能够正确的关闭资源,能够使用with 只是说明了实现了上下文管理器。比如:

@contextmanager
def tag():
    print("<h1>")
    yield 
    print("</h1>")


with tag():
    print("Hello world")

这里也可以使用 with 但是只是实现了tag 函数的上下文管理,没有关闭任何资源。其输出:

<h1>
Hello world
</h1>

如果跟关闭资源有关,我们需要在函数的上下文中定义好关闭资源的方法,比如以连接 mysql 数据库为例:

from contextlib import contextmanager
import pymysql

# 定义连接上下文
@contextmanager
def create_connection():
    connection = pymysql.connect(host='localhost',
                             user='root',
                             password='qwe123',
                             db='apitesting',
                             charset='utf8mb4',
                             cursorclass=pymysql.cursors.DictCursor)
    try:
        yield connection
    finally:
        connection.close()

# 使用 with...as 输出上下文管理
with create_connection() as conn:
    with conn.cursor() as cursor:
        sql = "select * from account;"
        cursor.execute(sql)
        result = cursor.fetchone()
        print(result)
    conn.commit()

所以关闭资源是关闭资源,跟上下文管理不存在关系。

closing

如果一个对象没有实现上下文,就不能使用with语句. 这个时候可以使用closing 将其装饰为一个实现了上下文管理的对象,但是注意的是,对象必须要有 close 方法。

from contextlib import closing

class TestObj():
    def __init__(self):
        print("init test obj")
    def test(slef):
        print("test")
    def close(self):
        print("close")
    

with closing(TestObj()) as t:
    t.test()

以上所述就是小编给大家介绍的《6-contextlib》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

结构化计算机组成

结构化计算机组成

Andrew S.Tanenbaum / 刘卫东 / 机械工业出版社 / 2001-10-1 / 46.00

AndrewcS.Tanenbaum获得过美国麻省理工学院的理学学士学位和加利福尼亚大学伯克利分校的哲学博士学位,目前是荷兰阿姆斯特丹Vrije大学计算机科学系的教授,并领导着一个计算机系统的研究小组.同时,他还是一所计算与图像处理学院的院长,这是由几所大学合作成立的研究生院.尽管社会工作很多,但他并没有中断学术研究. 多年来,他在编译技术.操作系统.网络及局域分布式系统方面进行了大量的一起来看看 《结构化计算机组成》 这本书的介绍吧!

JS 压缩/解压工具
JS 压缩/解压工具

在线压缩/解压 JS 代码

MD5 加密
MD5 加密

MD5 加密工具

XML、JSON 在线转换
XML、JSON 在线转换

在线XML、JSON转换工具