Skip to content

Commit ba2b35c

Browse files
committed
Added PEP249 optional cursor.rownumber
1 parent da7ce17 commit ba2b35c

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pynuodb/cursor.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(self, session, prepared_statement_cache_size):
5757
self.description = None
5858
self.rowcount = -1
5959
self.colcount = -1
60-
60+
self.rownumber = 0
6161
self.__query = None
6262

6363
@property
@@ -123,6 +123,7 @@ def execute(self, operation, parameters=None):
123123
# TODO: ???
124124
if self.rowcount < 0:
125125
self.rowcount = -1
126+
self.rownumber = 0
126127

127128
def _execute(self, operation):
128129
"""Handles operations without parameters."""
@@ -153,7 +154,7 @@ def fetchone(self):
153154
self._check_closed()
154155
if self._result_set is None:
155156
raise Error("Previous execute did not produce any results or no call was issued yet")
156-
157+
self.rownumber += 1
157158
return self._result_set.fetchone(self.session)
158159

159160
def fetchmany(self, size=None):
@@ -172,7 +173,6 @@ def fetchmany(self, size=None):
172173
else:
173174
fetched_rows.append(row)
174175
num_fetched_rows += 1
175-
176176
return fetched_rows
177177

178178
def fetchall(self):
@@ -186,7 +186,6 @@ def fetchall(self):
186186
break
187187
else:
188188
fetched_rows.append(row)
189-
190189
return fetched_rows
191190

192191
def nextset(self):

0 commit comments

Comments
 (0)