#

Author: yutiansut

QUANTAXIS/QAStrategy/qactabase.py

@@ -19,6 +19,7 @@
 import QUANTAXIS as QA
 from QUANTAXIS.QAStrategy.util import QA_data_futuremin_resample
 from QUANTAXIS.QIFI.QifiAccount import ORDER_DIRECTION, QIFI_Account
+from QUANTAXIS.QAMarket.market_preset import MARKET_PRESET
 from QUANTAXIS.QAEngine.QAThreadEngine import QA_Thread
 from QUANTAXIS.QAUtil.QAParameter import MARKET_TYPE, RUNNING_ENVIRONMENT
 
@@ -62,7 +63,7 @@ def __init__(self, code='rb2005', frequence='1min', strategy_id='QA_STRATEGY', r
 
         self.running_time = ''
 
-        self.market_preset = QA.QAARP.MARKET_PRESET()
+        self.market_preset = MARKET_PRESET()
         self._market_data = []
         self.risk_check_gap = risk_check_gap
         self.latest_price = {}

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,model='REALTIME')
+    manager = QA_QIFISMANAGER(mongo_ip,model='BACKTEST')
 
     def get(self):
         action = self.get_argument('action', 'acchistory')
@@ -98,6 +98,43 @@ def get(self):
 
 
 class QAQIFIS_Handler(QABaseHandler):
+    #manager = QA_QIFIMANAGER(mongo_ip)
+    manager = QA_QIFISMANAGER(mongo_ip,model='BACKTEST')
+
+    def get(self):
+        action = self.get_argument('action', 'acchistory')
+
+        if action == 'accountlist':
+            res = self.manager.get_allaccountname()
+            self.write({'res': res})
+        elif action == 'portfoliolist':
+            res = self.manager.get_allportfolio()
+            self.write({'res': res})
+        elif action == 'accountinportfolio':
+            portfolio = self.get_argument('portfolio', 't12')
+            res = self.manager.get_portfolio_panel(portfolio)
+
+            self.write({'res': QA_util_to_json_from_pandas(res)})
+
+    def post(self):
+        action = self.get_argument('action', 'change_name')
+        if action == 'drop_account':
+
+            account_cookie = self.get_argument('account_cookie')
+            res = self.manager.drop_account(account_cookie)
+            self.write({
+                'res': res,
+                'status': 200
+            })
+        elif action == 'drop_many':
+            account_cookies = self.get_argument('account_cookies')
+            res = self.manager.drop_many(account_cookies)
+            self.write({
+                'res': res,
+                'status': 200
+            })
+
+class QAQIFIS_REALTIME_Handler(QABaseHandler):
     #manager = QA_QIFIMANAGER(mongo_ip)
     manager = QA_QIFISMANAGER(mongo_ip,model='REALTIME')
 

QUANTAXIS/QAWebServer/server.py

@@ -35,7 +35,7 @@
 from QUANTAXIS.QAWebServer.schedulehandler import (QAScheduleQuery,
                                                    QASchedulerHandler,
                                                    init_scheduler)
-from QUANTAXIS.QAWebServer.qifiserver import QAQIFI_Handler, QAQIFIS_Handler
+from QUANTAXIS.QAWebServer.qifiserver import QAQIFI_Handler, QAQIFIS_Handler, QAQIFIS_REALTIME_Handler
 from tornado.options import (define, options, parse_command_line,
                              parse_config_file)
 from tornado.web import Application, RequestHandler, authenticated
@@ -71,6 +71,7 @@ def get(self):
     (r"/scheduler/query", QAScheduleQuery),
     (r"/qifi", QAQIFI_Handler),
     (r"/qifis", QAQIFIS_Handler),
+    (r"/qifirealtime", QAQIFIS_REALTIME_Handler),
     (r"/user", QAUserhander)
 
 ]