mns-common 1.3.9.2__py3-none-any.whl → 1.6.1.4__py3-none-any.whl

mns_common/component/main_line/main_line_zt_reason_service.py

@@ -0,0 +1,257 @@
+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)
+import mns_common.constant.db_name_constant as db_name_constant
+from mns_common.db.MongodbUtil import MongodbUtil
+import mns_common.api.ths.zt.ths_stock_zt_pool_api as ths_stock_zt_pool_api
+from mns_common.utils.async_fun import async_fun
+import mns_common.component.trade_date.trade_date_common_service_api as trade_date_common_service_api
+import mns_common.api.ths.zt.ths_stock_zt_reason_web_api as ths_stock_zt_reason_web_api
+
+mongodb_util = MongodbUtil('27017')
+import mns_common.utils.data_frame_util as data_frame_util
+import time
+from loguru import logger
+import pandas as pd
+from datetime import datetime
+import mns_common.component.cookie.cookie_info_service as cookie_info_service
+
+
+# 添加主线和涨停分析临时数据
+def merge_main_line_info(str_day, data_df):
+    # 保证数据完整性
+    if 'main_line' not in data_df.columns:
+        data_df['main_line'] = ''
+    else:
+        data_df.fillna({'main_line': ''}, inplace=True)
+
+    if 'sub_main_line' not in data_df.columns:
+        data_df['sub_main_line'] = ''
+    else:
+        data_df.fillna({'sub_main_line': ''}, inplace=True)
+
+    if 'zt_analysis' not in data_df.columns:
+        data_df['zt_analysis'] = ''
+
+    else:
+        data_df.fillna({'zt_analysis': ''}, inplace=True)
+
+    if 'zt_reason' not in data_df.columns:
+        data_df['zt_reason'] = ''
+    else:
+        data_df.fillna({'zt_reason': ''}, inplace=True)
+
+    if 'main_line_choose_source' not in data_df.columns:
+        data_df['main_line_choose_source'] = 'now_zt'
+    else:
+        data_df.fillna({'main_line_choose_source': 'now_zt'}, inplace=True)
+
+    if 'main_line_grade' not in data_df.columns:
+        data_df['main_line_grade'] = 1
+    else:
+        data_df.fillna({'main_line_grade': 1}, inplace=True)
+
+    query_zt_now = {'symbol': {"$in": list(data_df['symbol'])}, 'str_day': str_day}
+    # merge 主线 涨停详情
+    main_line_detail_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL, query_zt_now)
+    if data_frame_util.is_not_empty(main_line_detail_df):
+        symbol_mapping_zt_reason_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['zt_reason']))
+
+        symbol_mapping_zt_analysis_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['zt_analysis']))
+
+        symbol_mapping_main_line_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['main_line']))
+
+        symbol_mapping_sub_main_line_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['sub_main_line']))
+
+        symbol_mapping_main_line_choose_source = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['main_line_choose_source']))
+
+        symbol_mapping_main_line_grade = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['main_line_grade']))
+
+        data_df['main_line_grade'] = data_df['symbol'].map(
+            symbol_mapping_main_line_grade).fillna(
+            data_df['main_line_grade'])
+
+        data_df['main_line'] = data_df['symbol'].map(
+            symbol_mapping_main_line_now).fillna(
+            data_df['main_line'])
+
+        data_df['sub_main_line'] = data_df['symbol'].map(
+            symbol_mapping_sub_main_line_now).fillna(
+            data_df['sub_main_line'])
+        data_df['zt_reason'] = data_df['symbol'].map(
+            symbol_mapping_zt_reason_now).fillna(
+            data_df['zt_reason'])
+        data_df['zt_analysis'] = data_df['symbol'].map(
+            symbol_mapping_zt_analysis_now).fillna(
+            data_df['zt_analysis'])
+
+        data_df['main_line_choose_source'] = data_df['symbol'].map(
+            symbol_mapping_main_line_choose_source).fillna(
+            data_df['main_line_choose_source'])
+
+    return data_df
+
+
+# merge涨停分析 原因
+def merge_zt_reason_info(str_day, data_df):
+    if 'zt_analysis' not in data_df.columns:
+        data_df['zt_analysis'] = ''
+
+    else:
+        data_df.fillna({'zt_analysis': ''}, inplace=True)
+
+    if 'zt_reason' not in data_df.columns:
+        data_df['zt_reason'] = ''
+    else:
+        data_df.fillna({'zt_reason': ''}, inplace=True)
+
+    query_zt_now = {'symbol': {"$in": list(data_df['symbol'])}, 'str_day': str_day}
+    # merge 主线 涨停详情
+    zt_reason_analysis_df = mongodb_util.find_query_data(db_name_constant.ZT_REASON_ANALYSIS, query_zt_now)
+    if data_frame_util.is_not_empty(zt_reason_analysis_df):
+        symbol_mapping_zt_reason_now = dict(
+            zip(zt_reason_analysis_df['symbol'], zt_reason_analysis_df['zt_reason']))
+
+        symbol_mapping_zt_analysis_now = dict(
+            zip(zt_reason_analysis_df['symbol'], zt_reason_analysis_df['zt_analysis']))
+
+        data_df['zt_reason'] = data_df['symbol'].map(
+            symbol_mapping_zt_reason_now).fillna(
+            data_df['zt_reason'])
+        data_df['zt_analysis'] = data_df['symbol'].map(
+            symbol_mapping_zt_analysis_now).fillna(
+            data_df['zt_analysis'])
+
+    return data_df
+
+
+def update_zt_reason_analysis(symbol, str_day, name, need_update):
+    try:
+        key_id = symbol + "_" + str_day
+        query_zt = {"_id": key_id}
+
+        # 已经存在的数据
+        zt_reason_analysis_exists_df = mongodb_util.find_query_data(db_name_constant.ZT_REASON_ANALYSIS,
+                                                                    query_zt)
+
+        if data_frame_util.is_not_empty(zt_reason_analysis_exists_df):
+            zt_analysis = list(zt_reason_analysis_exists_df['zt_analysis'])[0]
+            zt_reason = list(zt_reason_analysis_exists_df['zt_reason'])[0]
+            # 需要更新数据
+            if data_frame_util.is_string_empty(zt_analysis) or data_frame_util.is_string_empty(
+                    zt_reason) or need_update:
+
+                # web端更新数据
+                try:
+                    web_zt_result_dict = ths_stock_zt_reason_web_api.get_ths_web_zt_reason_info(symbol,
+                                                                                                cookie_info_service.get_ths_cookie())
+                    zt_analysis = web_zt_result_dict['zt_analysis']
+                    zt_reason = web_zt_result_dict['zt_reason']
+                    time.sleep(1)
+                except BaseException as e:
+                    time.sleep(1)
+                    zt_analysis = ''
+                    zt_reason = ''
+
+                # 问财更新数据
+                if data_frame_util.is_string_empty(zt_analysis) or data_frame_util.is_string_empty(
+                        zt_reason):
+                    try:
+                        zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(symbol)
+                        zt_analysis = zt_result_dict['zt_analyse_detail']
+                        zt_reason = zt_result_dict['zt_reason']
+                        time.sleep(1)
+                    except BaseException as e:
+                        time.sleep(1)
+                        zt_analysis = ''
+                        zt_reason = ''
+
+                zt_reason_analysis_exists_df['zt_analysis'] = zt_analysis
+                zt_reason_analysis_exists_df['zt_reason'] = zt_reason
+                now_date = datetime.now()
+                str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+                zt_reason_analysis_exists_df['update_time'] = str_now_date
+                mongodb_util.save_mongo(zt_reason_analysis_exists_df, db_name_constant.ZT_REASON_ANALYSIS)
+        else:
+            # 不存在临时主线数据
+            try:
+                zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(symbol)
+                zt_analysis = zt_result_dict['zt_analyse_detail']
+                zt_reason = zt_result_dict['zt_reason']
+                time.sleep(1)
+            except BaseException as e:
+                time.sleep(1)
+                zt_analysis = ''
+                zt_reason = ''
+            now_date = datetime.now()
+            str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+            reason_dict = {'_id': key_id,
+                           'symbol': symbol,
+                           'name': name,
+                           'zt_analysis': zt_analysis,
+                           'zt_reason': zt_reason,
+                           'str_day': str_day,
+                           'update_time': str_now_date,
+                           }
+            reason_df = pd.DataFrame(reason_dict, index=[1])
+            mongodb_util.save_mongo(reason_df, db_name_constant.ZT_REASON_ANALYSIS)
+    except BaseException as e:
+        logger.error("添加涨停原因详情异常:{},{}", e, name)
+
+
+def update_symbol_list_zt_reason_analysis(data_df, need_update):
+    for stock_one in data_df.itertuples():
+        str_day = stock_one.str_day
+        symbol = stock_one.symbol
+        name = stock_one.name
+        update_zt_reason_analysis(symbol, str_day, name, need_update)
+        if need_update:
+            time.sleep(5)
+        else:
+            time.sleep(1)
+
+
+# 保存连板股票主线
+@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, 'symbol': {"$in": list(
+                                                                        last_trade_day_main_line_detail_df['symbol'])}})
+            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__':
+    update_zt_reason_analysis('600362', '2025-12-26', '江西铜业', True)

mns_common/component/proxies/proxy_common_api.py

@@ -7,78 +7,246 @@ project_path = file_path[0:end]
 sys.path.append(project_path)
 
 import mns_common.api.proxies.liu_guan_proxy_api as liu_guan_proxy_api
-from functools import lru_cache
 import pandas as pd
 import mns_common.utils.data_frame_util as data_frame_util
 from mns_common.db.MongodbUtil import MongodbUtil
 import mns_common.constant.db_name_constant as db_name_constant
 import datetime
+import time
+from loguru import logger
+from functools import lru_cache
+import mns_common.api.em.real_time.east_money_stock_a_api as east_money_stock_a_api
+import threading
 
 mongodb_util = MongodbUtil('27017')
 
+IP_POOL = 'ip_pool'
+ONE_IP = 'one_ip'
+query_one = {'ip_type': ONE_IP}
+query_pool = {'ip_type': IP_POOL}
+
 
-@lru_cache(maxsize=None)
 def query_liu_guan_proxy_ip():
-    ip_proxy_pool = mongodb_util.find_all_data(db_name_constant.IP_PROXY_POOL)
+    ip_proxy_pool = mongodb_util.find_query_data(db_name_constant.IP_PROXY_POOL, query_one)
     return ip_proxy_pool
 
 
-def clear_proxy_ip_cache():
-    query_liu_guan_proxy_ip.cache_clear()
-    mongodb_util.remove_data({}, db_name_constant.IP_PROXY_POOL)
+def remove_one_proxy_ip():
+    mongodb_util.remove_data(query_one, db_name_constant.IP_PROXY_POOL)
 
 
-def check_valid(str_now_date, ip_proxy_pool):
+def check_valid(ip_proxy_pool):
     effect_time = list(ip_proxy_pool['effect_time'])[0]
+    now_date = datetime.datetime.now()
+    str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
     if effect_time > str_now_date:
         return True
     else:
-        clear_proxy_ip_cache()
+        remove_one_proxy_ip()
         return False
 
 
+@lru_cache(maxsize=None)
+def get_account_cache():
+    query = {"type": "liu_guan_proxy"}
+    return mongodb_util.find_query_data(db_name_constant.STOCK_ACCOUNT_INFO, query)
+
+
+def generate_proxy_ip_api(minutes):
+    try_numer = 3
+    while try_numer > 0:
+        try:
+            stock_account_info = get_account_cache()
+            order_id = list(stock_account_info['password'])[0]
+            secret = list(stock_account_info['account'])[0]
+            # 获取10分钟动态ip
+            liu_guan_ip = liu_guan_proxy_api.get_proxy_api(order_id, secret, str(60 * minutes))
+            try_numer = try_numer
+            logger.info("生成新的ip:{}", liu_guan_ip)
+            return liu_guan_ip
+        except BaseException as e:
+            logger.error("获取ip失败:{}", str(e))
+            time.sleep(1)
+            continue
+
+
 def generate_proxy_ip(minutes):
-    ip_proxy_pool = mongodb_util.find_all_data(db_name_constant.IP_PROXY_POOL)
+    ip_proxy_pool = query_liu_guan_proxy_ip()
     if data_frame_util.is_not_empty(ip_proxy_pool):
         return list(ip_proxy_pool['ip'])[0]
     else:
-        query = {"type": "liu_guan_proxy", }
-        stock_account_info = mongodb_util.find_query_data(db_name_constant.STOCK_ACCOUNT_INFO, query)
-        order_id = list(stock_account_info['password'])[0]
-        secret = list(stock_account_info['account'])[0]
-
+        remove_one_proxy_ip()
         now_date = datetime.datetime.now()
-        time_to_add = datetime.timedelta(minutes=minutes - 1, seconds=50)
+        # 加上分钟
+        time_to_add = datetime.timedelta(minutes=minutes)
         new_date = now_date + time_to_add
         str_now_date = new_date.strftime('%Y-%m-%d %H:%M:%S')
+        try:
+            ip_proxy = generate_proxy_ip_api(minutes)
 
-        # 获取10分钟动态ip
-        ip = liu_guan_proxy_api.get_proxy_api(order_id, secret, str(60 * minutes))
+            result_dict = {"_id": ip_proxy,
+                           'ip_type': ONE_IP,
+                           'effect_time': str_now_date,
+                           'ip': ip_proxy}
+            result_df = pd.DataFrame(result_dict, index=[1])
 
-        result_dict = {"_id": ip,
-                       'effect_time': str_now_date,
-                       'ip': ip}
-        result_df = pd.DataFrame(result_dict, index=[1])
+            mongodb_util.insert_mongo(result_df, db_name_constant.IP_PROXY_POOL)
+        except BaseException as e:
+            logger.error("获取ip失败:{}", str(e))
 
-        mongodb_util.insert_mongo(result_df, db_name_constant.IP_PROXY_POOL)
-
-        return ip
+        return ip_proxy
 
 
-def get_proxy_ip(str_now_date, minutes):
+def get_proxy_ip(minutes):
     ip_proxy_pool = query_liu_guan_proxy_ip()
     if data_frame_util.is_empty(ip_proxy_pool):
         return generate_proxy_ip(minutes)
     else:
-        if check_valid(str_now_date, ip_proxy_pool):
+        if check_valid(ip_proxy_pool):
             return list(ip_proxy_pool['ip'])[0]
         else:
             return generate_proxy_ip(minutes)
 
 
-if __name__ == '__main__':
-    while True:
+#
+# def check_proxy(proxy_ip):
+#     try:
+#         # 两秒超时
+#         test_df = call_with_timeout(get_em_real_time_data, proxy_ip, timeout=2)
+#         if data_frame_util.is_not_empty(test_df):
+#             logger.info("可用代理ip:{}", proxy_ip)
+#             return True
+#         else:
+#             return False
+#     except Exception as e:
+#         logger.error("代理ip不可用:{},{}", proxy_ip, e)
+#         return False
+
+
+# 查询ip池子
+def query_liu_guan_proxy_ip_pool():
+    ip_proxy_pool = mongodb_util.find_query_data(db_name_constant.IP_PROXY_POOL, query_pool)
+    return ip_proxy_pool
+
+
+def remove_proxy_ip_pool():
+    mongodb_util.remove_data(query_pool, db_name_constant.IP_PROXY_POOL)
+
+
+def generate_proxy_ip_pool_api(minutes, ip_num):
+    stock_account_info = get_account_cache()
+    order_id = list(stock_account_info['password'])[0]
+    secret = list(stock_account_info['account'])[0]
+    # 获取10分钟动态ip
+    ip_pool = liu_guan_proxy_api.get_proxy_pool_api(order_id, secret, str(60 * minutes), ip_num)
+    return ip_pool
+
+
+def get_proxy_ip_pool(minutes, seconds, ip_num):
+    ip_proxy_pool = query_liu_guan_proxy_ip_pool()
+    if data_frame_util.is_empty(ip_proxy_pool):
+        return generate_proxy_ip_pool(minutes, seconds, ip_num)
+    else:
+        if check_valid(ip_proxy_pool):
+            ip_pool = list(ip_proxy_pool['ip_pool'])[0]
+            effect_time = list(ip_proxy_pool['effect_time'])[0]
+            result = {'ip_pool': ip_pool,
+                      'effect_time': effect_time}
+            return result
+        else:
+            # 已经失效 移除ip pool
+            remove_proxy_ip_pool()
+            # 重新生成
+            return generate_proxy_ip_pool(minutes, seconds, ip_num)
+
+
+# seconds 有效秒数,minutes 需要减1
+def generate_proxy_ip_pool(minutes, seconds, ip_num):
+    ip_proxy_pool = query_liu_guan_proxy_ip_pool()
+    if data_frame_util.is_not_empty(ip_proxy_pool):
+        ip_pool = list(ip_proxy_pool['ip_pool'])[0]
+        effect_time = list(ip_proxy_pool['effect_time'])[0]
+
+
+    else:
+        remove_proxy_ip_pool()
         now_date = datetime.datetime.now()
-        str_now_date_test = now_date.strftime('%Y-%m-%d %H:%M:%S')
-        ip_test = get_proxy_ip(str_now_date_test)
-        print(ip_test)
+        # 加上分钟 少10秒
+        time_to_add = datetime.timedelta(minutes=minutes - 1, seconds=seconds)
+        new_date = now_date + time_to_add
+        effect_time = new_date.strftime('%Y-%m-%d %H:%M:%S')
+        ip_pool = generate_proxy_ip_pool_api(minutes, ip_num)
+        result_dict = {
+            "_id": [IP_POOL],
+            'ip_type': [IP_POOL],
+            'effect_time': [effect_time],
+            'ip_pool': [ip_pool]  # 每个字段都包装成列表
+        }
+        result_df = pd.DataFrame(result_dict)
+
+        mongodb_util.insert_mongo(result_df, db_name_constant.IP_PROXY_POOL)
+    result = {'ip_pool': ip_pool,
+              'effect_time': effect_time}
+    return result
+
+
+def get_em_real_time_data(proxy_ip):
+    proxies = {
+        "http": proxy_ip,
+        "https": proxy_ip
+    }
+    return east_money_stock_a_api.get_stock_page_data(1, proxies, 20, 10)
+
+
+# 定义一个带超时的函数调用
+def call_with_timeout(func, *args, timeout=2, **kwargs):
+    # 用于存储函数执行结果
+    result = None
+    exception = None
+
+    # 定义一个线程目标函数
+    def target():
+        nonlocal result, exception
+        try:
+            result = func(*args, **kwargs)
+        except Exception as e:
+            exception = e
+
+    # 创建线程并启动
+    thread = threading.Thread(target=target)
+    thread.start()
+
+    # 等待线程完成，最多等待 timeout 秒
+    thread.join(timeout)
+
+    # 如果线程仍然存活，说明函数超时了
+    if thread.is_alive():
+        raise TimeoutError(f"Function exceeded timeout of {timeout} seconds")
+
+    # 如果函数抛出了异常，重新抛出
+    if exception is not None:
+        raise exception
+    return result
+
+
+@lru_cache(maxsize=None)
+def query_province_and_city_info():
+    return mongodb_util.find_all_data(db_name_constant.IP_PROXY_CITY_PROVINCE)
+
+
+def import_province_and_city():
+    # 设置文件夹路径
+    folder_path = r'E:\province-and-city.xlsx'
+    df = pd.read_excel(folder_path)
+    df['_id'] = df['cid']
+
+    mongodb_util.save_mongo(df, db_name_constant.IP_PROXY_CITY_PROVINCE)
+    return df
+
+
+if __name__ == "__main__":
+    stock_account_info_test = get_account_cache()
+    order_id_test = list(stock_account_info_test['password'])[0]
+    secret_test = list(stock_account_info_test['account'])[0]
+    # 获取10分钟动态ip
+    ip = liu_guan_proxy_api.get_proxy_api(order_id_test, secret_test, str(60 * 1))

mns_common/component/tfp/stock_tfp_api.py

@@ -5,7 +5,6 @@ file_path = os.path.abspath(__file__)
 end = file_path.index('mns') + 16
 project_path = file_path[0:end]
 sys.path.append(project_path)
-import akshare as ak
 import mns_common.utils.date_handle_util as date_handle_util
 from loguru import logger
 import mns_common.utils.data_frame_util as data_frame_util
@@ -15,20 +14,87 @@ import mns_common.constant.db_name_constant as db_name_constant
 
 mongodb_util = MongodbUtil('27017')
 
+"""
+Date: 2024/4/29 15:00
+Desc: 东方财富网-数据中心-特色数据-停复牌信息
+https://data.eastmoney.com/tfpxx/
+"""
+
+import pandas as pd
+import requests
+
+
+def stock_tfp_em(date: str = "20240426") -> pd.DataFrame:
+    """
+    东方财富网-数据中心-特色数据-停复牌信息
+    https://data.eastmoney.com/tfpxx/
+    :param date: specific date as "2020-03-19"
+    :type date: str
+    :return: 停复牌信息表
+    :rtype: pandas.DataFrame
+    """
+    url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
+    params = {
+        "sortColumns": "SUSPEND_START_DATE",
+        "sortTypes": "-1",
+        "pageSize": "500",
+        "pageNumber": "1",
+        "reportName": "RPT_CUSTOM_SUSPEND_DATA_INTERFACE",
+        "columns": "ALL",
+        "source": "WEB",
+        "client": "WEB",
+        "filter": f"""(MARKET="全部")(DATETIME='{"-".join([date[:4], date[4:6], date[6:]])}')""",
+    }
+    r = requests.get(url, params=params)
+    data_json = r.json()
+    total_page = data_json["result"]["pages"]
+    big_df = pd.DataFrame()
+    for page in range(1, total_page + 1):
+        params.update({"pageNumber": page})
+        r = requests.get(url, params=params)
+        data_json = r.json()
+        temp_df = pd.DataFrame(data_json["result"]["data"])
+        big_df = pd.concat(objs=[big_df, temp_df], ignore_index=True)
+
+    big_df.reset_index(inplace=True)
+
+    big_df["SUSPEND_START_TIME"] = pd.to_datetime(big_df["SUSPEND_START_TIME"], errors="coerce").dt.date
+    big_df["SUSPEND_END_TIME"] = pd.to_datetime(
+        big_df["SUSPEND_END_TIME"], errors="coerce"
+    ).dt.date
+
+    big_df["SUSPEND_START_DATE"] = pd.to_datetime(
+        big_df["SUSPEND_START_DATE"], errors="coerce"
+    ).dt.date
+    big_df["PREDICT_RESUME_DATE"] = pd.to_datetime(
+        big_df["PREDICT_RESUME_DATE"], errors="coerce"
+    ).dt.date
+
+    big_df = big_df[['index', 'SECURITY_CODE', 'SECURITY_NAME_ABBR', 'SUSPEND_START_TIME',
+                     'SUSPEND_END_TIME', 'SUSPEND_EXPIRE', 'SUSPEND_REASON', 'TRADE_MARKET',
+                     'SUSPEND_START_DATE',
+                     'PREDICT_RESUME_DATE'
+                     ]]
+
+    return big_df
+
 
 def get_stock_tfp_by_day(str_day):
-    stock_tfp_em_df = ak.stock_tfp_em(date_handle_util.no_slash_date(str_day))
+    stock_tfp_em_df = stock_tfp_em(date_handle_util.no_slash_date(str_day))
     stock_tfp_em_df = stock_tfp_em_df.rename(
-        columns={'序号': 'index',
-                 '代码': 'symbol',
-                 '名称': 'name',
-                 '停牌时间': 'sus_begin_time',
-                 '停牌截止时间': 'sus_end_time',
-                 '停牌截止时间': 'sus_end_time',
-                 '停牌期限': 'sus_period',
-                 '停牌原因': 'sus_reason',
-                 '所属市场': 'market',
-                 '预计复牌时间': 'resume_time'
+        columns={'index': 'index',
+                 'SECURITY_CODE': 'symbol',
+                 'SECURITY_NAME_ABBR': 'name',
+                 'SUSPEND_START_TIME': 'sus_begin_time',
+                 'SUSPEND_END_TIME': 'sus_end_time',
+
+                 'SUSPEND_START_DATE': 'sus_begin_date',
+                 'PREDICT_RESUME_DATE': 'resume_time',
+
+                 'SUSPEND_EXPIRE': 'sus_period',
+                 'SUSPEND_REASON': 'sus_reason',
+                 'TRADE_MARKET': 'market',
+
                  })
     return stock_tfp_em_df
 
@@ -60,3 +126,7 @@ def get_stock_tfp_symbol_from_db(str_day):
     except BaseException as e:
         logger.error("获取停牌信息异常:{}", e)
         return ['666666']
+
+
+if __name__ == '__main__':
+    get_stock_tfp_symbol_list_by_day('2025-06-21')

mns_common/component/us/__init__.py

@@ -0,0 +1,7 @@
+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)