Support timeout when querying orders from online quant sites

Author: sinall

shipane_sdk/joinquant/client.py

@@ -18,6 +18,7 @@ def __init__(self, **kwargs):
         self._username = kwargs.get('username', None)
         self._password = kwargs.get('password', None)
         self._backtest_id = kwargs.get('backtest_id', None)
+        self._timeout = kwargs.pop('timeout', (5.0, 10.0))
 
     def login(self):
         self._session.headers = {
@@ -30,12 +31,12 @@ def login(self):
             'Origin': self.BASE_URL,
             'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
         }
-        self._session.get(self.BASE_URL)
+        self._session.get(self.BASE_URL, timeout=self._timeout)
         response = self._session.post('{}/user/login/doLogin?ajax=1'.format(self.BASE_URL), data={
             'CyLoginForm[username]': self._username,
             'CyLoginForm[pwd]': self._password,
             'ajax': 1
-        })
+        }, timeout=self._timeout)
         self._session.headers.update({
             'cookie': response.headers['Set-Cookie']
         })
@@ -48,7 +49,7 @@ def query(self):
             'backtestId': self._backtest_id,
             'date': today_str,
             'ajax': 1
-        })
+        }, timeout=self._timeout)
         transaction_detail = response.json()
         raw_transactions = transaction_detail['data']['transaction']
         transactions = []

shipane_sdk/ricequant/client.py

@@ -10,15 +10,16 @@ class RiceQuantClient(BaseQuantClient):
     def __init__(self, **kwargs):
         super(RiceQuantClient, self).__init__('RiceQuant')
 
-        self._rq_client = RQOpenClient(kwargs.get('username', None), kwargs.get('password', None))
+        self._rq_client = RQOpenClient(kwargs.get('username', None), kwargs.get('password', None),
+                                       timeout=kwargs.pop('timeout', (5.0, 10.0)))
         self._run_id = kwargs.get('run_id', None)
 
     def login(self):
-        self._rq_client.login()
+        self._rq_client.login(timeout=self._timeout)
         super(RiceQuantClient, self).login()
 
     def query(self):
-        response = self._rq_client.get_day_trades(self._run_id)
+        response = self._rq_client.get_day_trades(self._run_id, timeout=self._timeout)
         raw_transactions = response['resp']['trades']
         transactions = []
         for raw_transaction in raw_transactions:

shipane_sdk/uqer/client.py

@@ -18,6 +18,7 @@ def __init__(self, **kwargs):
         self._username = kwargs.get('username', None)
         self._password = kwargs.get('password', None)
         self._strategy = kwargs.get('strategy', None)
+        self._timeout = kwargs.pop('timeout', (5.0, 10.0))
 
     def login(self):
         self._session.headers = {
@@ -30,12 +31,12 @@ def login(self):
             'Origin': self.BASE_URL,
             'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
         }
-        self._session.get(self.BASE_URL)
+        self._session.get(self.BASE_URL, timeout=self._timeout)
         response = self._session.post('{}/usermaster/authenticate/v1.json'.format(self.BASE_URL), data={
             'username': self._username,
             'password': self._password,
             'rememberMe': 'false'
-        })
+        }, timeout=self._timeout)
         self._session.headers.update({
             'cookie': response.headers['Set-Cookie']
         })
@@ -44,9 +45,10 @@ def login(self):
 
     def query(self):
         today_str = datetime.today().strftime('%Y-%m-%d')
-        response = self._session.get('{}/mercury_trade/strategy/{}/order'.format(self.BASE_URL, self._strategy), params={
-            'date': today_str,
-        })
+        response = self._session.get('{}/mercury_trade/strategy/{}/order'.format(self.BASE_URL, self._strategy),
+                                     params={
+                                         'date': today_str,
+                                     }, timeout=self._timeout)
         raw_transactions = response.json()
         transactions = []
         for raw_transaction in raw_transactions: