Fix attaching with UNIX sockets

Signed-off-by: Ben Firshman <ben@firshman.co.uk>
This commit is contained in:
Ben Firshman 2014-10-16 16:59:50 +01:00
parent 721edfd526
commit 537b468140
1 changed files with 10 additions and 1 deletions

View File

@ -262,7 +262,16 @@ class Client(requests.Session):
sock = response.raw._fp.fp.raw._sock
else:
sock = response.raw._fp.fp._sock
sock._response = response
try:
# Keep a reference to the response to stop it being garbage
# collected. If the response is garbage collected, it will close
# TLS sockets.
sock._response = response
except AttributeError:
# UNIX sockets can't have attributes set on them, but that's fine
# because we won't be doing TLS over them
pass
return sock
def _stream_helper(self, response):