Skip to content

Commit b2ea173

Browse files
authored
fix rowcount (#26)
* fix rowcount
1 parent 1d305ef commit b2ea173

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

pydataapi/dbapi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def execute(
130130
)
131131
rows: List[List] = getattr(result, '_rows')
132132
self._rows = rows
133-
self._rowcount = len(rows)
133+
self._rowcount = len(rows) or result.number_of_records_updated
134134
return self
135135

136136
def executemany(

pydataapi/pydataapi.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -174,8 +174,8 @@ def __init__(self, response: Dict):
174174
super().__init__(response.get('generatedFields', []))
175175

176176
@property
177-
def number_of_records_updated(self) -> Optional[int]:
178-
return self._response.get('numberOfRecordsUpdated')
177+
def number_of_records_updated(self) -> int:
178+
return self._response.get('numberOfRecordsUpdated', 0)
179179

180180
@property
181181
def headers(self) -> List[str]:

0 commit comments

Comments
 (0)