From 9b005decb9b5164a741a9217d505a0a6c19fa77f Mon Sep 17 00:00:00 2001 From: Kemp Po Date: Mon, 14 Apr 2025 22:16:29 +0800 Subject: [PATCH 1/2] add SESSION_AUTHORIZATION to client capabilities --- trino/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trino/client.py b/trino/client.py index 730a6f87..47724d2f 100644 --- a/trino/client.py +++ b/trino/client.py @@ -522,7 +522,7 @@ def http_headers(self) -> CaseInsensitiveDict[str]: headers[constants.HEADER_ENCODING] = self._client_session.encoding else: raise ValueError("Invalid type for encoding: expected str or list") - headers[constants.HEADER_CLIENT_CAPABILITIES] = 'PARAMETRIC_DATETIME' + headers[constants.HEADER_CLIENT_CAPABILITIES] = 'PARAMETRIC_DATETIME,SESSION_AUTHORIZATION' headers["user-agent"] = f"{constants.CLIENT_NAME}/{__version__}" if len(self._client_session.roles.values()): headers[constants.HEADER_ROLE] = ",".join( From 0ca3977a49791b8f3fffd3a1a9d0a88ee92779b6 Mon Sep 17 00:00:00 2001 From: Kemp Po Date: Mon, 14 Apr 2025 22:31:22 +0800 Subject: [PATCH 2/2] update tests --- tests/unit/test_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py index b33b72f5..a9225cd8 100644 --- a/tests/unit/test_client.py +++ b/tests/unit/test_client.py @@ -135,7 +135,7 @@ def assert_headers(headers): assert headers[constants.HEADER_SESSION] == "" assert headers[constants.HEADER_TRANSACTION] is None assert headers[constants.HEADER_TIMEZONE] == timezone - assert headers[constants.HEADER_CLIENT_CAPABILITIES] == "PARAMETRIC_DATETIME" + assert headers[constants.HEADER_CLIENT_CAPABILITIES] == "PARAMETRIC_DATETIME,SESSION_AUTHORIZATION" assert headers[accept_encoding_header] == accept_encoding_value assert headers[client_info_header] == client_info_value assert headers[constants.HEADER_ROLE] == ( @@ -1173,7 +1173,7 @@ def json(self): sql = 'execute my_stament using 1, 2, 3' additional_headers = { constants.HEADER_PREPARED_STATEMENT: 'my_statement=added_prepare_statement_header', - constants.HEADER_CLIENT_CAPABILITIES: 'PARAMETRIC_DATETIME' + constants.HEADER_CLIENT_CAPABILITIES: 'PARAMETRIC_DATETIME,SESSION_AUTHORIZATION' } # Patch the post function to avoid making the requests, as well as to