14
14
from fakettypes import *
15
15
import socket
16
16
import urllib
17
+
17
18
__all__ = ['connect' , 'connect_thread_local' ]
18
19
19
20
"""
@@ -55,10 +56,11 @@ def execute(self, request):
55
56
headers .update (request .headers )
56
57
s = requests .session ()
57
58
response = s .request (method = Method ._VALUES_TO_NAMES [request .method ],
58
- url = "http://%s:%s%s" % (self .host , self .port , request .uri ), params = request .parameters ,
59
- data = request .body , headers = request .headers )
59
+ url = "http://%s:%s%s" % (self .host , self .port , request .uri ), params = request .parameters ,
60
+ data = request .body , headers = request .headers , timeout = self . timeout )
60
61
return RestResponse (status = response .status_code , body = response .content , headers = response .headers )
61
62
63
+
62
64
def connect (servers = None , framed_transport = False , timeout = None ,
63
65
retry_time = 60 , recycle = None , round_robin = None ,
64
66
max_retries = 3 , basic_auth = None ):
@@ -102,7 +104,7 @@ def connect(servers=None, framed_transport=False, timeout=None,
102
104
Expects keys:
103
105
* username
104
106
* password
105
-
107
+
106
108
round_robin: bool
107
109
*DEPRECATED*
108
110
@@ -114,8 +116,8 @@ def connect(servers=None, framed_transport=False, timeout=None,
114
116
if servers is None :
115
117
servers = [DEFAULT_SERVER ]
116
118
return ThreadLocalConnection (servers , framed_transport , timeout ,
117
- retry_time , recycle , max_retries = max_retries ,
118
- basic_auth = basic_auth )
119
+ retry_time , recycle , max_retries = max_retries ,
120
+ basic_auth = basic_auth )
119
121
120
122
connect_thread_local = connect
121
123
@@ -170,7 +172,6 @@ def __init__(self, servers, framed_transport=False, timeout=None,
170
172
171
173
def __getattr__ (self , attr ):
172
174
def _client_call (* args , ** kwargs ):
173
-
174
175
for retry in xrange (self ._max_retries + 1 ):
175
176
try :
176
177
return getattr (self ._ensure_connection (), attr )(* args , ** kwargs )
@@ -202,8 +203,8 @@ def connect(self):
202
203
server = self ._servers .get ()
203
204
log .debug ('Connecting to %s' , server )
204
205
self ._local .conn = ClientTransport (server , self ._framed_transport ,
205
- self ._timeout , self ._recycle ,
206
- self ._basic_auth )
206
+ self ._timeout , self ._recycle ,
207
+ self ._basic_auth )
207
208
except (socket .timeout , socket .error ):
208
209
log .warning ('Connection to %s failed.' , server )
209
210
self ._servers .mark_dead (server )
@@ -212,6 +213,6 @@ def connect(self):
212
213
213
214
def close (self ):
214
215
"""If a connection is open, close it."""
215
- # if self._local.conn:
216
- # self._local.conn.transport.close()
216
+ # if self._local.conn:
217
+ # self._local.conn.transport.close()
217
218
self ._local .conn = None
0 commit comments