Add compatibility code to retrieve headers

Python3 removed the getheaders() function and replaced it with a
get_all() one. Add compat code.

https://fedorahosted.org/freeipa/ticket/6558

Signed-off-by: Simo Sorce <simo@redhat.com>
Reviewed-By: Martin Basti <mbasti@redhat.com>
This commit is contained in:
Simo Sorce
2016-12-14 06:20:15 -05:00
committed by Martin Basti
parent b1a20599c4
commit 397f2be9df

View File

@@ -729,7 +729,11 @@ class KerbTransport(SSLTransport):
pass
def parse_response(self, response):
self.store_session_cookie(response.msg.getheaders('Set-Cookie'))
if six.PY2:
header = response.msg.getheaders('Set-Cookie')
else:
header = response.msg.get_all('Set-Cookie')
self.store_session_cookie(header)
return SSLTransport.parse_response(self, response)