def __next__()

in azure-kusto-data/azure/kusto/data/response.py [0:0]

• 21 lines of code
• 9 McCabe index (conditional complexity)

    def __next__(self) -> Union[KustoResultTable, KustoStreamingResultTable]:
        if self.finished:
            raise StopIteration

        if type(self._current_table) is KustoStreamingResultTable and not self._current_table.finished and not self._skip_incomplete_tables:
            raise KustoStreamingQueryError(
                "Tried retrieving a new primary_result table before the old one was finished. To override call `set_skip_incomplete_tables(True)`"
            )

        while True:
            try:
                table = next(self.streamed_data)
            except StopIteration:
                self.finished = True
                raise
            if table["FrameType"] == FrameType.DataTable:
                break

        if table["TableKind"] == WellKnownDataSet.PrimaryResult.value:
            self._current_table = KustoStreamingResultTable(table)
        else:
            self._current_table = KustoResultTable(table)

        self.tables.append(self._current_table)
        return self._current_table