#

Author: yutiansut

QUANTAXIS/QAWebServer/qifiserver.py

@@ -5,7 +5,7 @@
 from QUANTAXIS.QIFI.QifiManager import QA_QIFIMANAGER, QA_QIFISMANAGER
 class QAQIFI_Handler(QABaseHandler):
     #manager = QA_QIFIMANAGER(mongo_ip)
-    manager = QA_QIFISMANAGER(mongo_ip)
+    manager = QA_QIFISMANAGER(mongo_ip,model='REALTIME')
 
     def get(self):
         action = self.get_argument('action', 'acchistory')
@@ -99,7 +99,7 @@ def get(self):
 
 class QAQIFIS_Handler(QABaseHandler):
     #manager = QA_QIFIMANAGER(mongo_ip)
-    manager = QA_QIFISMANAGER(mongo_ip)
+    manager = QA_QIFISMANAGER(mongo_ip,model='REALTIME')
 
     def get(self):
         action = self.get_argument('action', 'acchistory')

QUANTAXIS/QIFI/QifiManager.py

@@ -171,9 +171,10 @@ def query_re(self, text) -> list:
 
     def get_portfolio_panel(self, portfolio) -> pd.DataFrame:
         r = self.get_portfolio_account(portfolio)
+        
         rp = [self.database.find_one({'account_cookie': i}, {
                                      "accounts": 1, 'trading_day': 1, '_id': 0}) for i in r]
-        return pd.DataFrame([mergex(i['accounts'], {'trading_day': i['trading_day']}) for i in rp])
+        return pd.DataFrame([mergex(i['accounts'], {'trading_day': i['trading_day']}) for i in rp]).query('user_id in {}'.format(r))
 
     def get_allaccountname(self) -> list:
         return list(set([i['account_cookie'] for i in self.database.find({}, {'account_cookie': 1, '_id': 0})]))