mns-scheduler 1.4.3.2__py3-none-any.whl → 1.4.5.7__py3-none-any.whl

mns_scheduler/irm/stock_question_id_service.py

@@ -0,0 +1,169 @@
+import sys
+import os
+
+file_path = os.path.abspath(__file__)
+end = file_path.index('mns') + 16
+project_path = file_path[0:end]
+sys.path.append(project_path)
+
+"""
+"""
+
+from functools import lru_cache
+import pandas as pd
+import requests
+from bs4 import BeautifulSoup
+from mns_common.db.MongodbUtil import MongodbUtil
+import mns_common.constant.db_name_constant as db_name_constant
+import mns_common.component.common_service_fun_api as common_service_fun_api
+import mns_common.component.em.em_stock_info_api as em_stock_info_api
+import mns_common.component.company.company_common_service_api as company_common_service_api
+from loguru import logger
+import mns_common.utils.data_frame_util as data_frame_util
+from datetime import datetime, timedelta
+
+mongodb_util = MongodbUtil('27017')
+from tqdm import tqdm
+
+
+# 同步所有映射表
+@lru_cache()
+def sync_stock_uid() -> pd.DataFrame:
+    """
+    上证e互动-代码ID映射
+    https://sns.sseinfo.com/list/company.do
+    :return: 代码ID映射
+    :rtype: str
+    """
+    url = "https://sns.sseinfo.com/allcompany.do"
+    data = {
+        "code": "0",
+        "order": "2",
+        "areaId": "0",
+        "page": "1",
+    }
+    uid_list = list()
+    code_list = list()
+    for page in tqdm(range(1, 74), leave=False):
+        data.update({"page": page})
+        r = requests.post(url, data=data)
+        data_json = r.json()
+        soup = BeautifulSoup(data_json["content"], "lxml")
+        soup.find_all("a", attrs={"rel": "tag"})
+        uid_list.extend(
+            [item["uid"] for item in soup.find_all("a", attrs={"rel": "tag"})]
+        )
+        code_list.extend(
+            [
+                item.find("img")["src"].split("?")[0].split("/")[-1].split(".")[0]
+                for item in soup.find_all("a", attrs={"rel": "tag"})
+            ]
+        )
+    code_uid_df = pd.DataFrame()
+    code_uid_df['symbol'] = code_list
+    code_uid_df['uid'] = uid_list
+    code_uid_df['_id'] = uid_list
+    return code_uid_df
+
+
+# 获取上海问答id
+@lru_cache()
+def get_sh_stock_all_uid() -> pd.DataFrame:
+    return mongodb_util.find_all_data(db_name_constant.SH_INFO_UID)
+
+
+# 获取深圳问答id
+@lru_cache()
+def get_sz_stock_all_uid() -> pd.DataFrame:
+    return mongodb_util.find_all_data(db_name_constant.SZ_INFO_UID)
+
+
+# 获取深圳互动回答单个ID
+def get_one_sz_symbol_org_id(symbol):
+    sz_info_uid_df = get_sz_stock_all_uid()
+    sz_info_uid_one_df = sz_info_uid_df.loc[sz_info_uid_df['symbol'] == symbol]
+    if data_frame_util.is_not_empty(sz_info_uid_one_df):
+        return list(sz_info_uid_one_df['uid'])[0]
+    else:
+        try:
+            return fetch_sz_org_id(symbol)
+        except BaseException as e:
+            logger.error("获取组织代码异常:{},{}", symbol, e)
+            return '0'
+
+
+# 获取上海互动回答单个ID
+def get_one_sh_symbol_org_id(symbol):
+    sh_info_uid_df = get_sh_stock_all_uid()
+    sh_info_uid_one_df = sh_info_uid_df.loc[sh_info_uid_df['symbol'] == symbol]
+    if data_frame_util.is_not_empty(sh_info_uid_one_df):
+        return list(sh_info_uid_one_df['uid'])[0]
+    else:
+        return '0'
+
+
+# 深圳股票-互动易-组织代码 单个获取
+def fetch_sz_org_id(symbol: str = "000001") -> str:
+    """
+    股票-互动易-组织代码
+    https://irm.cninfo.com.cn/
+    :return: 组织代码
+    :rtype: str
+    """
+    url = "https://irm.cninfo.com.cn/newircs/index/queryKeyboardInfo"
+    params = {"_t": "1691144074"}
+    data = {"keyWord": symbol}
+    r = requests.post(url, params=params, data=data)
+    data_json = r.json()
+    org_id = data_json["data"][0]["secid"]
+    return org_id
+
+
+# 同步上证互动uid
+def sync_sh_stock_uid():
+    code_uid_df = sync_stock_uid()
+    mongodb_util.save_mongo(code_uid_df, db_name_constant.SH_INFO_UID)
+
+
+# 同步深圳互动uid
+def sync_sz_stock_uid(symbol_list):
+    real_time_quotes_all_stocks = em_stock_info_api.get_a_stock_info()
+    de_list_company_symbols = company_common_service_api.get_de_list_company()
+    real_time_quotes_all_stocks = real_time_quotes_all_stocks.loc[
+        ~(real_time_quotes_all_stocks['symbol'].isin(de_list_company_symbols))]
+    real_time_quotes_all_stocks = common_service_fun_api.classify_symbol(real_time_quotes_all_stocks)
+
+    real_time_quotes_all_stocks = real_time_quotes_all_stocks.loc[
+        real_time_quotes_all_stocks['classification'].isin(['S', 'C'])]
+    if len(symbol_list) != 0:
+        real_time_quotes_all_stocks = real_time_quotes_all_stocks.loc[
+            real_time_quotes_all_stocks['symbol'].isin(symbol_list)]
+    else:
+        # 获取当前时间
+        now_date = datetime.now()
+        # 计算前7天的时间（timedelta 用于表示时间间隔）
+        seven_days_ago = now_date - timedelta(days=30)
+        str_day_number = int(seven_days_ago.strftime('%Y%m%d'))
+        real_time_quotes_all_stocks = real_time_quotes_all_stocks.loc[
+            (real_time_quotes_all_stocks['list_date'] >= str_day_number) | (
+                    real_time_quotes_all_stocks['list_date'] == 19890604)]
+    for stock_one in real_time_quotes_all_stocks.itertuples():
+        try:
+            symbol = stock_one.symbol
+            uid = fetch_sz_org_id(symbol)
+            result_dict = {'_id': symbol, 'symbol': symbol, 'uid': uid}
+            result_dict_df = pd.DataFrame(result_dict, index=[1])
+            mongodb_util.save_mongo(result_dict_df, db_name_constant.SZ_INFO_UID)
+            logger.info("同步SZ互动ID:{}", stock_one.symbol)
+        except Exception as e:
+            logger.error("同步SZ互动ID异常:{},{}", stock_one.symbol, e)
+
+
+if __name__ == '__main__':
+    sz_symbol_org_id = get_one_sz_symbol_org_id('300085')
+    sh_symbol_org_id = get_one_sh_symbol_org_id('600000')
+    print(sz_symbol_org_id)
+    print(sh_symbol_org_id)
+    sync_sz_stock_uid([])
+    # sync_sz_stock_uid([])
+    # sync_sh_stock_uid([])

mns_scheduler/self_choose/ths_self_choose_service.py

@@ -17,6 +17,7 @@ import mns_common.component.trade_date.trade_date_common_service_api as trade_da
 
 mongodb_util = MongodbUtil('27017')
 
+
 # 固定的选择
 # fixed_optional_list = ['USDCNH', 'XAUUSD',
 #                        '881279',
@@ -37,17 +38,16 @@ mongodb_util = MongodbUtil('27017')
 #                        ]
 
 # 固定的选择
-fixed_optional_list = ['899050', '881157']
-
+# fixed_optional_list = ['899050', '881157']
 
-def add_fixed_optional():
-    ths_cookie = cookie_info_service.get_ths_cookie()
-    for symbol in fixed_optional_list:
-        ths_self_choose_api.add_stock_to_account(symbol, ths_cookie)
+#
+# def add_fixed_optional():
+#     ths_cookie = cookie_info_service.get_ths_cookie()
+#     for symbol in fixed_optional_list:
+#         ths_self_choose_api.add_stock_to_account(symbol, ths_cookie)
 
-
-def delete_all_self_choose_stocks():
-    ths_cookie = cookie_info_service.get_ths_cookie()
+# 删除所有自选股票
+def delete_all_self_choose_stocks(ths_cookie):
     all_self_choose_stock_list = ths_self_choose_api.get_all_self_choose_stock_list(ths_cookie)
     for stock_one in all_self_choose_stock_list.itertuples():
         symbol = stock_one.code
@@ -55,8 +55,7 @@ def delete_all_self_choose_stocks():
 
 
 # 添加固定自选股票
-def add_self_choose_symbol():
-    ths_cookie = cookie_info_service.get_ths_cookie()
+def add_self_choose_symbol(ths_cookie):
     # 固定自选
     self_choose_symbol_df = mongodb_util.find_all_data(db_name_constant.SELF_CHOOSE_STOCK)
     if data_frame_util.is_not_empty(self_choose_symbol_df):
@@ -66,8 +65,7 @@ def add_self_choose_symbol():
 
 
 # 添加今日选择股票
-def add_today_choose_symbol():
-    ths_cookie = cookie_info_service.get_ths_cookie()
+def add_today_choose_symbol(ths_cookie):
     now_date = datetime.now()
     str_day = now_date.strftime('%Y-%m-%d')
     last_trade_day = trade_date_common_service_api.get_last_trade_day(str_day)
@@ -81,8 +79,7 @@ def add_today_choose_symbol():
 
 
 # 添加同花顺概念
-def add_self_choose_concept():
-    ths_cookie = cookie_info_service.get_ths_cookie()
+def add_self_choose_concept(ths_cookie):
     query_plate = {'self_type': {
         "$in": [self_choose_constant.SELF_CHOOSE_THS_CONCEPT,
                 self_choose_constant.SELF_CHOOSE_THS_INDUSTRY]}}
@@ -95,7 +92,7 @@ def add_self_choose_concept():
 
 
 # 添加最近交易股票
-def add_trade_stocks():
+def add_trade_stocks(ths_cookie):
     now_date = datetime.now()
     str_day = now_date.strftime('%Y-%m-%d')
     last_trade_day = trade_date_common_service_api.get_last_trade_day(str_day)
@@ -109,15 +106,13 @@ def add_trade_stocks():
     trade_stocks_df = mongodb_util.find_query_data(db_name_constant.BUY_STOCK_NAME, query)
     if data_frame_util.is_not_empty(trade_stocks_df):
         stock_list = stock_list.union(set(trade_stocks_df['symbol']))
-    ths_cookie = cookie_info_service.get_ths_cookie()
     if len(stock_list) > 0:
         for symbol in stock_list:
             ths_self_choose_api.add_stock_to_account(symbol, ths_cookie)
 
 
 # 添加连板到自选
-def add_continue_boards_zt_stocks():
-    ths_cookie = cookie_info_service.get_ths_cookie()
+def add_continue_boards_zt_stocks(ths_cookie):
     now_date = datetime.now()
     str_day = now_date.strftime('%Y-%m-%d')
     if trade_date_common_service_api.is_trade_day(str_day):
@@ -138,19 +133,52 @@ def add_continue_boards_zt_stocks():
         ths_self_choose_api.add_stock_to_account(stock_one.symbol, ths_cookie)
 
 
+# 添加主线龙头
+def add_main_line_leader(ths_cookie):
+    main_line_leader_df = mongodb_util.find_query_data('main_line_leader', {})
+    if data_frame_util.is_not_empty(main_line_leader_df):
+        main_line_leader_df = count_main_line_number(main_line_leader_df)
+
+        main_line_leader_df = main_line_leader_df.sort_values(
+            by=['main_line_number', 'main_line_leader_grade'],
+            ascending=[False, False,]
+        )
+        for stock_one in main_line_leader_df.itertuples():
+            ths_self_choose_api.add_stock_to_account(stock_one.symbol, ths_cookie)
+
+
+# 统计主线的数量
+def count_main_line_number(main_line_detail_df):
+    # 统计主线的数量
+    main_line_detail_df['main_line_number'] = main_line_detail_df.groupby('main_line')['main_line'].transform(
+        'count')
+    # 相同主线数量 数量加1
+    main_line_detail_df['main_line_number'] = (
+                                                      main_line_detail_df.groupby('main_line_number')['main_line']
+                                                      .transform('rank', method='dense') - 1
+                                              ) * 0.1 + main_line_detail_df['main_line_number']
+
+    return main_line_detail_df
+
+
 # 自选股操作 删除当天自选股 增加新的连板股票  添加固定选择自选
 def self_choose_stock_handle():
-    delete_all_self_choose_stocks()
+    ths_cookie = cookie_info_service.get_ths_cookie()
+
+    delete_all_self_choose_stocks(ths_cookie)
     # 固定自选板块
-    add_fixed_optional()
+    # add_fixed_optional()
     # 添加同花顺概念
-    add_self_choose_concept()
+    add_self_choose_concept(ths_cookie)
+    # 添加主线龙头
+    add_main_line_leader(ths_cookie)
     # 连板股票
-    add_continue_boards_zt_stocks()
+    add_continue_boards_zt_stocks(ths_cookie)
+
     # 自己买入的股票
-    add_trade_stocks()
+    add_trade_stocks(ths_cookie)
     # 添加自选股票
-    add_self_choose_symbol()
+    add_self_choose_symbol(ths_cookie)
 
 
 if __name__ == '__main__':

mns_scheduler/trade/auto_login/trader_auto_service.py

@@ -10,6 +10,7 @@ from mns_common.component.deal.terminal_enum import TerminalEnum
 import time
 from loguru import logger
 
+
 # qmt 自动登录
 def qmt_auto_login():
     deal_service_api.auto_login(TerminalEnum.QMT.terminal_code)
@@ -29,4 +30,4 @@ def auto_login():
 
 
 if __name__ == '__main__':
-    auto_login()
+    qmt_auto_login()

mns_scheduler/trade/task/trader_task_service.py

@@ -17,11 +17,22 @@ TRADER_SERVER_PATH = 'H:\\mns_trader_server.bat'
 TRADER_SERVER_NAME = "mns_trader_server"
 
 
+# 打开qmt
+def open_qmt_terminal():
+    logger.info('打开 [交易web服务端]')
+    if bool(1 - is_open_trader_server()):
+        cmd_util.open_bat_file(TRADER_SERVER_PATH)
+    else:
+        logger.info('交易服务端已经运行')
+    logger.info('自动登陆qmt')
+    trader_auto_service.qmt_auto_login()
+
+
 # 打开交易客户端
 def open_trader_terminal():
     # 打开 [交易web服务端]
     logger.info('打开 [交易web服务端]')
-    if bool(1- is_open_trader_server()):
+    if bool(1 - is_open_trader_server()):
         cmd_util.open_bat_file(TRADER_SERVER_PATH)
     time.sleep(10)
     logger.info('自动登陆ths和qmt')
@@ -62,4 +73,4 @@ def kill_server():
 
 if __name__ == '__main__':
     # kill_server()
-    open_trader_terminal()
+    open_qmt_terminal()

mns_scheduler/trade/tfp/stock_tfp_info_sync.py

@@ -52,5 +52,5 @@ def sync_stock_tfp(str_day):
 
 
 if __name__ == '__main__':
-    df = sync_stock_tfp('2025-07-30')
+    df = sync_stock_tfp('2025-12-25')
     print(df)

mns_scheduler/zb/stock_zb_pool_sync.py

@@ -27,4 +27,4 @@ def sync_stock_zb_pool(str_now_day):
 
 
 if __name__ == '__main__':
-    sync_stock_zb_pool('2025-04-18')
+    sync_stock_zb_pool('2025-12-11')

mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py

@@ -18,10 +18,11 @@ import mns_common.api.ths.zt.ths_stock_zt_pool_v2_api as ths_stock_zt_pool_v2_ap
 import mns_common.component.zt.zt_common_service_api as zt_common_service_api
 import mns_common.component.em.em_real_time_quotes_api as em_real_time_quotes_api
 from datetime import datetime
-import mns_common.api.ths.zt.ths_stock_zt_pool_api as ths_stock_zt_pool_api
 import mns_common.constant.db_name_constant as db_name_constant
 import mns_common.component.deal.deal_service_api as deal_service_api
 import mns_common.component.company.company_common_service_new_api as company_common_service_new_api
+import mns_common.component.main_line.main_line_zt_reason_service as main_line_zt_reason_service
+from mns_common.utils.async_fun import async_fun
 
 '''
 东方财富涨停池
@@ -63,10 +64,12 @@ def save_zt_info(str_day):
         # 同花顺问财涨停池
         ths_zt_pool_df_data = ths_stock_zt_pool_v2_api.get_ths_stock_zt_reason_with_cache(str_day)
     except BaseException as e:
-        logger.error("同步ths涨停数据异常")
+        logger.error("使用问财同步ths涨停数据异常:{}", e)
         ths_zt_pool_df_data = pd.DataFrame()
 
     stock_em_zt_pool_df_data = handle_ths_em_diff_data(ths_zt_pool_df_data, stock_em_zt_pool_df_data)
+    # 更新涨停分析和原因数据
+    update_zt_reason_analysis(stock_em_zt_pool_df_data, str_day)
 
     stock_em_zt_pool_df_data = common_service_fun_api.total_mv_classification(stock_em_zt_pool_df_data.copy())
 
@@ -76,12 +79,18 @@ def save_zt_info(str_day):
 
     stock_em_zt_pool_df_data = company_common_service_api.amendment_industry(stock_em_zt_pool_df_data.copy())
 
-    # 主线标记 复盘用
-    stock_em_zt_pool_df_data['main_line'] = '无'
-    stock_em_zt_pool_df_data['sub_main_line'] = '无'
-
     # 上个交易交易日涨停股票
     last_trade_day_zt_df = zt_common_service_api.get_last_trade_day_zt(str_day)
+    # 设置连板
+    stock_em_zt_pool_df_data = set_connected_boards_numbers(stock_em_zt_pool_df_data.copy(),
+                                                            last_trade_day_zt_df.copy())
+
+    # 添加主线信息
+    stock_em_zt_pool_df_data = main_line_zt_reason_service.merge_main_line_info(str_day,
+                                                                                stock_em_zt_pool_df_data.copy())
+    # 添加涨停原因和分析信息
+    stock_em_zt_pool_df_data = main_line_zt_reason_service.merge_zt_reason_info(str_day,
+                                                                                stock_em_zt_pool_df_data.copy())
 
     stock_em_zt_pool_df_data['first_closure_time'] = stock_em_zt_pool_df_data['first_closure_time'].str.strip()
     stock_em_zt_pool_df_data['list_date'] = stock_em_zt_pool_df_data['list_date'].apply(
@@ -104,105 +113,47 @@ def save_zt_info(str_day):
     stock_em_zt_pool_df_data = stock_em_zt_pool_df_data.sort_values(by=['first_closure_time'])
 
     # 重置索引，并将排序结果保存到新的"index"列中
-
     stock_em_zt_pool_df_data['str_day'] = str_day
     stock_em_zt_pool_df_data['_id'] = stock_em_zt_pool_df_data['symbol'] + "_" + str_day
     stock_em_zt_pool_df_data.drop_duplicates('symbol', keep='last', inplace=True)
 
     query_today_zt = {'str_day': str_day}
-
-    stock_today_zt_pool_df = mongodb_util.find_query_data(db_name_constant.STOCK_ZT_POOL, query_today_zt)
-
-    if data_frame_util.is_empty(stock_today_zt_pool_df):
-
+    stock_exist_zt_pool_df = mongodb_util.find_query_data(db_name_constant.STOCK_ZT_POOL, query_today_zt)
+    if data_frame_util.is_empty(stock_exist_zt_pool_df):
         today_new_zt_pool_df = stock_em_zt_pool_df_data.copy()
     else:
         today_new_zt_pool_df = stock_em_zt_pool_df_data.loc[
-            ~stock_em_zt_pool_df_data['symbol'].isin(stock_today_zt_pool_df['symbol'])]
-
-    try:
-
-        today_main_line_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL, {'str_day': str_day})
-
-        for stock_one in today_new_zt_pool_df.itertuples():
-            try:
-
-                # 设置连板数目
-                stock_em_zt_pool_df_data = set_connected_boards_numbers(stock_em_zt_pool_df_data.copy(),
-                                                                        stock_one.symbol,
-                                                                        last_trade_day_zt_df.copy())
-
-                # 涨停分析
-                zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(stock_one.symbol)
-                zt_analysis = zt_result_dict['zt_analyse_detail']
-                zt_reason = zt_result_dict['zt_reason']
-
-                stock_em_zt_pool_df_data.loc[
-                    stock_em_zt_pool_df_data['symbol'] == stock_one.symbol, 'zt_reason'] = zt_reason
-
-                stock_em_zt_pool_df_data.loc[
-                    stock_em_zt_pool_df_data['symbol'] == stock_one.symbol, 'zt_analysis'] = zt_analysis
-
-                today_main_line_one_df = today_main_line_df.loc[today_main_line_df['symbol'] == stock_one.symbol]
-                if data_frame_util.is_not_empty(today_main_line_one_df):
-                    stock_em_zt_pool_df_data.loc[
-                        stock_em_zt_pool_df_data['symbol'] == stock_one.symbol, 'main_line'] = \
-                        list(today_main_line_one_df['main_line'])[0]
-
-                    stock_em_zt_pool_df_data.loc[
-                        stock_em_zt_pool_df_data['symbol'] == stock_one.symbol, 'sub_main_line'] = \
-                        list(today_main_line_one_df['sub_main_line'])[0]
-
-                query_exist = {'symbol': stock_one.symbol, 'str_day': str_day}
-                if mongodb_util.exist_data_query(db_name_constant.STOCK_ZT_POOL, query_exist):
-                    continue
-                else:
-
-                    stock_em_zt_pool_df_data_one = stock_em_zt_pool_df_data.loc[
-                        stock_em_zt_pool_df_data['symbol'] == stock_one.symbol]
-                    stock_em_zt_pool_df_data_one = stock_em_zt_pool_df_data_one[ZT_FIELD]
-
-                    chg = round(float(list(stock_em_zt_pool_df_data_one['chg'])[0]), 2)
-                    stock_em_zt_pool_df_data_one['chg'] = chg
+            ~stock_em_zt_pool_df_data['symbol'].isin(stock_exist_zt_pool_df['symbol'])]
 
-                    mongodb_util.save_mongo(stock_em_zt_pool_df_data_one, db_name_constant.STOCK_ZT_POOL)
+    mongodb_util.save_mongo(today_new_zt_pool_df, db_name_constant.STOCK_ZT_POOL)
+    stock_em_zt_pool_df_data = pd.concat([stock_exist_zt_pool_df, today_new_zt_pool_df])
+    # 保存连板股票主线
+    save_last_trade_day_main_line(str_day, stock_em_zt_pool_df_data)
 
-            except BaseException as e:
-                stock_em_zt_pool_df_data['zt_reason'] = '0'
-                logger.error("同步涨停信息出现异常:{},{}", stock_one.symbol, e)
-    except BaseException as e:
-        stock_em_zt_pool_df_data['zt_reason'] = '0'
-        logger.error("出现异常:{}", e)
-
-    stock_em_zt_pool_df_data = pd.concat([stock_today_zt_pool_df, today_new_zt_pool_df])
+    stock_em_zt_pool_df_data.fillna('', inplace=True)
     stock_em_zt_pool_df_data = stock_em_zt_pool_df_data[ZT_FIELD]
     return stock_em_zt_pool_df_data
 
 
 # 设置连板数目
-def set_connected_boards_numbers(stock_em_zt_pool_df_data, symbol, last_trade_day_zt_df):
+def set_connected_boards_numbers(stock_em_zt_pool_df_data, last_trade_day_zt_df):
+    if data_frame_util.is_empty(stock_em_zt_pool_df_data):
+        return stock_em_zt_pool_df_data
+    if data_frame_util.is_empty(last_trade_day_zt_df):
+        return stock_em_zt_pool_df_data
     # 连板股票
     connected_boards_df_copy = last_trade_day_zt_df.loc[
         last_trade_day_zt_df['symbol'].isin(stock_em_zt_pool_df_data['symbol'])]
 
     connected_boards_df = connected_boards_df_copy.copy()
+    #
     connected_boards_df['connected_boards_numbers'] = connected_boards_df['connected_boards_numbers'] + 1
 
-    # 连板股票
-    connected_boards_df_one = connected_boards_df.loc[connected_boards_df['symbol'] == symbol]
-
-    if data_frame_util.is_not_empty(connected_boards_df_one):
-        stock_em_zt_pool_df_data.loc[stock_em_zt_pool_df_data['symbol'] == symbol, 'connected_boards_numbers'] = \
-            list(connected_boards_df_one['connected_boards_numbers'])[0]
-
-        if 'main_line' in connected_boards_df_one.columns:
-            stock_em_zt_pool_df_data.loc[stock_em_zt_pool_df_data['symbol'] == symbol, 'main_line'] = \
-                list(connected_boards_df_one['main_line'])[0]
-
-        if 'sub_main_line' in connected_boards_df_one.columns:
-            stock_em_zt_pool_df_data.loc[stock_em_zt_pool_df_data['symbol'] == symbol, 'sub_main_line'] = \
-                list(connected_boards_df_one['sub_main_line'])[0]
-
+    symbol_mapping_connected_boards_numbers = dict(
+        zip(connected_boards_df['symbol'], connected_boards_df['connected_boards_numbers']))
+    # 使用map进行替换，不匹配的保持原值
+    stock_em_zt_pool_df_data['connected_boards_numbers'] = stock_em_zt_pool_df_data['symbol'].map(
+        symbol_mapping_connected_boards_numbers).fillna(1)
     return stock_em_zt_pool_df_data
 
 
@@ -389,25 +340,50 @@ def query_company_info_with_share():
     return company_info_df
 
 
+# 异步更新涨停分析和数据
+@async_fun
+def update_zt_reason_analysis(stock_em_zt_pool_df_data, str_day):
+    if data_frame_util.is_empty(stock_em_zt_pool_df_data):
+        return
+    now_date = datetime.now()
+    now_str_day = now_date.strftime('%Y-%m-%d')
+    if now_str_day != str_day:
+        return
+
+    stock_em_zt_pool_df_data['str_day'] = str_day
+    main_line_zt_reason_service.update_symbol_list_zt_reason_analysis(stock_em_zt_pool_df_data, True)
+
+
+# 保存连板股票主线
+@async_fun
+def save_last_trade_day_main_line(str_day, stock_em_zt_pool_df_data):
+    last_trade_day = trade_date_common_service_api.get_last_trade_day(str_day)
+    stock_em_zt_pool_connected_df = stock_em_zt_pool_df_data.loc[
+        stock_em_zt_pool_df_data['connected_boards_numbers'] > 1]
+    if data_frame_util.is_empty(stock_em_zt_pool_connected_df):
+        return
+    else:
+        query = {'str_day': last_trade_day, 'symbol': {"$in": list(stock_em_zt_pool_connected_df['symbol'])}}
+        last_trade_day_main_line_detail_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL, query)
+        if data_frame_util.is_empty(last_trade_day_main_line_detail_df):
+            return
+        else:
+            last_trade_day_main_line_detail_df['_id'] = last_trade_day_main_line_detail_df['symbol'] + '_' + str_day
+            last_trade_day_main_line_detail_df['str_day'] = str_day
+            last_trade_day_main_line_detail_df['connected_boards_numbers'] = last_trade_day_main_line_detail_df[
+                                                                                 'connected_boards_numbers'] + 1
+            now_date = datetime.now()
+            str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+            last_trade_day_main_line_detail_df['update_time'] = str_now_date
+            today_exist_main_line_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL,
+                                                                    {'str_day': str_day})
+            if data_frame_util.is_not_empty(today_exist_main_line_df):
+                today_new_main_line_df = last_trade_day_main_line_detail_df.loc[
+                    last_trade_day_main_line_detail_df['symbol'].isin(list(today_exist_main_line_df['symbol']))]
+            else:
+                today_new_main_line_df = last_trade_day_main_line_detail_df.copy()
+            mongodb_util.save_mongo(today_new_main_line_df, db_name_constant.MAIN_LINE_DETAIL)
+
+
 if __name__ == '__main__':
-    save_zt_info('2025-11-17')
-# from datetime import datetime
-#
-# if __name__ == '__main__':
-#
-#     sync_date = date_handle_util.add_date_day('20240110', 0)
-#
-#     now_date = datetime.now()
-#
-#     str_now_day = sync_date.strftime('%Y-%m-%d')
-#
-#     while now_date > sync_date:
-#         try:
-#             save_zt_info(str_now_day)
-#             sync_date = date_handle_util.add_date_day(date_handle_util.no_slash_date(str_now_day), 1)
-#             print(str_now_day)
-#             str_now_day = sync_date.strftime('%Y-%m-%d')
-#
-#         except BaseException as e:
-#             sync_date = date_handle_util.add_date_day(date_handle_util.no_slash_date(str_now_day), 1)
-#             str_now_day = sync_date.strftime('%Y-%m-%d')
+    save_zt_info('2025-12-26')

mns_scheduler/zt/zt_pool/ths_zt_pool_sync_api.py

@@ -188,6 +188,6 @@ if __name__ == '__main__':
     # trade_date = '2024-08-01'
     # zt_df = ths_zt_pool(trade_date, None)
     # save_ths_zt_pool(zt_df, trade_date)
-    trade_date = '2025-09-05'
+    trade_date = '2025-11-04'
     ths_zt_pool_df_test = ths_zt_pool(trade_date, None)
     save_ths_zt_pool(ths_zt_pool_df_test, trade_date)

mns_scheduler/zt/zt_pool/update_null_zt_reason_api.py

@@ -11,6 +11,7 @@ mongodb_util = MongodbUtil('27017')
 import mns_common.api.ths.zt.ths_stock_zt_pool_api as ths_stock_zt_pool_api
 import mns_common.utils.data_frame_util as data_frame_util
 from loguru import logger
+import time
 
 
 def update_null_zt_reason(str_day):
@@ -38,6 +39,8 @@ def update_null_zt_reason(str_day):
                     stock_zt_pool_df_null_zt_reason['symbol'] == stock_zt_one.symbol]
                 # 涨停分析
                 zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(stock_zt_one.symbol)
+                time.sleep(5)
+
                 zt_analysis = zt_result_dict['zt_analyse_detail']
                 zt_reason = zt_result_dict['zt_reason']
 
@@ -55,4 +58,4 @@ def update_null_zt_reason(str_day):
 
 
 if __name__ == '__main__':
-    update_null_zt_reason('2025-11-14')
+    update_null_zt_reason('2025-12-08')

mns_scheduler/zz_task/compensation/compensate_task_one_day.py

@@ -134,7 +134,7 @@ def col_data_move():
 
 if __name__ == '__main__':
     # todo 修改日期
-    str_day = '2025-11-13'
+    str_day = '2025-11-20'
     # col_data_move()
     sync_daily_data_info()
     sync_stock_zt_pool()