Skip to content

Commit fd1ab6d

Browse files
committed
Updated thrift files and added check for protocol version
1 parent fcc262f commit fd1ab6d

3 files changed

Lines changed: 3960 additions & 324 deletions

File tree

src/databricks/sql/client.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import requests
66
import json
77
import os
8+
from databricks.sql.thrift_api.TCLIService import ttypes
89

910
from databricks.sql import __version__
1011
from databricks.sql import *
@@ -225,6 +226,9 @@ def __del__(self):
225226
def get_session_id(self):
226227
return self.thrift_backend.handle_to_id(self._session_handle)
227228

229+
def get_session_protocol_version(self):
230+
return self.thrift_backend.handle_to_protocol_version(self._session_handle)
231+
228232
def get_session_id_hex(self):
229233
return self.thrift_backend.handle_to_hex_id(self._session_handle)
230234

@@ -501,6 +505,11 @@ def execute(
501505
"""
502506
if parameters is None:
503507
parameters = []
508+
elif (
509+
self.connection.get_session_protocol_version()
510+
< ttypes.TProtocolVersion.SPARK_CLI_SERVICE_PROTOCOL_V8
511+
):
512+
raise Error("Parameterized operations are not supported by this server")
504513
else:
505514
parameters = named_parameters_to_tsparkparams(parameters)
506515

0 commit comments

Comments
 (0)