Browse Source

修正实时交易判断表为whole

Daniel 2 years ago
parent
commit
5b0e22d103
1 changed files with 8 additions and 8 deletions
  1. 8 8
      QMT/real_time.py

+ 8 - 8
QMT/real_time.py

@@ -19,7 +19,7 @@ db_pool = pymysql.connect(host='localhost',
                           password='r6kEwqWU9!v3',
                           database='hlfx_pool')
 cursor_pool = db_pool.cursor()
-engine_stock = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/qmt_stocks_front?charset=utf8')
+engine_stock = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/qmt_stocks_whole?charset=utf8')
 
 
 def real_price(datas):
@@ -36,11 +36,11 @@ def ma(stock, num, data):
     try:
         i = (num - 1) * -1
         df = pd.read_sql_query(
-            'select close from `%s_1d`' % stock, engine_stock)
+            'select close_front from `%s_1d`' % stock, engine_stock)
     except:
         return 9999999
     else:
-        ma_num = (sum(df['close'][i:]) + data[stock]['lastPrice'])/num
+        ma_num = (sum(df['close_front'][i:]) + data[stock]['lastPrice'])/num
         return ma_num
 
 
@@ -49,11 +49,11 @@ def ma_1(stock, num):
     i = (num) * -1
     try:
         df = pd.read_sql_query(
-            'select close from `%s_1d`' % stock, engine_stock)
+            'select close_front from `%s_1d`' % stock, engine_stock)
     except BaseException:
         return 9999999
     else:
-        ma_num_1 = df['close'][i:].mean()
+        ma_num_1 = df['close_front'][i:].mean()
         return ma_num_1
 
 
@@ -62,18 +62,18 @@ def his_vol(stock, num):
     num = num * -1
     try:
         df = pd.read_sql_query(
-            'select volume from `%s_1d`' % stock, engine_stock)
+            'select volume_front from `%s_1d`' % stock, engine_stock)
     except BaseException:
         return 9999999
     else:
-        return df['volume'].iloc[num]
+        return df['volume_front'].iloc[num]
 
 
 def ma_judge(data, stock_list, results):
     print('这个ma_judge的PID为:', os.getpid())
     for stock in data:
         i = stock.replace('XSHG', 'SH').replace('XSHE', 'SZ')
-        current_price, open_price = data[i]['lastPrice'], data[i]['open']
+        current_price, open_price = data[i]['lastPrice'], data[i]['open_front']
         MA5, MA10, MA20 = ma(i, 5, data), ma(i, 10, data), ma(i, 20, data)
         MA5_1 = ma_1(i, 5)
         # print(i, current_price, open_price, MA5, MA10, MA20, MA5_1)