Commit241e934

nithinkdb

and

Jesse

authored

[PECO-1083] Updated thrift files and added check for protocol version (#229)

* Updated thrift files and added check for protocol versionSigned-off-by: nithinkdb <nithin.krishnamurthi@databricks.com>* Made error message more clearSigned-off-by: nithinkdb <nithin.krishnamurthi@databricks.com>* Changed name of fnSigned-off-by: nithinkdb <nithin.krishnamurthi@databricks.com>* Ran linterSigned-off-by: nithinkdb <nithin.krishnamurthi@databricks.com>* Update src/databricks/sql/client.pyCo-authored-by: Jesse <jwhitehouse@airpost.net>---------Signed-off-by: nithinkdb <nithin.krishnamurthi@databricks.com>Co-authored-by: Jesse <jwhitehouse@airpost.net>

1 parent7c72cf4 commit241e934Copy full SHA for 241e934

File tree

3 files changed

+3964

-324

lines changed

src/databricks/sql
- client.py
- thrift_api/TCLIService
  - ttypes.py
- thrift_backend.py

3 files changed

+3964

-324

lines changed

`‎src/databricks/sql/client.py‎`

Lines changed: 13 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@`
`5`	`5`	`importrequests`
`6`	`6`	`importjson`
`7`	`7`	`importos`
	`8`	`+fromdatabricks.sql.thrift_api.TCLIServiceimportttypes`
`8`	`9`
`9`	`10`	`fromdatabricks.sqlimport__version__`
`10`	`11`	`fromdatabricks.sqlimport*`
`@@ -225,6 +226,11 @@ def __del__(self):`
`225`	`226`	`defget_session_id(self):`
`226`	`227`	`returnself.thrift_backend.handle_to_id(self._session_handle)`
`227`	`228`
	`229`	`+defget_session_protocol_version(self):`
	`230`	`+returnself.thrift_backend.extract_protocol_version_from_handle(`
	`231`	`+self._session_handle`
	`232`	`+ )`
	`233`	`+`
`228`	`234`	`defget_session_id_hex(self):`
`229`	`235`	`returnself.thrift_backend.handle_to_hex_id(self._session_handle)`
`230`	`236`
`@@ -501,6 +507,13 @@ def execute(`
`501`	`507`	`"""`
`502`	`508`	`ifparametersisNone:`
`503`	`509`	`parameters= []`
	`510`	`+elif (`
	`511`	`+self.connection.get_session_protocol_version()`
	`512`	`+<ttypes.TProtocolVersion.SPARK_CLI_SERVICE_PROTOCOL_V8`
	`513`	`+ ):`
	`514`	`+raiseError(`
	`515`	`+"Parameterized operations are not supported by this server. DBR 14.1 is required."`
	`516`	`+ )`
`504`	`517`	`else:`
`505`	`518`	`parameters=named_parameters_to_tsparkparams(parameters)`
`506`	`519`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit241e934

File tree

3 files changed

3 files changed

`‎src/databricks/sql/client.py‎`

0 commit comments