@@ -416,75 +416,3 @@ def map_col_type(type_):
416416 (column .name , map_col_type (column .datatype ), None , None , None , None , None )
417417 for column in table_schema_message .columns
418418 ]
419-
420-
421- class SeaResultSet (ResultSet ):
422- """ResultSet implementation for SEA backend."""
423-
424- def __init__ (
425- self ,
426- connection : "Connection" ,
427- execute_response : "ExecuteResponse" ,
428- sea_client : "SeaDatabricksClient" ,
429- buffer_size_bytes : int = 104857600 ,
430- arraysize : int = 10000 ,
431- ):
432- """
433- Initialize a SeaResultSet with the response from a SEA query execution.
434-
435- Args:
436- connection: The parent connection
437- sea_client: The SeaDatabricksClient instance for direct access
438- buffer_size_bytes: Buffer size for fetching results
439- arraysize: Default number of rows to fetch
440- execute_response: Response from the execute command
441- """
442-
443- super ().__init__ (
444- connection = connection ,
445- backend = sea_client ,
446- arraysize = arraysize ,
447- buffer_size_bytes = buffer_size_bytes ,
448- command_id = execute_response .command_id ,
449- status = execute_response .status ,
450- has_been_closed_server_side = execute_response .has_been_closed_server_side ,
451- has_more_rows = execute_response .has_more_rows ,
452- results_queue = execute_response .results_queue ,
453- description = execute_response .description ,
454- is_staging_operation = execute_response .is_staging_operation ,
455- )
456-
457- def _fill_results_buffer (self ):
458- """Fill the results buffer from the backend."""
459- raise NotImplementedError ("fetchone is not implemented for SEA backend" )
460-
461- def fetchone (self ) -> Optional [Row ]:
462- """
463- Fetch the next row of a query result set, returning a single sequence,
464- or None when no more data is available.
465- """
466-
467- raise NotImplementedError ("fetchone is not implemented for SEA backend" )
468-
469- def fetchmany (self , size : Optional [int ] = None ) -> List [Row ]:
470- """
471- Fetch the next set of rows of a query result, returning a list of rows.
472-
473- An empty sequence is returned when no more rows are available.
474- """
475-
476- raise NotImplementedError ("fetchmany is not implemented for SEA backend" )
477-
478- def fetchall (self ) -> List [Row ]:
479- """
480- Fetch all (remaining) rows of a query result, returning them as a list of rows.
481- """
482- raise NotImplementedError ("fetchall is not implemented for SEA backend" )
483-
484- def fetchmany_arrow (self , size : int ) -> Any :
485- """Fetch the next set of rows as an Arrow table."""
486- raise NotImplementedError ("fetchmany_arrow is not implemented for SEA backend" )
487-
488- def fetchall_arrow (self ) -> Any :
489- """Fetch all remaining rows as an Arrow table."""
490- raise NotImplementedError ("fetchall_arrow is not implemented for SEA backend" )
0 commit comments