Support customized requests session

Author: messense

bosonnlp/client.py

@@ -57,13 +57,13 @@ class BosonNLP(object):
 
     """
 
-    def __init__(self, token, bosonnlp_url=DEFAULT_BOSONNLP_URL, compress=True):
+    def __init__(self, token, bosonnlp_url=DEFAULT_BOSONNLP_URL, compress=True, session=None):
         self.token = token
         self.bosonnlp_url = bosonnlp_url.rstrip('/')
         self.compress = compress
 
         # Enable keep-alive and connection-pooling.
-        self.session = requests.session()
+        self.session = session or requests.session()
         self.session.headers['X-Token'] = token
         self.session.headers['Accept'] = 'application/json'
         self.session.headers['User-Agent'] = 'bosonnlp.py/{} {}'.format(

tests.py

@@ -321,3 +321,14 @@ def test_comments_task_with_multiple_push(nlp, input1, input2):
     comments.clear()
     assert len(result) == 4
     assert {'opinion', '_id', 'list', 'num'} == set(result[0].keys())
+
+
+def test_custom_requests_session():
+    import requests
+
+    session = requests.session()
+    nlp = BosonNLP(os.environ['BOSON_API_TOKEN'], session=session)
+    assert nlp.session == session
+    result = nlp.sentiment(['再也不来了', '美好的世界'])
+    assert result[0][1] > result[0][0]
+    assert result[1][0] > result[1][1]