python之with语句结合上下文管理器

所谓上下文管理器即在一个类中重写了__enter__方法和__exit__方法的类就能够成为上下文管理器类。

咱们能够经过with语句结合上下文管理器简化一些操做。

使用with语句结合自定义上下文管理器完成数据库相应的操做，代码实现以下：

# 1. 导入模块
import pymysql # 建立自定义上下文管理器对象
class MyDatabase(object): # 接收参数并建立数据库链接对象
    def __init__(self, host, port, user, passwd, database): self.__db = pymysql.Connection(host, port, user, passwd, database, charset='utf8') # 返回数据库链接对象
    def __enter__(self): return self.__db

    # 关闭数据库链接
    def __exit__(self, exc_type, exc_val, exc_tb): self.__db.close() def main(): # 使用with关键字接收enter返回的对象给db
    with MyDatabase('localhost', 3306, 'root', 'mysql', 'JDDB') as db: # 利用db建立游标
        cur = db.cursor() sql = '''select * from %s''' cur.execute(sql, (goods,)) result = cur.fetchall() for i in result: print(i) # 关闭游标
 cur.close() # 程序入口
if __name__ == '__main__': main()

上下文管理器类的代码流程：

1.编写__init__方法用来接收参数，并建立数据库链接对象；

2.重写__enter__方法，返回数据库链接对象；

3.重写__exit__方法，用来关闭数据库链接；

with语句代码流程：

1.当将建立对象的语句放到with语句里时不会建立对象，而是接受__enter__方法返回的对象并给对象起个别名；

2.使用接受到的对象即数据库链接对象，建立游标；

3.编写SQL语句，并经过游标执行SQL语句；

4.获取SQL语句的查询结果，并显示出来；

5.关闭游标；

6.当with语句内的代码执行完毕后自动执行__exit__方法关闭数据库链接。

注意：with MyDatabase() as db ---> db = MyDatabase().__enter__()

利用with结合自定义上下文类实现HTTP服务端：

# 1.导入socket模块
import socket class MySocket(object): # 2.编写init方法接收port参数
    def __init__(self, port): self.__port = port # 3.建立socket对象
        self.__sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 4.编写enter方法返回套接字对象
    def __enter__(self): # 设置端口复用
        self.__sk.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True) # 绑定端口
        self.__sk.bind(self.__port) # 设置端口监听
        self.__sk.listen(128) # 返回套接字对象
        return self.__sk

    # 5.编写exit方法关闭套接字对象
    def __exit__(self, exc_type, exc_val, exc_tb): self.__sk.close() def main(): # 使用with关键字 并接受返回的套接字对象给sk
    with MySocket(8000) as sk: # 等待客户端链接
        clicent, ip_port = sk.accept() recv_data = clicent.recv(1024) print(recv_data.decode('utf-8')) # 编写HTTP响应报文
        http_line = 'HTTP/1.1 GET 200 OK\r\n' http_header = 'Server PWS/1.0\r\n' http_body = 'Welcome to index!\r\n' send_data = (http_line + http_header + '\r\n' + http_body).encode('utf-8') clicent.send(send_data) # 关闭客户端链接
 clicent.close() # 编写程序入口
if __name__ == '__main__': main()

自定义上下文管理器类的代码解读：

1.编写__init__方法，用来接收参数并建立套接字对象；

2.编写__enter__方法，并使用套接字对象设置端口复用、绑定端口、并设置监听，而后返回套接字对象；

3.编写__exit__方法，关闭套接字对象。

with语句代码解读：

1.接收enter返回的套接字对象，并起个别名，

2.经过返回套接字对象等待客户端链接，

3.接收客户端链接成功后会返回一个新的套接字和IP端口号，

4.使用客户端套接字发送HTTP响应报文

5.关闭客户端链接

6.当with语句中的代码执行完毕后自动执行__exit__方法，关闭服务器链接

with管理器例题：

# 要求1：能够使用with Mysql('heima', 'root', 'mysql') as cursor:
#
#
# 语句获取游标对象cursor,操做cursor便可完成数据库操做；
#
#
# 要求2：Mysql类中三个参数分别对应要链接的数据库名、用户名、密码；
#
#
# 要求3：若是数据库操做正常，上下文中自动提交（自动执行commit()方法）,
# # 关闭游标而且关闭数据库链接；
#
#
# 要求4：若是数据库操做出现异常，上下文中自动回滚（自动执行rollback()方法），
# 关闭游标而且关闭数据库链接，捕获异常并打印报错信息。
import pymysql

global bool


class Mysql(object):
    def __init__(self, database, user, password):
        self.__db = pymysql.Connection(host='localhost', port=3306, database=database, user=user, password=password,
                                       charset='utf8')

    def __enter__(self):
        self.__cur = self.__db.cursor()
        return self.__cur, self.__db

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


def main():
    with Mysql('heima', 'root', 'mysql') as cursor:
        cur, conn = cursor
        sql = '''SQL语句'''
        cur.execute(sql)

if __name__ == '__main__':
    main()

 

注意：__exit(self, exc_type, exc_val, exc_tb)__函数中的三个参数exc_type, exc_val, exc_tb只有在程序异常退出时才会有值，

只要判断这三个参数中的其中一个，若是有值那就说明程序存在异常。

当__exit()__函数返回值为Flase（默认）时，为显示程序异常信息（红色字），

当函数返回值为True时，为不显示程序异常信息。