mns-common 1.3.3.5__py3-none-any.whl → 1.5.7.2__py3-none-any.whl

mns_common/api/ths/zt/ths_stock_zt_pool_v2_api.py

@@ -14,7 +14,7 @@ end = file_path.index('mns') + 14
 project_path = file_path[0:end]
 sys.path.append(project_path)
 import re
-import mns_common.component.company.company_common_service_api as company_common_service_api
+import mns_common.component.company.company_common_service_new_api as company_common_service_new_api
 import mns_common.component.zt.zt_common_service_api as zt_common_service_api
 from loguru import logger
 # question
@@ -40,7 +40,7 @@ from loguru import logger
 # 当设置具体数值n时，循环请求n页，返回n页合并数据。
 
 
-import mns_common.api.em.east_money_stock_api as east_money_stock_api
+import mns_common.component.em.em_real_time_quotes_api as em_real_time_quotes_api
 from mns_common.db.MongodbUtil import MongodbUtil
 import pandas as pd
 import mns_common.component.common_service_fun_api as common_service_fun_api
@@ -139,13 +139,13 @@ def convert_statistics(stat):
 
 # 获取实时行情涨停列表
 def get_real_time_zt_info():
-    real_time_df = east_money_stock_api.get_real_time_quotes_all_stocks()
-    real_time_df_zt = real_time_df.loc[real_time_df['chg'] > common_service_fun_api.ZT_CHG]
+    real_time_df = em_real_time_quotes_api.get_real_time_quotes_now(None, None)
+    real_time_df_zt = real_time_df.loc[
+        (real_time_df['chg'] > common_service_fun_api.ZT_CHG) | (real_time_df['wei_bi'] == 100)]
     if data_frame_util.is_empty(real_time_df_zt):
         return pd.DataFrame()
     real_time_df_zt = real_time_df_zt[[
         'symbol',
-        'name',
         'chg',
         'amount',
         'quantity_ratio',
@@ -153,16 +153,17 @@ def get_real_time_zt_info():
         'high',
         'low',
         'open',
-        'list_date',
         'exchange',
         'wei_bi',
         'flow_mv',
         'total_mv',
         'buy_1_num'
     ]]
-    company_df = company_common_service_api.get_company_info_industry()
+    company_df = company_common_service_new_api.get_company_all_info_info()
     company_df = company_df[[
         '_id',
+        'name',
+        'list_date',
         "industry",
         "first_sw_industry",
         "second_sw_industry",
@@ -173,9 +174,6 @@ def get_real_time_zt_info():
         "em_industry",
         "company_type",
         "mv_circulation_ratio",
-        "ths_concept_list_info",
-        "kpl_plate_name",
-        "kpl_plate_list_info",
         "diff_days"
     ]]
     company_df = company_df.loc[company_df['_id'].isin(real_time_df_zt['symbol'])]
@@ -200,24 +198,83 @@ def get_now_zt_pool_with_reason(str_day):
     real_time_zt_df = get_real_time_zt_info()
     # 昨日涨停列表
     last_trade_zt = zt_common_service_api.get_last_trade_day_zt(str_day)
-    # 东方财富涨停列表
-    em_now_zt_pool = stock_zt_pool_api.stock_em_zt_pool_df(date_handle_util.no_slash_date(str_day))
+
+    try:
+        # 东方财富涨停列表
+        em_now_zt_pool = stock_zt_pool_api.stock_em_zt_pool_df(date_handle_util.no_slash_date(str_day))
+    except BaseException as e:
+        em_now_zt_pool = pd.DataFrame()
+        logger.error("获取东方财富涨停列表异常:{}", e)
+
     if data_frame_util.is_not_empty(em_now_zt_pool):
         real_time_zt_df_wei_bi_100 = real_time_zt_df.loc[real_time_zt_df['symbol'].isin(em_now_zt_pool['symbol'])]
         real_time_zt_df_high_chg = real_time_zt_df.loc[~(real_time_zt_df['symbol'].isin(em_now_zt_pool['symbol']))]
+
+        em_now_zt_pool = em_now_zt_pool[['symbol',
+                                         'connected_boards_numbers',
+                                         'statistics',
+                                         'closure_funds',
+                                         'first_closure_time',
+                                         'last_closure_time',
+                                         'frying_plates_numbers']]
+
+        em_now_zt_pool = em_now_zt_pool.set_index(['symbol'], drop=True)
+        real_time_zt_df_wei_bi_100 = real_time_zt_df_wei_bi_100.set_index(['symbol'], drop=False)
+        real_time_zt_df_wei_bi_100 = pd.merge(real_time_zt_df_wei_bi_100, em_now_zt_pool, how='outer',
+                                              left_index=True, right_index=True)
+
+        now_continue_zt = last_trade_zt.loc[
+            last_trade_zt['symbol'].isin(list(real_time_zt_df_wei_bi_100['symbol']))]
+
     else:
-        real_time_zt_df_high_chg = real_time_zt_df
+        real_time_zt_df_wei_bi_100 = real_time_zt_df.loc[real_time_zt_df['wei_bi'] == 100]
+
+        real_time_zt_df_wei_bi_100['connected_boards_numbers'] = 1
+        real_time_zt_df_wei_bi_100['statistics'] = '1/1'
+        real_time_zt_df_wei_bi_100['closure_funds'] = real_time_zt_df_wei_bi_100['now_price'] * \
+                                                      real_time_zt_df_wei_bi_100['buy_1_num'] * 100
+        real_time_zt_df_wei_bi_100['first_closure_time'] = '153000'
+        real_time_zt_df_wei_bi_100['last_closure_time'] = '153000'
+        real_time_zt_df_wei_bi_100['frying_plates_numbers'] = 0
+        now_continue_zt = last_trade_zt.loc[
+            last_trade_zt['symbol'].isin(list(real_time_zt_df_wei_bi_100['symbol']))]
+
+        if data_frame_util.is_not_empty(now_continue_zt):
+            now_continue_zt = now_continue_zt[['symbol', 'connected_boards_numbers', 'statistics']]
+
+            connected_boards_map = now_continue_zt.set_index('symbol')['connected_boards_numbers'].to_dict()
+
+            # 将映射的值加到原DataFrame上
+            real_time_zt_df_wei_bi_100['connected_boards_numbers'] = real_time_zt_df_wei_bi_100[
+                                                                         'connected_boards_numbers'] + \
+                                                                     real_time_zt_df_wei_bi_100['symbol'].map(
+                                                                         connected_boards_map).fillna(0)
+
+            real_time_zt_df_wei_bi_100 = real_time_zt_df_wei_bi_100.sort_values(by=['connected_boards_numbers'],
+                                                                                ascending=False)
+
+        real_time_zt_df_high_chg = real_time_zt_df.loc[real_time_zt_df['wei_bi'] != 100]
+
+        # todo
+    # 高涨幅处理
+    now_zt_pool_df = handle_high_chg(real_time_zt_df_high_chg, last_trade_zt, real_time_zt_df_wei_bi_100)
+
+    now_zt_pool_df = merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt, now_continue_zt)
+
+    result_first = now_zt_pool_df.loc[now_zt_pool_df['connected_boards_numbers'] == 1]
+    result_connected_boards = now_zt_pool_df.loc[now_zt_pool_df['connected_boards_numbers'] > 1]
+    result_connected_boards = result_connected_boards.sort_values(by=['connected_boards_numbers'], ascending=False)
+    result_first = result_first.sort_values(by=['first_closure_time'], ascending=True)
+
+    now_zt_pool_df = pd.concat([result_connected_boards,
+                                result_first])
+    now_zt_pool_df = now_zt_pool_df.fillna('未知数据')
+    now_zt_pool_df.drop_duplicates('symbol', keep='last', inplace=True)
 
-    em_now_zt_pool = em_now_zt_pool[
-        ['symbol', 'connected_boards_numbers',
-         'statistics', 'closure_funds', 'first_closure_time',
-         'last_closure_time', 'frying_plates_numbers']]
+    return now_zt_pool_df
 
-    em_now_zt_pool = em_now_zt_pool.set_index(['symbol'], drop=True)
-    real_time_zt_df_wei_bi_100 = real_time_zt_df_wei_bi_100.set_index(['symbol'], drop=False)
-    real_time_zt_df_wei_bi_100 = pd.merge(real_time_zt_df_wei_bi_100, em_now_zt_pool, how='outer',
-                                          left_index=True, right_index=True)
 
+def handle_high_chg(real_time_zt_df_high_chg, last_trade_zt, real_time_zt_df_wei_bi_100):
     # 初始化数据
     real_time_zt_df_high_chg['closure_funds'] = 0
     real_time_zt_df_high_chg['first_closure_time'] = '153000'
@@ -231,6 +288,7 @@ def get_now_zt_pool_with_reason(str_day):
 
     last_trade_zt_copy_today_high_chg = last_trade_zt.loc[
         last_trade_zt['symbol'].isin(real_time_zt_df_high_chg_last_trade_zt['symbol'])]
+
     last_trade_zt_copy_today_high_chg = last_trade_zt_copy_today_high_chg[
         ['symbol', 'connected_boards_numbers', 'statistics']]
     last_trade_zt_copy_today_high_chg['connected_boards_numbers'] = last_trade_zt_copy_today_high_chg[
@@ -250,21 +308,10 @@ def get_now_zt_pool_with_reason(str_day):
                                 real_time_zt_df_high_chg_last_trade_zt,
                                 real_time_zt_df_high_chg_last_trade_no_zt])
 
-    now_zt_pool_df = merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt)
-
-    result_first = now_zt_pool_df.loc[now_zt_pool_df['connected_boards_numbers'] == 1]
-    result_connected_boards = now_zt_pool_df.loc[now_zt_pool_df['connected_boards_numbers'] > 1]
-    result_connected_boards = result_connected_boards.sort_values(by=['connected_boards_numbers'], ascending=False)
-    result_first = result_first.sort_values(by=['first_closure_time'], ascending=True)
-
-    now_zt_pool_df = pd.concat([result_connected_boards,
-                                result_first])
-    now_zt_pool_df = now_zt_pool_df.fillna('未知数据')
-    now_zt_pool_df.drop_duplicates('symbol', keep='last', inplace=True)
     return now_zt_pool_df
 
 
-def merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt):
+def merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt, now_continue_zt):
     try:
         zt_reason_df = get_ths_stock_zt_reason_with_cache(str_day)
     except BaseException as e:
@@ -272,9 +319,17 @@ def merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt):
         logger.error("获取涨停原因异常:{}", e)
 
     last_trade_zt_copy = last_trade_zt.copy()
+
     if "zt_reason" not in last_trade_zt.columns:
         last_trade_zt_copy['zt_reason'] = '暂无'
-    last_trade_zt_copy = last_trade_zt_copy[['symbol', 'zt_reason']]
+
+    if 'main_line' not in last_trade_zt.columns:
+        last_trade_zt['main_line'] = ''
+
+    if 'sub_main_line' not in last_trade_zt.columns:
+        last_trade_zt['sub_main_line'] = ''
+
+    last_trade_zt_copy = last_trade_zt_copy[['symbol', 'zt_reason', 'sub_main_line', 'main_line']]
 
     if data_frame_util.is_empty(zt_reason_df):
         now_zt_pool_df['zt_reason'] = '暂无'
@@ -292,6 +347,21 @@ def merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt):
     zt_reason_df_copy = zt_reason_df_copy.set_index(['symbol'], drop=True)
     now_zt_pool_df = now_zt_pool_df.set_index(['symbol'], drop=False)
 
+    if data_frame_util.is_not_empty(now_continue_zt):
+        now_continue_zt = now_continue_zt[['symbol', 'main_line', 'sub_main_line']]
+
+        # 创建一个 symbol 到 main_line 的映射
+        main_line_mapping = now_continue_zt.set_index('symbol')['main_line'].to_dict()
+
+        # 使用 map 更新 main_line 列
+        now_zt_pool_df['main_line'] = now_zt_pool_df['symbol'].map(main_line_mapping).fillna("")
+
+        # 创建一个 symbol 到 main_line 的映射
+        sub_main_line_mapping = now_continue_zt.set_index('symbol')['sub_main_line'].to_dict()
+
+        # 使用 map 更新 sub_main_line 列
+        now_zt_pool_df['sub_main_line'] = now_zt_pool_df['symbol'].map(sub_main_line_mapping).fillna("")
+
     result_zt_df = pd.merge(now_zt_pool_df, zt_reason_df_copy, how='outer',
                             left_index=True, right_index=True)
 
@@ -302,8 +372,12 @@ def merge_zt_reason(now_zt_pool_df, str_day, last_trade_zt):
     last_trade_zt_copy = last_trade_zt_copy.set_index(['symbol'], drop=True)
     del zt_reason_na['zt_reason']
     zt_reason_na = zt_reason_na.set_index(['symbol'], drop=False)
+    del last_trade_zt_copy['main_line']
+    del last_trade_zt_copy['sub_main_line']
+
     zt_reason_na = pd.merge(zt_reason_na, last_trade_zt_copy, how='outer',
                             left_index=True, right_index=True)
+
     zt_reason_na['zt_reason'] = zt_reason_na['zt_reason'].fillna('0')
     # 删除昨日涨停 今日未涨停的
     zt_reason_na.dropna(subset=['symbol'], inplace=True)
@@ -366,14 +440,11 @@ def zt_reason_group(zt_pool_df):
 
 
 if __name__ == '__main__':
-    str_day_test = '2024-10-09'
-
+    str_day_test = '2025-10-28'
+    # 同花顺涨停
+    ths_zt_pool_df = get_now_zt_pool_with_reason(str_day_test)
     # 东财涨停池
     em_now_zt_pool_test = stock_zt_pool_api.stock_em_zt_pool_df(date_handle_util.no_slash_date(str_day_test))
-    # 同花顺涨停
-    ths_zt_pool_df = get_zt_reason(str_day_test)
 
     miss_zt_df = em_now_zt_pool_test.loc[~(em_now_zt_pool_test['symbol'].isin(ths_zt_pool_df['symbol']))]
     pass
-
-

mns_common/api/xueqiu/__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)

mns_common/api/xueqiu/xue_qiu_k_line_api.py

@@ -0,0 +1,83 @@
+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 requests
+import pandas as pd
+
+
+# year 年
+#  quarter 季度
+# month 月度
+# week 周
+# day 日
+def get_xue_qiu_k_line(symbol, period, cookie, end_time, hq):
+    url = "https://stock.xueqiu.com/v5/stock/chart/kline.json"
+
+    params = {
+        "symbol": symbol,
+        "begin": end_time,
+        "period": period,
+        "type": hq,
+        "count": "-120084",
+        "indicator": "kline,pe,pb,ps,pcf,market_capital,agt,ggt,balance"
+    }
+
+    headers = {
+        "accept": "application/json, text/plain, */*",
+        "accept-language": "zh-CN,zh;q=0.9",
+        "origin": "https://xueqiu.com",
+        "priority": "u=1, i",
+        "referer": "https://xueqiu.com/S/SZ300879?md5__1038=n4%2BxgDniDQeWqxYwq0y%2BbDyG%2BYDtODuD7q%2BqRYID",
+        "sec-ch-ua": '"Chromium";v="134", "Not:A-Brand";v="24", "Google Chrome";v="134"',
+        "sec-ch-ua-mobile": "?0",
+        "sec-ch-ua-platform": '"Windows"',
+        "sec-fetch-dest": "empty",
+        "sec-fetch-mode": "cors",
+        "sec-fetch-site": "same-site",
+        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/134.0.0.0 Safari/537.36",
+        "cookie": cookie
+    }
+
+    response = requests.get(
+        url=url,
+        params=params,
+        headers=headers
+    )
+
+    if response.status_code == 200:
+        response_data = response.json()
+        df = pd.DataFrame(
+            data=response_data['data']['item'],
+            columns=response_data['data']['column']
+        )
+
+        # 1. 转换为 datetime（自动处理毫秒级时间戳）
+        df["beijing_time"] = pd.to_datetime(df["timestamp"], unit="ms")
+
+        # 2. 设置 UTC 时区
+        df["beijing_time"] = df["beijing_time"].dt.tz_localize("UTC")
+
+        # 3. 转换为北京时间（UTC+8）
+        df["beijing_time"] = df["beijing_time"].dt.tz_convert("Asia/Shanghai")
+
+        # 4. 提取年月日（格式：YYYY-MM-DD）
+        df["str_day"] = df["beijing_time"].dt.strftime("%Y-%m-%d %H:%M:%S")
+        del df["beijing_time"]
+
+        return df
+    else:
+        # 直接抛出带有明确信息的异常
+        raise ValueError("调用雪球接口失败")
+
+
+if __name__ == '__main__':
+    number = 1
+    cookies ='cookiesu=291756518429127; device_id=8c5eef4332e51dbcbcd92943b0dd91e4; s=bz17174k1c; xq_a_token=b9c7e702181cba3ed732d5019efe2dfe2fb054b0; xqat=b9c7e702181cba3ed732d5019efe2dfe2fb054b0; xq_r_token=c1edaf05e1c6fdf8122671eced8049e8df8a4290; xq_id_token=eyJ0eXAiOiJKV1QiLCJhbGciOiJSUzI1NiJ9.eyJ1aWQiOi0xLCJpc3MiOiJ1YyIsImV4cCI6MTc2MTYxNDEwNywiY3RtIjoxNzU5OTc2NzUxOTI3LCJjaWQiOiJkOWQwbjRBWnVwIn0.JwQMnAT6E1lPSqfK-GnFpLNQz-Jr-xtZ6HoUfnCfeRTNKrDJqf0l5_iY7dhEo6A5m38MmXj0hgR-xYkzPm35fFcQCQnZ06dOK0XCP6AE9r5KVFd73GyqbXLijKDFPQVM0sPgbgCs9fTOmB7Nlju_7B_raPmZ8IIHzZRU6uSK9oXsrnwpo4TD0Mr2nMF3ktVwEQHdkZsjm9mdSZaT9fkDvayFKKmg_O-JbHDZ6mTF1i-zR6oH-5r3g6HJJU9-_NwzZMeBEVnLG8IWwO8_n9CWdpbqQ-qC-iue4yEHFdiTLBCRRgyFauVv0kJS1ZNKvvefxkETTVr-WgXobT82ED7ApA; u=291756518429127; Hm_lvt_1db88642e346389874251b5a1eded6e3=1759765103,1759808830,1759970243,1759976801; HMACCOUNT=4C4F802C04F4D70A; Hm_lpvt_1db88642e346389874251b5a1eded6e3=1759978805'
+    while True:
+        test_df = get_xue_qiu_k_line('SZ300188', '1m', cookies, '1760065317680', '')
+        print(number)
+        number = number + 1

mns_common/component/__init__.py

@@ -4,4 +4,4 @@ import os
 file_path = os.path.abspath(__file__)
 end = file_path.index('mns') + 14
 project_path = file_path[0:end]
-sys.path.append(project_path)
+sys.path.append(project_path)

mns_common/component/classify/symbol_classify_api.py

@@ -35,26 +35,26 @@ def set_stock_type_by_deal_days(real_time_quotes_now_init):
     real_time_quotes_now.loc[:, "stock_type"] = stock_type_classify_param['normal_stock']
     # 交易天数五个交易日的股票
     real_time_quotes_now.loc[(real_time_quotes_now['name'].str.startswith('C'))
-                             | (real_time_quotes_now['deal_days'] <= stock_type_classify_param[
+                             | (real_time_quotes_now['deal_days'] < stock_type_classify_param[
         'sub_stock_new_min_deal_days']), "stock_type"] = stock_type_classify_param['new_stock']
 
     # 交易上市6-100天的次新股票
     real_time_quotes_now.loc[
         (real_time_quotes_now['stock_type'] != stock_type_classify_param['new_stock'])
-        & (real_time_quotes_now['deal_days'] > stock_type_classify_param['sub_stock_new_min_deal_days'])
-        & (real_time_quotes_now['deal_days'] <= stock_type_classify_param['sub_new_stock_max_deal_days']),
+        & (real_time_quotes_now['deal_days'] >= stock_type_classify_param['sub_stock_new_min_deal_days'])
+        & (real_time_quotes_now['deal_days'] < stock_type_classify_param['sub_new_stock_max_deal_days']),
         "stock_type"] = stock_type_classify_param['sub_stock_new']
 
     # 上市次新股 100-365
     real_time_quotes_now.loc[
-        (real_time_quotes_now['deal_days'] > stock_type_classify_param['sub_new_stock_max_deal_days'])
-        & (real_time_quotes_now['deal_days'] <= stock_type_classify_param['sub_stock_max_deal_days']),
+        (real_time_quotes_now['deal_days'] >= stock_type_classify_param['sub_new_stock_max_deal_days'])
+        & (real_time_quotes_now['deal_days'] < stock_type_classify_param['sub_stock_max_deal_days']),
         "stock_type"] = stock_type_classify_param['sub_stock']
 
     # 交易天数 366-730
     real_time_quotes_now.loc[
-        (real_time_quotes_now['deal_days'] > stock_type_classify_param['sub_stock_max_deal_days'])
-        & (real_time_quotes_now['deal_days'] <= stock_type_classify_param['normal_stock_max_deal_days']),
+        (real_time_quotes_now['deal_days'] >= stock_type_classify_param['sub_stock_max_deal_days'])
+        & (real_time_quotes_now['deal_days'] < stock_type_classify_param['normal_stock_max_deal_days']),
         "stock_type"] = stock_type_classify_param['normal_sub_stock']
 
     return real_time_quotes_now

mns_common/component/common_service_fun_api.py

@@ -22,9 +22,9 @@ TEN_THOUSAND = 10000
 
 
 def total_mv_classification(real_time_quotes_now):
-    real_time_quotes_now['flow_mv_sp'] = round((real_time_quotes_now['flow_mv'] / HUNDRED_MILLION), 2)
-    real_time_quotes_now['total_mv_sp'] = round((real_time_quotes_now['total_mv'] / HUNDRED_MILLION), 2)
-    real_time_quotes_now['flow_mv_level'] = 0
+    real_time_quotes_now.loc[:, 'flow_mv_sp'] = round((real_time_quotes_now['flow_mv'] / HUNDRED_MILLION), 2)
+    real_time_quotes_now.loc[:, 'total_mv_sp'] = round((real_time_quotes_now['total_mv'] / HUNDRED_MILLION), 2)
+    real_time_quotes_now.loc[:, 'flow_mv_level'] = 0
 
     real_time_quotes_now.reset_index(drop=True, inplace=True)
     real_time_quotes_now.loc[(real_time_quotes_now["flow_mv_sp"] >= 0), ['flow_mv_level']] \
@@ -39,6 +39,20 @@ def classify_symbol(real_time_quotes_now_df):
     return real_time_quotes_now_df
 
 
+# 增加前缀
+def add_pre_prefix(real_time_quotes_now_df):
+    real_time_quotes_now_df['symbol_prefix'] = real_time_quotes_now_df['symbol'].apply(
+        lambda symbol: add_pre_prefix_one(symbol))
+    return real_time_quotes_now_df
+
+
+# 增加后缀
+def add_after_prefix(real_time_quotes_now_df):
+    real_time_quotes_now_df['symbol_prefix'] = real_time_quotes_now_df['symbol'].apply(
+        lambda symbol: add_after_prefix_one(symbol))
+    return real_time_quotes_now_df
+
+
 # 单个股票分类
 def classify_symbol_one(symbol):
     if symbol.startswith('3'):
@@ -54,6 +68,41 @@ def classify_symbol_one(symbol):
         return 'X'
 
 
+# 添加前缀
+def add_pre_prefix_one(symbol):
+    symbol_simple = symbol[0:6]
+    if bool(1 - is_valid_symbol(symbol_simple)):
+        return symbol
+
+    if symbol_simple.startswith('6'):
+        return 'SH' + symbol_simple
+    elif symbol_simple.startswith('0') or symbol_simple.startswith('3'):
+        return 'SZ' + symbol_simple
+    else:
+        return 'BJ' + symbol_simple
+
+
+def is_valid_symbol(symbol):
+    # 确保输入是字符串（避免数字或其他类型）
+    if not isinstance(symbol, str):
+        return False
+    # 检查长度是否为6且所有字符都是数字
+    return len(symbol) == 6 and symbol.isdigit()
+
+
+# 添加后缀
+def add_after_prefix_one(symbol):
+    symbol_simple = symbol[0:6]
+    if bool(1 - is_valid_symbol(symbol_simple)):
+        return symbol
+    if symbol_simple.startswith('6'):
+        return symbol_simple + '.SH'
+    elif symbol_simple.startswith('0') or symbol_simple.startswith('3'):
+        return symbol_simple + '.SZ'
+    else:
+        return symbol_simple + '.BJ'
+
+
 def symbol_amount_simple(real_time_quotes_now_df):
     real_time_quotes_now_df['amount_level'] = round(real_time_quotes_now_df['amount'] / HUNDRED_MILLION, 2)
     return real_time_quotes_now_df
@@ -82,8 +131,7 @@ def exclude_st_symbol(real_time_quotes_now_df):
 # 排除带星的ST 容易退市
 def exclude_star_st_symbol(df):
     exclude_st_symbol_list = list(
-        df.loc[(df['name'].str.contains('\*'))
-               | (df['name'].str.contains('退'))]['symbol'])
+        df.loc[(df['name'].str.contains(r'\*')) | (df['name'].str.contains('退'))]['symbol'])
     return df.loc[
         ~(df['symbol'].isin(
             exclude_st_symbol_list))]
@@ -145,6 +193,18 @@ def group_by_industry_sum(real_time_quotes_now, field1, field2):
     return pd.DataFrame({field1: df_series.index, field2: df_series.values})
 
 
+def symbol_add_prefix(symbol):
+    if bool(1 - symbol.isdigit()):
+        return symbol
+    symbol_simple = symbol[0:6]
+    if symbol_simple.startswith('6'):
+        return '1.' + symbol_simple
+    elif symbol_simple.startswith('0') or symbol_simple.startswith('3'):
+        return '0.' + symbol_simple
+    else:
+        return '0.' + symbol_simple
 
 
-
+# 排除改变代码的北交所
+def exclude_change_bjs_code(df):
+    return df[~df['symbol'].str.startswith(('8', '4'))]

mns_common/component/company/company_common_service_api.py

@@ -15,6 +15,26 @@ import mns_common.constant.db_name_constant as db_name_constant
 mongodb_util = MongodbUtil('27017')
 
 
+@lru_cache(maxsize=None)
+def get_company_info_name():
+    return mongodb_util.find_query_data_choose_field('company_info', {},
+                                                     {"_id": 1, 'name': 1,
+                                                      "industry": 1, "company_type": 1,
+                                                      "ths_concept_name": 1, "ths_concept_code": 1,
+                                                      "ths_concept_sync_day": 1, 'sub_stock': 1,
+                                                      "first_sw_industry": 1, "second_sw_industry": 1,
+                                                      "third_sw_industry": 1, "mv_circulation_ratio": 1,
+                                                      "diff_days": 1,
+                                                      'em_industry': 1,
+                                                      'ths_concept_list_info': 1,
+                                                      'ths_concept_name_list_str': 1,
+                                                      "kpl_plate_name": 1,
+                                                      "kpl_plate_list_info": 1,
+                                                      'operate_profit': 1,
+                                                      'total_operate_income': 1
+                                                      })
+
+
 @lru_cache(maxsize=None)
 def get_company_info_industry():
     return mongodb_util.find_query_data_choose_field('company_info', {},
@@ -26,7 +46,7 @@ def get_company_info_industry():
                                                       "diff_days": 1,
                                                       'em_industry': 1,
                                                       'ths_concept_list_info': 1,
-                                                      'ths_concept_name_list_str':1,
+                                                      'ths_concept_name_list_str': 1,
                                                       "kpl_plate_name": 1,
                                                       "kpl_plate_list_info": 1,
                                                       'operate_profit': 1,

mns_common/component/company/company_common_service_new_api.py

@@ -9,6 +9,8 @@ from functools import lru_cache
 from mns_common.db.MongodbUtil import MongodbUtil
 import mns_common.constant.db_name_constant as db_name_constant
 import pandas as pd
+pd.set_option('future.no_silent_downcasting', True)
+
 
 mongodb_util = MongodbUtil('27017')
 
@@ -102,7 +104,8 @@ def amend_ths_industry(real_time_quotes_now_init):
     na_industry_data = na_industry_data.loc[na_industry_data['amount'] != 0]
 
     real_time_quotes_now_result = pd.concat([real_time_quotes_now, na_industry_data], axis=0)
-    real_time_quotes_now['mv_circulation_ratio'].fillna(1, inplace=True)
+    real_time_quotes_now.fillna({'connected_boards_numbers': 1}, inplace=True)
+
     # 处理na值
     real_time_quotes_now_result.fillna(0, inplace=True)
     return real_time_quotes_now_result

mns_common/component/cookie/cookie_enum.py

@@ -0,0 +1,15 @@
+from enum import Enum
+
+'''
+价格枚举
+'''
+
+
+class CookieEnum(Enum):
+    THS_COOKIE = ('ths_cookie', '同花顺cookie')
+    EM_COOKIE = ('em_cookie', '东方财富cookie')
+    XUE_QIU_COOKIE = ('xue_qiu_cookie', '雪球cookie')
+
+    def __init__(self, cookie_code, cookie_name):
+        self.cookie_code = cookie_code
+        self.cookie_name = cookie_name

mns_common/component/cookie/cookie_info_service.py

@@ -9,18 +9,23 @@ from mns_common.db.MongodbUtil import MongodbUtil
 
 mongodb_util = MongodbUtil('27017')
 
-from functools import lru_cache
 
-
-@lru_cache(maxsize=None)
+# ths cookie
 def get_ths_cookie():
     stock_account_info = mongodb_util.find_query_data('stock_account_info', {"type": "ths_cookie", })
     ths_cookie = list(stock_account_info['cookie'])[0]
     return ths_cookie
 
 
-@lru_cache(maxsize=None)
+# 东财 cookie
 def get_em_cookie():
     stock_account_info = mongodb_util.find_query_data('stock_account_info', {"type": "em_cookie", })
     em_cookie = list(stock_account_info['cookie'])[0]
     return em_cookie
+
+
+# 雪球 cookie
+def get_xue_qiu_cookie():
+    stock_account_info = mongodb_util.find_query_data('stock_account_info', {"type": "xue_qiu_cookie", })
+    cookie = list(stock_account_info['cookie'])[0]
+    return cookie

mns_common/component/data/data_init_api.py

@@ -60,16 +60,21 @@ def calculate_parameter_factor(real_time_quotes_now):
     real_time_quotes_now['large_inflow_multiple'] = round(
         (real_time_quotes_now['large_order_net_inflow'] / real_time_quotes_now['reference_main_inflow']), 2)
 
-    real_time_quotes_now.loc[:, 'real_disk_diff_amount_exchange'] = round(
-        (real_time_quotes_now['disk_diff_amount_exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+    if 'real_disk_diff_amount_exchange' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_disk_diff_amount_exchange'] = round(
+            (real_time_quotes_now['disk_diff_amount_exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
 
-    real_time_quotes_now.loc[:, 'real_main_inflow_multiple'] = round(
-        (real_time_quotes_now['main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+    if 'real_main_inflow_multiple' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_main_inflow_multiple'] = round(
+            (real_time_quotes_now['main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
 
-    real_time_quotes_now.loc[:, 'real_super_main_inflow_multiple'] = round(
-        (real_time_quotes_now['super_main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
-    real_time_quotes_now.loc[:, 'real_exchange'] = round(
-        (real_time_quotes_now['exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+    if 'real_super_main_inflow_multiple' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_super_main_inflow_multiple'] = round(
+            (real_time_quotes_now['super_main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+
+    if 'real_exchange' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_exchange'] = round(
+            (real_time_quotes_now['exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
 
     real_time_quotes_now.loc[:, 'max_real_main_inflow_multiple'] = real_time_quotes_now[
         ['real_main_inflow_multiple', 'real_super_main_inflow_multiple']].max(axis=1)