psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-09-21 03:08:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

NamedTupleCursor doesn't change exception when fetching with no result.

Browse Source
This commit is contained in:
Daniele Varrazzo 2010-11-11 10:30:01 +00:00
parent ef7a5ee8a9
commit 3cae0f3f5d
3 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ChangeLog
View File

@ -3,6 +3,9 @@
* lib/extras.py: build the namedtuple only once per execution, not once
per fetch.
* lib/extras.py: don't change the exception raised when fetching without a
result from NamedTupleCursor.
2010-11-10 Daniele Varrazzo <daniele.varrazzo@gmail.com>
* psycopg/green.c: functions unused outside the module marked static.

2
lib/extras.py
View File

@ -305,7 +305,7 @@ class NamedTupleCursor(_cursor):
raise self._exc
else:
def _make_nt(self, namedtuple=namedtuple):
return namedtuple("Record", [d[0] for d in self.description])
return namedtuple("Record", [d[0] for d in self.description or ()])
class LoggingConnection(_connection):

9
tests/extras_dictcursor.py
View File

@ -219,6 +219,15 @@ class NamedTupleCursorTest(unittest.TestCase):
self.assertEqual(r.bar, 2)
self.assertRaises(AttributeError, getattr, r, 'foo')
@if_has_namedtuple
def test_no_result_no_surprise(self):
curs = self.conn.cursor()
curs.execute("update nttest set s = s")
self.assertRaises(psycopg2.ProgrammingError, curs.fetchone)
curs.execute("update nttest set s = s")
self.assertRaises(psycopg2.ProgrammingError, curs.fetchall)
@if_has_namedtuple
def test_minimal_generation(self):
# Instrument the class to verify it gets called the minimum number of times.