通常,操作一个数据库(如mysql)要依次执行connect、execute、commit、close操作,如
mysql_conn = pymysql.connect(host="10.255.255.xx",
user="xxxr",
passwd="xx",
db='bi_mdata',
charset='utf8')
mysql_cursor = mysql_conn.cursor()
insert_sql ='xxxxx'
mysql_cursor.executemany(insert_sql,)
mysql_conn.commit()
mysql_cursor.close()
mysql_conn.close()
上述代码无疑是繁琐的。不过我们可以经常看到如下的代码格式
with open(file_name,'r',encoding='UTF-8') as f:
.....
上述代码即可实现文件的打开和关闭。这就是上下文管理器,执行with 后面的表达式得到的结果是上下文管理器对象。
上下文管理器通过with语句实现。with 语句会设置一个临时的上下文,交给上下文管理器对象控制,并且负责清理上下文。这么做能避免错误并减少样板代码,因此 API 更安全,而且更易于使用。
我们可以仿照此格式自定义一个管理数据库的上下文管理器。不过上下文管理器协议包含 __enter__ 和 __exit__ 两个方法。with 语句开始运行时,会在上下文管理器对象上调用 __enter__ 方法。with 语句运行结束后,会在上下文管理器对象上调用 __exit__ 方法。
如
class DB:
def __init__(self, hostname, port,user,passwd,db):
self.hostname = hostname
self.port = port
self.user=user
self.passwd=passwd
self.db=db
self.connection = None
self.cursor=None
def __enter__(self):
self.connection = pymysql.connect(host=self.hostname,port=self.port,user=self.user,password=self.passwd,database=self.db)
self.cursor=self.connection.cursor()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.cursor.close()
self.connection.close()
使用:
with DB('10.255.255.xx', 80xx,'root','123456','TESTDB') as db_client:
sql='select cust_id from customer where cust_id<11040'
db_client.cursor.execute(sql)
ans=db_client.cursor.fetchall()
print(ans)
执行with 后面的表达式得到的结果是上下文管理器对象,不过,把值绑定到目标变量上(as 子句)是在上下文管理器对象上调用 __enter__ 方法的结果。with 语句的 as 子句是可选的。
不管控制流程以哪种方式退出 with 块,都会在上下文管理器对象上调用 __exit__ 方法,而不是在 __enter__ 方法返回的对象上调用。
>>> db_client
<__main__.DB instance at 0x7f824d565fc8>
>>> db_client.cursor
<pymysql.cursors.Cursor object at 0x7f824d560f10>
再次使用cursor
>>> sql='xxxxxxxxxxxxxxxxxx'
>>> db_client.cursor.execute(sql)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/lib64/python2.7/site-packages/pymysql/cursors.py", line 165, in execute
while self.nextset():
File "/usr/lib64/python2.7/site-packages/pymysql/cursors.py", line 107, in nextset
return self._nextset(False)
File "/usr/lib64/python2.7/site-packages/pymysql/cursors.py", line 91, in _nextset
conn = self._get_db()
File "/usr/lib64/python2.7/site-packages/pymysql/cursors.py", line 73, in _get_db
raise err.ProgrammingError("Cursor closed")
pymysql.err.ProgrammingError: Cursor closed
>>>
这样一来,只要你写完了 DB 这个类,那么在程序每次连接数据库时,我们都只需要简单地调用 with 语句即可,并不需要关心数据库的关闭、异常等等,显然大大提高了开发的效率。
网友评论