修正L* H* 问题

MA_Close_futures.py

@@ -0,0 +1,63 @@
+from jqdatasdk import *
+from datetime import datetime as dt
+import pandas as pd
+from jqdatasdk.technical_analysis import *
+from sqlalchemy import create_engine
+import threading
+
+
+auth('18616891214', 'Ea?*7f68nD.dafcW34d!')
+# futures = ['IF9999.CCFX', 'IC9999.CCFX','IH9999.CCFX', 'IM9999.CCFX', 'FU9999.XSGE', 'RB9999.XSGE', 'MA9999.XZCE', 'TA9999.XZCE', 'SA9999.XZCE', 'M9999.XDCE', 'LH9999.XDCE']
+# futures = ['IF9999.CCFX', 'IC9999.CCFX','IH9999.CCFX', 'IM9999.CCFX']
+futures = ['FU9999.XSGE', 'RB9999.XSGE', 'MA9999.XZCE', 'TA9999.XZCE', 'SA9999.XZCE', 'M9999.XDCE', 'LH9999.XDCE']
+# futures = ['FU9999.XSGE']
+engine = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/futures?charset=utf8')
+# 获取各stock的去包含dataframe
+fut = locals()
+print(dt.now(), '开始赋值！')
+for fre in ['30m']:
+    for future in futures:
+        try:
+            fut[future] = pd.read_sql_query('select date,open,close,high,low,volume,money from `%s_%s`' % (future, fre),
+                                            engine)
+        except BaseException:
+            continue
+
+print(dt.now(), '数据库数据已赋值！')
+
+# print(fut['IF9999.CCFX'])
+
+start = dt.now()
+print(start)
+for f in futures:
+    fut[f] = pd.concat([fut[f], pd.DataFrame(columns=['MA5', 'derta'])],axis=1)
+    print(fut[f])
+
+    for i in range(len(fut[f])):
+        MA1 = MA(f, check_date=fut[f].loc[i, 'date'], timeperiod=5)
+        MA1 = MA1[f]
+        fut[f].loc[i, 'MA5'] = MA1
+        # print(fut[f].loc[i, ['close','high','low']])
+        if fut[f].loc[i, 'close'] > MA1:
+            derta = fut[f].loc[i, 'high']/MA1 - 1
+        else:
+            derta = (fut[f].loc[i, 'low']/MA1 -1)
+        fut[f].loc[i, 'derta'] = derta
+    print('___________________________________')
+    print(fut[f])
+    fut[f].to_excel('/Users/daniel/Downloads/MA_derta/30m_%s.xlsx'% f)
+
+end= dt.now()
+print('总时长：', (end - start).seconds)
+
+
+
+
+
+
+
+
+
+# MA1 = MA('MA9999.XZCE', check_date='2022-09-25', timeperiod=5)
+# print(MA1['MA9999.XZCE'])
+

MA_HLFX_order.py

@@ -0,0 +1,81 @@
+from jqdatasdk import *
+from datetime import datetime as dt
+import pandas as pd
+
+from sqlalchemy import create_engine
+
+
+
+
+auth('18616891214', 'Ea?*7f68nD.dafcW34d!')
+
+fre = ['30m', '1d']
+
+engine_hlfx_pool = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/hlfx_pool?charset=utf8')
+engine_stock = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/stocks?charset=utf8')
+
+
+
+fut = locals()
+print(dt.now(), '开始赋值！')
+for fre in ['1d']:
+
+        try:
+            stock_pool = pd.read_sql_query(
+                'select value from `%s`' % fre, engine_hlfx_pool)
+            print(stock_pool)
+            stock_pool = stock_pool.iloc[-1, 0].split(",")
+            print(stock_pool)
+        except BaseException:
+            continue
+        for stock in stock_pool[0:1]:
+            print(stock)
+
+            df_stock=df = get_bars(stock, count=10, unit=fre, fields=['date', 'open', 'close','high','low' ], include_now=True, df=True)
+            price = df_stock.iloc[-1].at['close']
+            price_open = df_stock.iloc[-1].at['open']
+            MA5_1 = df_stock['close'][-7:-2].mean()
+            MA5 = df_stock['close'][-6:-1].mean()
+            MA10 = df_stock['close'].mean()
+            print(price,price_open, 'ma5_1:',MA5_1, 'ma5:', MA5,MA10)
+            if (price > price_open) & (price > MA5) & (MA5 > MA5_1) & (price < MA5 * 1.07):
+                print('BUY')
+
+print(dt.now(), '数据库数据已赋值！')
+
+# print(fut['IF9999.CCFX'])
+
+# start = dt.now()
+# print(start)
+# for f in futures:
+#     fut[f] = pd.concat([fut[f], pd.DataFrame(columns=['MA5', 'derta'])],axis=1)
+#     print(fut[f])
+#
+#     for i in range(len(fut[f])):
+#         MA1 = MA(f, check_date=fut[f].loc[i, 'date'], timeperiod=5)
+#         MA1 = MA1[f]
+#         fut[f].loc[i, 'MA5'] = MA1
+#         # print(fut[f].loc[i, ['close','high','low']])
+#         if fut[f].loc[i, 'close'] > MA1:
+#             derta = fut[f].loc[i, 'high']/MA1 - 1
+#         else:
+#             derta = (fut[f].loc[i, 'low']/MA1 -1)
+#         fut[f].loc[i, 'derta'] = derta
+#     print('___________________________________')
+#     print(fut[f])
+#     fut[f].to_excel('/Users/daniel/Downloads/MA_derta/30m_%s.xlsx'% f)
+#
+# end= dt.now()
+# print('总时长：', (end - start).seconds)
+
+
+
+
+
+
+
+
+
+# MA1 = MA('MA9999.XZCE', check_date='2022-09-25', timeperiod=5)
+# print(MA1['MA9999.XZCE'])
+

factor_values.py

@@ -0,0 +1,68 @@
+from jqdatasdk import *
+from jqdatasdk.technical_analysis import *
+from sqlalchemy import create_engine
+import pandas as pd
+import pymysql
+import datetime
+
+
+auth('18616891214', 'Ea?*7f68nD.dafcW34d!')
+strattime = datetime.datetime.now()
+print(get_query_count())
+
+stocks = list(get_all_securities(['stock']).index)
+qihuo = get_price(['IF9999.CCFX', 'IC9999.CCFX','IH9999.CCFX', 'IM9999.CCFX'], start_date='2015-01-01', end_date='2015-12-31', frequency='daily', fields=None, skip_paused=False, fq='pre' ,panel=True)
+
+print(qihuo)
+for fre in ['30m', '1d']:
+    print(fre)
+    # 连接数据库
+    db = pymysql.connect(host='localhost',
+                         user='root',
+                         port=3307,
+                         password='r6kEwqWU9!v3',
+                         database='hlfx')
+    cursor = db.cursor()
+    cursor.execute("show tables like '%%%s%%' " % fre)
+    table_list = [tuple[0] for tuple in cursor.fetchall()]
+    print('取得 table_list %s' % fre)
+
+time = datetime.datetime(2010, 1, 4)
+print(time)
+
+for stock in table_list:
+    print(stock)
+    sql = ("select date_format(date, '%%Y-%%m-%%d') from `%s` where HL='L'" % stock)
+    cheak_time = cursor.execute(sql)
+    print(cheak_time)
+    time_list = [''.join(time) for time in cursor.fetchall()]
+    print(time_list)
+
+    break
+    #stock = normalize_code(stock[3:9])
+
+
+# 定义股票池列表
+security_list1 = '000001.XSHE'
+# security_list2 = ['000001.XSHE','000002.XSHE','601211.XSHG','603177.XSHG']
+# # 计算并输出 security_list1 的 MACD 值
+macd_dif, macd_dea, macd_macd = MACD(security_list1, check_date=time_list[1], SHORT = 12, LONG = 26, MID = 9)
+print(macd_dif[security_list1])
+print(macd_dea[security_list1])
+print(macd_macd[security_list1])
+
+# 输出 security_list2 的 MACD 值
+macd_dif, macd_dea, macd_macd = MACD(security_list2,check_date=datetime.datetime.today(), SHORT = 12, LONG = 26, MID = 9)
+for stock in security_list2:
+    print(macd_dif[stock])
+    print(macd_dea[stock])
+    print(macd_macd[stock])
+
+df = get_bars(stocks, count=10, unit='30m',
+              fields=['date','open','close','high','low','volume','money'],include_now=False,end_dt=datetime.date.today())
+print(df)
+endtime = datetime.datetime.now()
+print('单次时长为：', (endtime - strattime).seconds)
+
+
+#get_ticks("000001.XSHE", "2022-01-01", datetime.datetime.today())

get_history_futures.py

@@ -0,0 +1,34 @@
+from jqdatasdk import *
+auth('18616891214','Ea?*7f68nD.dafcW34d!')
+from sqlalchemy import create_engine
+import pandas as pd
+import datetime
+
+engine = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/hlfx_pool?charset=utf8')
+fre = '1d'
+pool =pd.read_sql_table('%s' % fre, con=engine).iloc[-1,2]
+stock_pool = pool.split(",")
+print(type(pool),pool)
+print(item for item in pool)
+print(type(stock_pool),stock_pool)
+print(item for item in stock_pool)
+
+
+# futures = ['IF9999.CCFX', 'IC9999.CCFX','IH9999.CCFX', 'IM9999.CCFX', 'FU9999.XSGE', 'RB9999.XSGE', 'MA9999.XZCE', 'TA9999.XZCE', 'SA9999.XZCE', 'M9999.XDCE', 'LH9999.XDCE']
+#
+# for fre in ['30m','1d','5m']:
+#     for future in futures:
+#         df_future = get_price(future, start_date='2015-01-01', end_date='2022-09-25', frequency=fre, fields=None, skip_paused=False, fq='pre' ,panel=True)
+#
+#         # 去除无数据日
+#         df_future = df_future.dropna(axis=0)
+#         # 重置index
+#         df_future.reset_index(inplace=True)
+#         # 保留日期数据
+#         df_future.rename(columns={'index': 'date'}, inplace=True)
+#         print(df_future)
+#         # 写入数据库
+#         df_future.to_sql('%s_%s' % (future, fre), con=engine, if_exists='append')
+#         with engine.connect() as con:
+#             con.execute("ALTER TABLE `%s_%s` ADD PRIMARY KEY (`date`);" % (future, fre))
+

real_time_signal.py

@@ -24,7 +24,7 @@ cursor = db.cursor()
 cursor.execute("show tables like '%%%s%%' "% fre)
 # stocks = [tuple[0] for tuple in cursor.fetchall()]
 stocks = list(get_all_securities(['stock'], date='2021-12-31').index)
-stocks = ['301058.XSHE']
+# stocks = ['301058.XSHE']
 # stocks = stocks[0:500]
 print(dt.now(), 'stocks范围已获取！')
 

updata_qbh_hlfx.py

@@ -4,18 +4,16 @@ import pymysql
 from sqlalchemy import create_engine
 import threading
 from datetime import datetime as dt
-import datetime
+from jqdatasdk.technical_analysis import *
 
-auth('18019403367', 'Qwer4321')
+auth('18616891214', 'Ea?*7f68nD.dafcW34d!')
 
 stocks = list(get_all_securities(['stock'], date=dt.today().strftime('%Y-%m-%d')).index)
-# stocks = stocks[0:1]
+# stocks = stocks[0:200]
 
 start = dt.now()
-# 确定级别
-# 注意修改time delta
-# fre = '30m'
-for fre in ['30m', '1d']:
+
+for fre in ['1d']:
     start = dt.now()
     print(fre)
     # 连接数据库
@@ -29,23 +27,32 @@ for fre in ['30m', '1d']:
     table_list = [tuple[0] for tuple in cursor.fetchall()]
     print('取得 table_list %s' % fre)
 
+    db_pool = pymysql.connect(host='localhost',
+                              user='root',
+                              port=3307,
+                              password='r6kEwqWU9!v3',
+                              database='hlfx_pool')
+    cursor_pool = db_pool.cursor()
+
+
     stk = locals()
     thd = threading.local()
 
-    def hlfx(stocks, engine, engine2):
+    def hlfx(stocks, engine_stock, engine_hlfx):
         for thd.stock in stocks:
             print(thd.stock)
             if ('stk%s_%s' % (thd.stock, fre)) in table_list:
                 # 有历史数据
-                index_len = pd.read_sql_table('stk%s_%s' % (thd.stock, fre), con=engine2).iloc[-1, 0]
-                # 注意修改time delta
-                startdate = pd.read_sql_table('stk%s_%s' % (thd.stock, fre), con=engine2).iloc[-1, 1]
-                # startdate = pd.read_sql_table('stk%s_%s' % (stock, fre), con=engine2).iloc[-1, 1] + datetime.timedelta(minutes= 5)
+                index_len = pd.read_sql_table('stk%s_%s' % (thd.stock, fre), con=engine_hlfx).iloc[-1, 0]
+                startdate = pd.read_sql_table('stk%s_%s' % (thd.stock, fre), con=engine_hlfx).iloc[-1, 1]
+
                 thd.get_price = pd.read_sql_query(
-                    'select date,open,close,high,low,volume,money from `stk%s_%s`' % (thd.stock, fre), engine)
+                    'select date,open,close,high,low,volume,money from `stk%s_%s`' % (thd.stock, fre), engine_stock)
                 thd.get_price = thd.get_price.loc[thd.get_price['date'] > startdate]
+
                 thd.df_day = pd.read_sql_query(
-                    'select date,open,close,high,low,volume,money,HL from `stk%s_%s`' % (thd.stock, fre), engine2)
+                    'select date,open,close,high,low,volume,money,HL from `stk%s_%s`' % (thd.stock, fre), engine_hlfx)
+
                 # 先处理去包含
                 for i in thd.get_price.index:
                     # 不包含
@@ -54,7 +61,7 @@ for fre in ['30m', '1d']:
                             or (thd.df_day.iloc[-1, 3] < thd.get_price.loc[i, 'high']
                                 and thd.df_day.iloc[-1, 4] < thd.get_price.loc[i, 'low']):
                         thd.df_day = pd.concat([thd.df_day, thd.get_price.loc[[i]]], ignore_index=True)
-                        # print(thd.df_day)
+
                     # 包含
                     else:
                         # (new_df.iloc[-1,3]>=df_day.loc[i,'high'] and new_df.iloc[-1,4]<= df_day.loc[i,'low']):
@@ -66,9 +73,9 @@ for fre in ['30m', '1d']:
                             # 右高，上升
                             thd.df_day.iloc[-1, 3] = max(thd.df_day.iloc[-1, 3], thd.get_price.loc[i, 'high'])
                             thd.df_day.iloc[-1, 4] = max(thd.df_day.iloc[-1, 4], thd.get_price.loc[i, 'low'])
-                            # 寻找顶底分型
+
+                # 寻找顶底分型
                 if len(thd.df_day.index) > 2:
-                    # 寻找顶底分型
                     for x in range(index_len, len(thd.df_day.index)):
                         m = x - 1
                         # 底
@@ -81,22 +88,41 @@ for fre in ['30m', '1d']:
                                 if thd.df_day.loc[m, 'HL'] == 'H':
                                     if (x - m) > 3:
                                         thd.df_day.loc[x, 'HL'] = 'L'
-                                        if x == len(thd.df_day.index) - 1:
-                                            print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'], '买买买买买！！')
-                                    break
+                                        # 此处可以获得MACD指标
+                                        # pre-macd_dif, pre-macd_dea, pre-macd_macd = MACD(thd.stock,check_date=thd.df_day.loc[m, 'datetime'], SHORT = 12, LONG = 26, MID = 9)
+                                        if x == len(thd.df_day.index) - 1 :
+                                            # pass
+                                            print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'], '\n', thd.df_day.loc[m, 'date'], '买买买买买！！')
+                                            results.append(thd.stock)
+                                            print('222')
+                                    # break
                                 elif (thd.df_day.loc[m, 'HL'] == 'L'):
                                     if thd.df_day.loc[x - 1, 'low'] < thd.df_day.loc[m - 1, 'low']:
                                         # 前一个为底，且中间存在不包含 or 更低的底
                                         thd.df_day.loc[x, 'HL'] = 'L'
-                                        if x == len(thd.df_day.index) - 1:
+                                        x_macd_dif, x_macd_dea, x_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[x, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        m_macd_dif, m_macd_dea, m_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[m, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        if x == len(thd.df_day.index) - 1 and x_macd_dif[thd.stock] > m_macd_dif[thd.stock]:
                                             # pass
-                                            print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'], '中继后的底————买吗？！')
+                                            # print(thd.df_day.loc[m, 'date'], thd.df_day.loc[m, 'low'],
+                                            #       m_macd_dif[thd.stock])
+                                            # print(thd.df_day.loc[x, 'date'], thd.df_day.loc[x, 'low'],
+                                            #       x_macd_dif[thd.stock])
+                                            print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'], 'MACD背驰————买吗？！')
+                                            results.append(thd.stock)
+                                            print('333')
                                         break
                                     else:
                                         break
                                 m = m - 1
                                 if m == 0:
                                     thd.df_day.loc[x, 'HL'] = 'L'
+                                    results.append(thd.stock)
+                                    print('444')
                         # 顶
                         elif ((thd.df_day.loc[x, 'high'] < thd.df_day.loc[x - 1, 'high']) and (
                                 thd.df_day.loc[x - 2, 'high'] < thd.df_day.loc[x - 1, 'high'])):
@@ -108,32 +134,47 @@ for fre in ['30m', '1d']:
                                     if x - m > 3:
                                         thd.df_day.loc[x, 'HL'] = 'H'
                                         if x == len(thd.df_day.index) - 1:
-                                            # print(stock, '!!!!!!!', '\n', '卖卖卖卖卖卖卖!')
-                                            pass
-                                        thd.df_day.loc[x, 9] = thd.df_day.loc[x, 'close'] - thd.df_day.loc[m, 'close']
-                                    break
+                                            print(thd.stock, '!!!!!!!', '\n', '卖卖卖卖卖卖卖!')
+                                            # pass
+                                            results_short.append(thd.stock)
+                                            if thd.stock in results:
+                                                results.remove(thd.stock)
+                                    # break
                                 elif (thd.df_day.loc[m, 'HL'] == 'H'):
                                     if thd.df_day.loc[x - 1, 'high'] > thd.df_day.loc[m - 1, 'high']:
                                         # 前一个为顶，且中间存在不包含 or 更高的顶
                                         thd.df_day.loc[x, 'HL'] = 'H'
-                                        if x == len(thd.df_day.index) - 1:
-                                            pass
-                                            # print(stock, '/\/\/\/\/\/\/', '一顶更有一顶高！')
+                                        x_macd_dif, x_macd_dea, x_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[x, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        m_macd_dif, m_macd_dea, m_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[m, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        if x == len(thd.df_day.index) - 1 and x_macd_dif[thd.stock] < m_macd_dif[thd.stock]:
+                                            # pass
+                                            print(thd.stock, '/\/\/\/\/\/\/', '顶背离了！！！！')
+                                            results_short.append(thd.stock)
+                                            if thd.stock in results:
+                                                results.remove(thd.stock)
                                         break
                                     break
                                 m = m - 1
                                 if m == 0:
                                     thd.df_day.loc[x, 'HL'] = 'H'
+                                    results_short.append(thd.stock)
+                                    if thd.stock in results:
+                                        results.remove(thd.stock)
                         else:
                             thd.df_day.loc[x, 'HL'] = '-'
 
                 # 更新数据库
-                thd.df_day[index_len + 1:].to_sql('stk%s_%s' % (thd.stock, fre), con=engine2, index=True, if_exists='append')
+                # 可以使用normalize_code(code) 方法 改变代码格式
+                # thd.df_day[index_len + 1:].to_sql('stk%s_%s' % (thd.stock, fre), con=engine_hlfx, index=True, if_exists='append')
             else:
                 # 没有历史数据表
                 thd.df_day = pd.DataFrame(columns=('date', 'open', 'close', 'high', 'low', 'volume', 'money', 'HL'))
                 thd.get_price = pd.read_sql_query(
-                    'select date,open,close,high,low,volume,money from `stk%s_%s`' % (thd.stock, fre), engine)
+                    'select date,open,close,high,low,volume,money from `stk%s_%s`' % (thd.stock, fre), engine_stock)
                 # 先处理去包含
                 for i in thd.get_price.index:
                     if i == 0 or i == 1:
@@ -168,22 +209,42 @@ for fre in ['30m', '1d']:
                                 if thd.df_day.loc[m, 'HL'] == 'H':
                                     if (x - m) > 3:
                                         thd.df_day.loc[x, 'HL'] = 'L'
+                                        # 此处可以获得MACD指标
+                                        # pre-macd_dif, pre-macd_dea, pre-macd_macd = MACD(thd.stock,check_date=thd.df_day.loc[m, 'datetime'], SHORT = 12, LONG = 26, MID = 9)
+
                                         if x == len(thd.df_day.index) - 1:
+                                            # pass
                                             print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'], '买买买买买！！')
-                                    break
+                                            results.append(thd.stock)
+                                    # break
                                 elif (thd.df_day.loc[m, 'HL'] == 'L'):
                                     if thd.df_day.loc[x - 1, 'low'] < thd.df_day.loc[m - 1, 'low']:
                                         # 前一个为底，且中间存在不包含 or 更低的底
                                         thd.df_day.loc[x, 'HL'] = 'L'
-                                        if x == len(thd.df_day.index) - 1:
+                                        x_macd_dif, x_macd_dea, x_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[x, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        m_macd_dif, m_macd_dea, m_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[m, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        if x == len(thd.df_day.index) - 1 and x_macd_dif[thd.stock] > m_macd_dif[
+                                            thd.stock]:
                                             # pass
-                                            print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'], '中继后的底————买吗？！')
+                                            # print(thd.df_day.loc[m, 'date'], thd.df_day.loc[m, 'low'],
+                                            #       m_macd_dif[thd.stock])
+                                            # print(thd.df_day.loc[x, 'date'], thd.df_day.loc[x, 'low'],
+                                            #       x_macd_dif[thd.stock])
+                                            print(thd.stock, '$$$$$$$', '\n', thd.df_day.loc[x, 'date'],
+                                                  'MACD背驰————买吗？！')
+                                            results.append(thd.stock)
+
                                         break
                                     else:
                                         break
                                 m = m - 1
                                 if m == 0:
                                     thd.df_day.loc[x, 'HL'] = 'L'
+                                    results.append(thd.stock)
                         # 顶
                         elif ((thd.df_day.loc[x, 'high'] < thd.df_day.loc[x - 1, 'high']) and (
                                 thd.df_day.loc[x - 2, 'high'] < thd.df_day.loc[x - 1, 'high'])):
@@ -195,44 +256,76 @@ for fre in ['30m', '1d']:
                                     if x - m > 3:
                                         thd.df_day.loc[x, 'HL'] = 'H'
                                         if x == len(thd.df_day.index) - 1:
-                                            # print(stock, '!!!!!!!', '\n', '卖卖卖卖卖卖卖!')
-                                            pass
-                                        thd.df_day.loc[x, 9] = thd.df_day.loc[x, 'close'] - thd.df_day.loc[m, 'close']
-                                    break
+                                            print(thd.stock, '!!!!!!!', '\n', '卖卖卖卖卖卖卖!')
+                                            # pass
+                                            results.remove(thd.stock)
+                                    # break
                                 elif (thd.df_day.loc[m, 'HL'] == 'H'):
                                     if thd.df_day.loc[x - 1, 'high'] > thd.df_day.loc[m - 1, 'high']:
                                         # 前一个为顶，且中间存在不包含 or 更高的顶
                                         thd.df_day.loc[x, 'HL'] = 'H'
-                                        if x == len(thd.df_day.index) - 1:
-                                            pass
-                                            # print(stock, '/\/\/\/\/\/\/', '一顶更有一顶高！')
+                                        x_macd_dif, x_macd_dea, x_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[x, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        m_macd_dif, m_macd_dea, m_macd_macd = MACD(thd.stock,
+                                                                                   check_date=thd.df_day.loc[m, 'date'],
+                                                                                   SHORT=12, LONG=26, MID=9, unit=fre)
+                                        if x == len(thd.df_day.index) - 1 and x_macd_dif[thd.stock] < m_macd_dif[
+                                            thd.stock]:
+                                            # pass
+                                            print(thd.stock, '/\/\/\/\/\/\/', '顶背离了！！！！')
+                                            results.remove(thd.stock)
                                         break
                                     break
                                 m = m - 1
                                 if m == 0:
                                     thd.df_day.loc[x, 'HL'] = 'H'
+                                    results.remove(thd.stock)
                         else:
                             thd.df_day.loc[x, 'HL'] = '-'
+            print(thd.df_day[-20:])
                 # 更新数据库
-                thd.df_day.to_sql('stk%s_%s' % (thd.stock, fre), con=engine2, index=True, if_exists='append')
+                # thd.df_day.to_sql('stk%s_%s' % (thd.stock, fre), con=engine_hlfx, index=True, if_exists='append')
 
     step = 500
     thread_list = []
-    engine = []
-    engine2 = []
+    engine_stock = []
+    engine_hlfx = []
     times_engine = 0
-    for i in range(0, len(stocks), step):
-        engine.append(create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/stocks?charset=utf8'))
-        engine2.append(create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/hlfx?charset=utf8'))
-        thread = threading.Thread(target=hlfx, args=(stocks[i:i + step], engine[times_engine], engine2[times_engine]))
-        times_engine = times_engine + 1
-        thread.start()
-        thread_list.append(thread)
-
-
-    for thread in thread_list:
-        thread.join()
-    db.close()
+    engine_hlfx_pool = create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/hlfx_pool?charset=utf8')
+
+    results= pd.read_sql_query(
+                    'select value from `%s`' % fre, engine_hlfx_pool)
+
+    results = results.iloc[-1, 0]
+    results = results.split(",")
+    results_short=[]
+
+    print('数据库读取', results, type(results))
+    # for i in range(0, len(stocks), step):
+    #     engine_stock.append(create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/stocks?charset=utf8'))
+    #     engine_hlfx.append(create_engine('mysql+pymysql://root:r6kEwqWU9!v3@localhost:3307/hlfx?charset=utf8'))
+    #     thread = threading.Thread(target=hlfx, args=(stocks[i:i + step], engine_stock[times_engine], engine_hlfx[times_engine]))
+    #     times_engine = times_engine + 1
+    #     thread.start()
+    #     thread_list.append(thread)
+    #
+    # for thread in thread_list:
+    #     thread.join()
+    # db.close()
+
+
+
+    time = dt.now().strftime('%Y-%m-%d %H:%M:%S')
+    results_list =','.join(set(results))
+    print(set(results))
+    # sql = "INSERT INTO %s (date,value) VALUES('%s','%s')" % (fre, dt.now().strftime('%Y-%m-%d %H:%M:%S'), results_list)
+    # cursor_pool.execute(sql)
+    # db_pool.commit()
+    print('做多', set(results))
+    print('做空', set(results_short))
+    print(len(set(results)))
+    print(len(set(results_short)))
 
     end= dt.now()
     print('总时长：', (end - start).seconds)

update_data_tosql.py

@@ -14,7 +14,6 @@ for fre in ['30m', '1d']:
         print(stock)
         try:
             index_len = pd.read_sql_table('stk%s_%s' % (stock, fre), con=engine).iloc[-1, 0]
-            # 注意修改time delta
             if fre == '1d':
                 startdate = pd.read_sql_table('stk%s_%s' % (stock, fre), con=engine).iloc[-1, 1] + datetime.timedelta(
                     days=1)