Skip to content

Commit 66d7517

Browse files
remove SeaResultSet
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 69ea238 commit 66d7517

File tree

1 file changed

+0
-72
lines changed

1 file changed

+0
-72
lines changed

src/databricks/sql/result_set.py

Lines changed: 0 additions & 72 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)