Merge pull request #352 from bfirsh/fix-tls-sockets-being-closed-when-attaching

Stop TLS sockets being closed when attaching
This commit is contained in:
Sam Alba 2014-10-16 07:54:01 -07:00
commit 91b1a93100
1 changed files with 4 additions and 2 deletions

View File

@ -259,9 +259,11 @@ class Client(requests.Session):
def _get_raw_response_socket(self, response):
self._raise_for_status(response)
if six.PY3:
return response.raw._fp.fp.raw._sock
sock = response.raw._fp.fp.raw._sock
else:
return response.raw._fp.fp._sock
sock = response.raw._fp.fp._sock
sock._response = response
return sock
def _stream_helper(self, response):
"""Generator for data coming from a chunked-encoded HTTP response."""