Fix SET SESSION support by using http_response.headers

dvecerdea · Jan 10, 2019 · 2caf04b · 2caf04b
1 parent 7c65767
commit 2caf04b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prestodb/client.py b/prestodb/client.py
@@ -413,14 +413,14 @@ def process(self, http_response):
 
         if constants.HEADER_CLEAR_SESSION in http_response.headers:
             for prop in get_header_values(
-                response.headers,
+                http_response.headers,
                 constants.HEADER_CLEAR_SESSION,
             ):
                 self._client_session.properties.pop(prop, None)
 
         if constants.HEADER_SET_SESSION in http_response.headers:
             for key, value in get_session_property_values(
-                response.headers,
+                http_response.headers,
                 constants.HEADER_SET_SESSION,
             ):
                 self._client_session.properties[key] = value