python/mysqlclient
1
0
Fork 0
mirror of https://github.com/PyMySQL/mysqlclient.git synced 2025-08-15 02:54:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cursor supports context manager interface

Browse Source
This commit is contained in:
INADA Naoki
2015-02-25 11:26:39 +09:00
parent 864e27905e
commit 1c818f8034
1 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
MySQLdb/cursors.py
View File

@ -85,13 +85,22 @@ class BaseCursor(object):
def close(self): def close(self):
"""Close the cursor. No further queries will be possible.""" """Close the cursor. No further queries will be possible."""
if self.connection is None or self.connection() is None: try:
return if self.connection is None or self.connection() is None:
while self.nextset(): return
pass while self.nextset():
self.connection = None pass
self.errorhandler = None finally:
self._result = None self.connection = None
self.errorhandler = None
self._result = None
def __enter__(self):
return self
def __exit__(self, *exc_info):
del exc_info
self.close()
def _check_executed(self): def _check_executed(self):
if not self._executed: if not self._executed: