diff --git a/iotdb-client/client-py/iotdb/utils/IoTDBRpcDataSet.py b/iotdb-client/client-py/iotdb/utils/IoTDBRpcDataSet.py index e9621bf81b963..2249e1c7c0d46 100644 --- a/iotdb-client/client-py/iotdb/utils/IoTDBRpcDataSet.py +++ b/iotdb-client/client-py/iotdb/utils/IoTDBRpcDataSet.py @@ -25,6 +25,7 @@ from thrift.transport import TTransport from iotdb.thrift.rpc.IClientRPCService import TSFetchResultsReq, TSCloseOperationReq from iotdb.tsfile.utils.DateUtils import parse_int_to_date +from iotdb.utils.IoTDBConnectionException import IoTDBConnectionException from iotdb.utils.IoTDBConstants import TSDataType logger = logging.getLogger("IoTDB") @@ -417,6 +418,8 @@ def result_set_to_pandas(self): return df def fetch_results(self): + if self.__is_closed: + raise IoTDBConnectionException("This DataSet is already closed") self.__rows_index = 0 request = TSFetchResultsReq( self.__session_id,