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

Merge pull request #26 from PyMySQL/feature/cursor-context-manager

Browse Source 
Add context manager interface to cursor
This commit is contained in:
INADA Naoki
2015-02-25 11:38:05 +09:00
parent 864e27905e 1c818f8034
commit a42fe113bf
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: