Skip to content

Commit 2caf04b

Browse files
rzeyde-varadaggreg
authored andcommitted
Fix SET SESSION support by using http_response.headers
1 parent 7c65767 commit 2caf04b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

prestodb/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -413,14 +413,14 @@ def process(self, http_response):
413413

414414
if constants.HEADER_CLEAR_SESSION in http_response.headers:
415415
for prop in get_header_values(
416-
response.headers,
416+
http_response.headers,
417417
constants.HEADER_CLEAR_SESSION,
418418
):
419419
self._client_session.properties.pop(prop, None)
420420

421421
if constants.HEADER_SET_SESSION in http_response.headers:
422422
for key, value in get_session_property_values(
423-
response.headers,
423+
http_response.headers,
424424
constants.HEADER_SET_SESSION,
425425
):
426426
self._client_session.properties[key] = value

0 commit comments

Comments
 (0)