def fetchallnumpy(self): self._assert_valid_result_set() if _has_numpy_support(): from turbodbc_numpy_support import make_numpy_result_set numpy_result_set = make_numpy_result_set(self.impl.get_result_set()) column_names = [description[0] for description in self.description] columns = zip(column_names, [_make_masked_array(data, mask) for data, mask in numpy_result_set.fetch_all()]) return OrderedDict(columns) else: raise Error("turbodbc was compiled without numpy support. Please install " "numpy and reinstall turbodbc")
def _numpy_batch_generator(self): self._assert_valid_result_set() if not _has_numpy_support(): raise Error(_NO_NUMPY_SUPPORT_MSG) from turbodbc_numpy_support import make_numpy_result_set numpy_result_set = make_numpy_result_set(self.impl.get_result_set()) first_run = True while True: result_batch = _make_masked_arrays(numpy_result_set.fetch_next_batch()) is_empty_batch = (len(result_batch[0]) == 0) if is_empty_batch and not first_run: raise StopIteration # Let us return a typed result set at least once first_run = False yield result_batch
def fetchallnumpy(self): self._assert_valid_result_set() if _has_numpy_support(): from turbodbc_numpy_support import make_numpy_result_set numpy_result_set = make_numpy_result_set( self.impl.get_result_set()) column_names = [description[0] for description in self.description] columns = zip(column_names, [ _make_masked_array(data, mask) for data, mask in numpy_result_set.fetch_all() ]) return OrderedDict(columns) else: raise Error( "turbodbc was compiled without numpy support. Please install " "numpy and reinstall turbodbc")
def _numpy_batch_generator(self): self._assert_valid_result_set() if _has_numpy_support(): from turbodbc_numpy_support import make_numpy_result_set else: raise Error( "turbodbc was compiled without numpy support. Please install " "numpy and reinstall turbodbc") numpy_result_set = make_numpy_result_set(self.impl.get_result_set()) first_run = True while True: result_batch = _make_masked_arrays( numpy_result_set.fetch_next_batch()) is_empty_batch = (len(result_batch[0]) == 0) if is_empty_batch and not first_run: raise StopIteration # Let us return a typed result set at least once first_run = False yield result_batch