Return generator through chunked result set.
(
self,
result,
exit_stack: ExitStack,
chunksize: int | None,
columns,
coerce_float: bool = True,
parse_dates=None,
dtype_backend: DtypeBackend | Literal["numpy"] = "numpy",
)
| 1129 | return total_inserted |
| 1130 | |
| 1131 | def _query_iterator( |
| 1132 | self, |
| 1133 | result, |
| 1134 | exit_stack: ExitStack, |
| 1135 | chunksize: int | None, |
| 1136 | columns, |
| 1137 | coerce_float: bool = True, |
| 1138 | parse_dates=None, |
| 1139 | dtype_backend: DtypeBackend | Literal["numpy"] = "numpy", |
| 1140 | ) -> Generator[DataFrame]: |
| 1141 | """Return generator through chunked result set.""" |
| 1142 | has_read_data = False |
| 1143 | with exit_stack: |
| 1144 | while True: |
| 1145 | data = result.fetchmany(chunksize) |
| 1146 | if not data: |
| 1147 | if not has_read_data: |
| 1148 | yield DataFrame.from_records( |
| 1149 | [], columns=columns, coerce_float=coerce_float |
| 1150 | ) |
| 1151 | break |
| 1152 | |
| 1153 | has_read_data = True |
| 1154 | self.frame = _convert_arrays_to_dataframe( |
| 1155 | data, columns, coerce_float, dtype_backend |
| 1156 | ) |
| 1157 | |
| 1158 | self._harmonize_columns( |
| 1159 | parse_dates=parse_dates, dtype_backend=dtype_backend |
| 1160 | ) |
| 1161 | |
| 1162 | if self.index is not None: |
| 1163 | self.frame.set_index(self.index, inplace=True) |
| 1164 | |
| 1165 | yield self.frame |
| 1166 | |
| 1167 | def read( |
| 1168 | self, |
no test coverage detected