mns-scheduler 1.4.3.3__py3-none-any.whl → 1.4.4.5__py3-none-any.whl

mns_scheduler/irm/stock_question_id_service.py

@@ -0,0 +1,159 @@
+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
+
+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)]
+
+    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_sh_stock_uid([])

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/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-11-18')
+    sync_stock_zb_pool('2025-12-11')

mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py

@@ -66,7 +66,7 @@ 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)
@@ -80,11 +80,20 @@ 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'] = '无'
+    stock_em_zt_pool_df_data['main_line'] = ''
+    stock_em_zt_pool_df_data['sub_main_line'] = ''
+    stock_em_zt_pool_df_data['zt_reason'] = ''
+    stock_em_zt_pool_df_data['zt_analysis'] = ''
 
     # 上个交易交易日涨停股票
     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 = sync_main_line_data(stock_em_zt_pool_df_data.copy(), str_day)
+    # 保存今日主线数据
+    stock_em_zt_pool_df_data = save_today_main_line(stock_em_zt_pool_df_data, str_day)
 
     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(
@@ -107,129 +116,44 @@ 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:
-                zt_reason = ''
-                zt_analyse_detail = ''
-
-                # 设置连板数目
-                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())
-
-                reason_tag = False
-                # 网页获取
-                try:
-                    ths_cookie = cookie_info_service.get_ths_cookie()
-                    # 问财获取涨停分析
-                    zt_analyse_detail = ths_company_info_api.get_company_hot_info(stock_one.symbol, ths_cookie)
-
-                    zt_reason = zt_analyse_detail.split("\n")[0]
-                    reason_tag = True
-                    time.sleep(2)
-                except BaseException as e:
-                    time.sleep(2)
-                    logger.error("网页获取涨停详情异常:{},{}", stock_one.symbol, e)
-
-                # 问财获取
-                if bool(1 - reason_tag):
-                    try:
-                        # 问财获取涨停分析
-                        zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(stock_one.symbol)
-                        zt_analyse_detail = zt_result_dict['zt_analyse_detail']
-                        zt_reason = zt_result_dict['zt_reason']
-                        time.sleep(2)
-                    except BaseException as e:
-                        time.sleep(2)
-                        logger.error("问财获取涨停详情异常:{},{}", stock_one.symbol, e)
-
-                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_analyse_detail
-
-                if data_frame_util.is_not_empty(today_main_line_df):
-                    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
-
-                    mongodb_util.save_mongo(stock_em_zt_pool_df_data_one, db_name_constant.STOCK_ZT_POOL)
-
-            except BaseException as e:
+            ~stock_em_zt_pool_df_data['symbol'].isin(stock_exist_zt_pool_df['symbol'])]
 
-                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])
+    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])
+    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
 
 
@@ -416,25 +340,192 @@ def query_company_info_with_share():
     return company_info_df
 
 
+def sync_main_line_data(stock_em_zt_pool_df_data, str_day):
+    # 主线标记 复盘用
+    stock_em_zt_pool_df_data['main_line'] = ''
+    stock_em_zt_pool_df_data['sub_main_line'] = ''
+    stock_em_zt_pool_df_data['zt_reason'] = ''
+    stock_em_zt_pool_df_data['zt_analysis'] = ''
+
+    last_trade_day = trade_date_common_service_api.get_last_trade_day(str_day)
+
+    last_trade_day_main_line_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL,
+                                                               {'str_day': last_trade_day, 'symbol': {
+                                                                   "$in": list(stock_em_zt_pool_df_data['symbol'])}})
+    if data_frame_util.is_not_empty(last_trade_day_main_line_df):
+        symbol_mapping_main_line = dict(
+            zip(last_trade_day_main_line_df['symbol'], last_trade_day_main_line_df['main_line']))
+
+        symbol_mapping_sub_main_line = dict(
+            zip(last_trade_day_main_line_df['symbol'], last_trade_day_main_line_df['sub_main_line']))
+
+        stock_em_zt_pool_df_data['main_line'] = (stock_em_zt_pool_df_data['symbol']
+        .map(symbol_mapping_main_line).fillna(
+            stock_em_zt_pool_df_data['main_line']))
+
+        stock_em_zt_pool_df_data['sub_main_line'] = stock_em_zt_pool_df_data['symbol'].map(
+            symbol_mapping_sub_main_line).fillna(
+            stock_em_zt_pool_df_data['sub_main_line'])
+
+    today_zt_reason_analysis = mongodb_util.find_query_data(db_name_constant.MAIN_REASON_ANALYSIS,
+                                                            {'str_day': str_day, 'symbol': {
+                                                                "$in": list(stock_em_zt_pool_df_data['symbol'])}})
+    if data_frame_util.is_not_empty(today_zt_reason_analysis):
+        # 创建一个映射字典
+        symbol_mapping_zt_reason = dict(zip(today_zt_reason_analysis['symbol'], today_zt_reason_analysis['zt_reason']))
+
+        symbol_mapping_zt_analysis = dict(
+            zip(today_zt_reason_analysis['symbol'], today_zt_reason_analysis['zt_analysis']))
+
+        symbol_mapping_main_line = dict(zip(today_zt_reason_analysis['symbol'], today_zt_reason_analysis['main_line']))
+
+        symbol_mapping_sub_main_line = dict(
+            zip(today_zt_reason_analysis['symbol'], today_zt_reason_analysis['sub_main_line']))
+
+        # 使用map进行替换，不匹配的保持原值
+        stock_em_zt_pool_df_data['zt_reason'] = stock_em_zt_pool_df_data['symbol'].map(symbol_mapping_zt_reason).fillna(
+            stock_em_zt_pool_df_data['zt_reason'])
+
+        stock_em_zt_pool_df_data['zt_analysis'] = stock_em_zt_pool_df_data['symbol'].map(
+            symbol_mapping_zt_analysis).fillna(
+            stock_em_zt_pool_df_data['zt_analysis'])
+
+        stock_em_zt_pool_df_data['main_line'] = stock_em_zt_pool_df_data['symbol'].map(symbol_mapping_main_line).fillna(
+            stock_em_zt_pool_df_data['main_line'])
+
+        stock_em_zt_pool_df_data['sub_main_line'] = stock_em_zt_pool_df_data['symbol'].map(
+            symbol_mapping_sub_main_line).fillna(
+            stock_em_zt_pool_df_data['sub_main_line'])
+
+    return stock_em_zt_pool_df_data
+
+
+def save_today_main_line(stock_em_zt_pool_df_data, str_day):
+    for stock_one in stock_em_zt_pool_df_data.itertuples():
+        try:
+            zt_reason = stock_one.zt_reason
+            zt_analyse_detail = stock_one.zt_analysis
+            reason_tag = False
+            if data_frame_util.is_string_empty(zt_reason) or data_frame_util.is_string_empty(zt_analyse_detail):
+
+                # 问财获取
+                if bool(1 - reason_tag):
+                    try:
+                        # 问财获取涨停分析
+                        zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(stock_one.symbol)
+                        zt_analyse_detail = zt_result_dict['zt_analyse_detail']
+                        zt_reason = zt_result_dict['zt_reason']
+                        time.sleep(2)
+                        reason_tag = True
+                    except BaseException as e:
+                        time.sleep(2)
+                        logger.error("问财获取涨停详情异常:{},{}", stock_one.symbol, e)
+
+                if bool(1 - reason_tag):
+                    # 网页获取
+                    try:
+                        ths_cookie = cookie_info_service.get_ths_cookie()
+                        # 问财获取涨停分析
+                        zt_analyse_detail = ths_company_info_api.get_company_hot_info(stock_one.symbol, ths_cookie)
+                        zt_reason = zt_analyse_detail.split("\n")[0]
+
+                        time.sleep(2)
+                    except BaseException as e:
+                        time.sleep(2)
+                        logger.error("网页获取涨停详情异常:{},{}", stock_one.symbol, e)
+
+            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_analyse_detail
+
+            main_line = stock_one.main_line
+            sub_main_line = stock_one.sub_main_line
+            connected_boards_numbers = stock_one.connected_boards_numbers
+
+            if data_frame_util.is_string_empty(zt_reason):
+                continue
+
+            if data_frame_util.is_string_empty(zt_analyse_detail):
+                continue
+
+            key_id = stock_one.symbol + "_" + str_day
+            # 更新临时表
+            query_zt = {"_id": key_id}
+            if mongodb_util.exist_data_query(db_name_constant.MAIN_REASON_ANALYSIS, query_zt):
+
+                if data_frame_util.is_string_not_empty(zt_reason):
+                    new_values_reason = {"$set": {
+                        'zt_reason': zt_reason}}
+
+                    mongodb_util.update_many(query_zt, new_values_reason, db_name_constant.MAIN_REASON_ANALYSIS)
+
+                if data_frame_util.is_string_not_empty(zt_analyse_detail):
+                    new_values_zt_analysis = {"$set": {
+                        'zt_analysis': zt_analyse_detail}}
+
+                    mongodb_util.update_many(query_zt, new_values_zt_analysis, db_name_constant.MAIN_REASON_ANALYSIS)
+            else:
+                main_dict = {'_id': key_id,
+                             'symbol': stock_one.symbol,
+                             'name': stock_one.name,
+                             'zt_analysis': zt_analyse_detail,
+                             'zt_reason': zt_reason,
+                             'main_line': main_line,
+                             'sub_main_line': sub_main_line,
+                             'str_day': str_day,
+                             'update_time': str_now_date,
+                             'connected_boards_numbers': connected_boards_numbers
+                             }
+
+                main_line_df = pd.DataFrame(main_dict, index=[1])
+                mongodb_util.insert_mongo(main_line_df, db_name_constant.MAIN_REASON_ANALYSIS)
+
+            if data_frame_util.is_string_empty(main_line):
+                continue
+
+            if data_frame_util.is_string_empty(sub_main_line):
+                continue
+
+            # 去除开头和结尾空格
+            if data_frame_util.is_string_not_empty(main_line):
+                main_line = main_line.strip()
+            if data_frame_util.is_string_not_empty(sub_main_line):
+                sub_main_line = sub_main_line.strip()
+            now_date = datetime.now()
+            str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+            hour = now_date.hour
+
+            main_dict = {'_id': key_id,
+                         'symbol': stock_one.symbol,
+                         'name': stock_one.name,
+                         'zt_analysis': zt_analyse_detail,
+                         'zt_reason': zt_reason,
+                         'main_line': main_line,
+                         'sub_main_line': sub_main_line,
+                         'str_day': str_day,
+                         'update_time': str_now_date,
+                         'connected_boards_numbers': connected_boards_numbers
+                         }
+
+            main_line_df = pd.DataFrame(main_dict, index=[1])
+
+            # 更新主线表
+            is_no_update_main_line = trade_date_common_service_api.is_trade_day(str_day) and 8 < hour < 15
+            if is_no_update_main_line:
+                continue
+
+            if mongodb_util.exist_data_query(db_name_constant.MAIN_LINE_DETAIL, query_zt):
+                continue
+            else:
+                mongodb_util.insert_mongo(main_line_df, db_name_constant.MAIN_LINE_DETAIL)
+
+        except BaseException as e:
+            logger.error("同步涨停信息出现异常:{},{}", stock_one.symbol, e)
+
+    return stock_em_zt_pool_df_data
+
+
 if __name__ == '__main__':
-    save_zt_info('2025-11-18')
-# 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-03')

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()