Merge pull request #9 from bosondata/feature/timeout

Add timeout support
bosondata · Sep 14, 2017 · 594287d · 594287d
2 parents 6e61bb2 + 63d3663
commit 594287d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/bosonnlp/client.py b/bosonnlp/client.py
@@ -55,12 +55,15 @@ class BosonNLP(object):
 
     :param bool compress: 是否压缩大于 10K 的请求体，默认为 True。
 
+    :param int timeout: HTTP 请求超时时间，默认为 60 秒。
+
     """
 
-    def __init__(self, token, bosonnlp_url=DEFAULT_BOSONNLP_URL, compress=True, session=None):
+    def __init__(self, token, bosonnlp_url=DEFAULT_BOSONNLP_URL, compress=True, session=None, timeout=60):
         self.token = token
         self.bosonnlp_url = bosonnlp_url.rstrip('/')
         self.compress = compress
+        self.timeout = timeout
 
         # Enable keep-alive and connection-pooling.
         self.session = session or requests.session()
@@ -71,6 +74,7 @@ def __init__(self, token, bosonnlp_url=DEFAULT_BOSONNLP_URL, compress=True, sess
         )
 
     def _api_request(self, method, path, **kwargs):
+        kwargs.setdefault('timeout', self.timeout)
         url = self.bosonnlp_url + path
         if method == 'POST':
             if 'data' in kwargs: