Browse Source

丰富提示信息

daniel-ali 2 years ago
parent
commit
0dd096da92
2 changed files with 12 additions and 10 deletions
  1. 6 6
      QMT/download_data_whole.py
  2. 6 4
      QMT/real_time.py

+ 6 - 6
QMT/download_data_whole.py

@@ -53,7 +53,9 @@ def to_sql(stock_list):
     print(f'Pid:{os.getpid()}已经完工了.应入库{len(stock_list)},共入库{m}支个股')
 
 
-def download_data(stock_list):
+def download_data():
+    stock_list = xtdata.get_stock_list_in_sector('沪深A股')
+    stock_list.sort()
     print(dt.now(), '开始下载!')
     xtdata.download_history_data2(stock_list=stock_list, period='1d', start_time='', end_time='')
     print(dt.now(), '下载完成,准备入库!')
@@ -69,16 +71,14 @@ def download_data(stock_list):
 
 
 if __name__ == '__main__':
-    stocks = xtdata.get_stock_list_in_sector('沪深A股')
     field = ['time', 'open', 'close', 'high', 'low', 'volume', 'amount']
     cpu_count = mp.cpu_count()
-    stocks.sort()
-    step = math.ceil(len(stocks) / cpu_count)
 
-    # download_data(stocks)
+
+    # download_data()
 
     scheduler = BlockingScheduler()
-    scheduler.add_job(func=download_data, trigger='cron', hour='15', minute='45', args=[stocks],
+    scheduler.add_job(func=download_data, trigger='cron', day_of_week='0-4', hour='15', minute='40',
                       timezone="Asia/Shanghai")
     try:
         scheduler.start()

+ 6 - 4
QMT/real_time.py

@@ -91,8 +91,8 @@ def sell_trader(data, positions_dict):
     # for m in data:
     #     print(m, data[m]['lastPrice'])
     print('卖出函数:', dt.now())
-    # positions = xt_trader.query_stock_positions(acc)
-    # print('持仓总数:', len(positions_list))
+    positions = xt_trader.query_stock_positions(acc)
+    print('持仓总数:', len(positions))
 
     for stock, volume in positions_dict.items():
         if stock in data:
@@ -107,6 +107,8 @@ def sell_trader(data, positions_dict):
                                                  xtconstant.LATEST_PRICE, 0, 'strategy1', 'order_test')
                 print('价格:', current_price, open_price, MA5, MA5_1)
                 print(order_id, stock, volume)
+        else:
+            print(f'本轮没有持仓股票信息!')
 
 
 def buy_trader(data, positions):
@@ -120,7 +122,7 @@ def buy_trader(data, positions):
             'select value from `%s`' % '1d', engine_hlfx_pool)
         stock_pool = stock_pool.iloc[-1, 0].split(",")
         stock_pool.sort()
-        print('stock_pool', stock_pool)
+        print('stock_pool',len(stock_pool))
     except BaseException:
         pass
     '''
@@ -182,7 +184,7 @@ def buy_trader(data, positions):
         new_keep_stock = [stock.replace('XSHG', 'SH').replace('XSHE', 'SZ') for stock in keep_stocks]
         print(f'new_keep_stock is:{len(new_keep_stock)},{new_keep_stock}')
 
-        max_pos = 12
+        max_pos = 15
         for stock in new_keep_stock:
             asset = xt_trader.query_stock_asset(acc)
             cash = asset.cash