mns-common 1.5.1.0__py3-none-any.whl → 1.5.1.2__py3-none-any.whl

mns_common/api/akshare/stock_zt_pool_api.py

@@ -43,5 +43,5 @@ def stock_em_zt_pool_df(date):
 
 
 if __name__ == '__main__':
-    df = stock_em_zt_pool_df('2024-09-04')
+    df = stock_em_zt_pool_df('2025-10-24')
     print(df)

mns_common/api/ths/zt/ths_stock_zt_pool_v2_api.py

@@ -140,7 +140,8 @@ def convert_statistics(stat):
 # 获取实时行情涨停列表
 def get_real_time_zt_info():
     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_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[[
@@ -197,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)
 
-    em_now_zt_pool = em_now_zt_pool[
-        ['symbol', 'connected_boards_numbers',
-         'statistics', 'closure_funds', 'first_closure_time',
-         'last_closure_time', 'frying_plates_numbers']]
+            real_time_zt_df_wei_bi_100 = real_time_zt_df_wei_bi_100.sort_values(by=['connected_boards_numbers'],
+                                                                                ascending=False)
 
-    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)
+        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)
 
+    return now_zt_pool_df
+
+
+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'
@@ -228,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[
@@ -247,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:
@@ -269,9 +319,14 @@ 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'] = ''
+
+    last_trade_zt_copy = last_trade_zt_copy[['symbol', 'zt_reason', 'main_line']]
 
     if data_frame_util.is_empty(zt_reason_df):
         now_zt_pool_df['zt_reason'] = '暂无'
@@ -289,6 +344,16 @@ 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)
 
+    now_zt_pool_df['main_line'] = ''
+    if data_frame_util.is_not_empty(now_continue_zt):
+        now_continue_zt = now_continue_zt[['symbol', '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("")
+
     result_zt_df = pd.merge(now_zt_pool_df, zt_reason_df_copy, how='outer',
                             left_index=True, right_index=True)
 
@@ -363,12 +428,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-1.5.1.0.dist-info → mns_common-1.5.1.2.dist-info}/METADATA

@@ -1,4 +1,4 @@
 Metadata-Version: 2.1
 Name: mns-common
-Version: 1.5.1.0
+Version: 1.5.1.2
 

{mns_common-1.5.1.0.dist-info → mns_common-1.5.1.2.dist-info}/RECORD

@@ -5,7 +5,7 @@ mns_common/api/akshare/k_line_api.py,sha256=IfqJDG2e-_Ry1f_MqjIff6jad9IIC3Tjnpma
 mns_common/api/akshare/stock_bid_ask_api.py,sha256=Af9t6Pv_-p7PJJ_7rF_JVaGBomkvePMMqALwuBh2Gfw,4139
 mns_common/api/akshare/stock_dt_pool.py,sha256=sKedOTzqsBZprJHJEr2sRYa8xbeSK7tRenqBE3wOdUc,2245
 mns_common/api/akshare/stock_zb_pool.py,sha256=ylcVgZTzsD3qz9WaItLtoqTf4t-Ex49MDtZUk0xaB5Q,2126
-mns_common/api/akshare/stock_zt_pool_api.py,sha256=_4PG_Wd88S4zweKNwZprwxaMMHBF4OQSB77DGW1l9UQ,1929
+mns_common/api/akshare/stock_zt_pool_api.py,sha256=-j_GZI3TTPV8XWnczgGjct5XZAHZdlptBcECv7Rfyzw,1929
 mns_common/api/akshare/yjyg_sync_api.py,sha256=cvk50_XhJWUqduOiC15SYvQTCQqECt6td_L2Hvnl7Jg,4108
 mns_common/api/em/__init__.py,sha256=2U9DiKslxsWwLLEcZKjS8UiQPN1QgALvnK3HiJNIZE0,163
 mns_common/api/em/concept/__init__.py,sha256=wEg73KlZo-dU0yKGwpA1C2y6LZm4IBb94tNda1tqLeg,163
@@ -68,7 +68,7 @@ mns_common/api/ths/wen_cai/__init__.py,sha256=2U9DiKslxsWwLLEcZKjS8UiQPN1QgALvnK
 mns_common/api/ths/wen_cai/ths_wen_cai_api.py,sha256=zDwZyh60sHdJ_folWDVZmLD1mndKc40Ltc4JbL12HF4,4935
 mns_common/api/ths/zt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mns_common/api/ths/zt/ths_stock_zt_pool_api.py,sha256=NAfyOWWROVpCRlw3jcCxPLQMwR8m0jt9tfIZT7Jr1UE,10690
-mns_common/api/ths/zt/ths_stock_zt_pool_v2_api.py,sha256=eICTYaHJYp2ccsHkeDnkzpJGT63FzKtaICDjCWCdQOA,16282
+mns_common/api/ths/zt/ths_stock_zt_pool_v2_api.py,sha256=0XCN98bEo_TsYim94ELln5tyiNvznJoYjq-ZoPZTdtk,19510
 mns_common/api/us/__init__.py,sha256=wEg73KlZo-dU0yKGwpA1C2y6LZm4IBb94tNda1tqLeg,163
 mns_common/api/us/ths_us_company_info_api.py,sha256=qQjv4F-ovQ2uuu-FlBAnxjvVA7qj9y_x5WZtUoyoEW4,241
 mns_common/api/xueqiu/__init__.py,sha256=wEg73KlZo-dU0yKGwpA1C2y6LZm4IBb94tNda1tqLeg,163
@@ -158,7 +158,7 @@ mns_common/utils/date_handle_util.py,sha256=XS-MyA8_7k35LOCFAYOHgVcVkMft_Kc4Wa9U
 mns_common/utils/db_util.py,sha256=hSmfNAN4vEeEaUva6_cicZEhb2jSnib-Gvk2reke1vc,2590
 mns_common/utils/file_util.py,sha256=egWu6PenGPRp_ixrNTHKarT4dAnOT6FETR82EHUZJnQ,1042
 mns_common/utils/ip_util.py,sha256=UTcYfz_uytB__6nlBf7T-izuI7hi4XdB6ET0sJgEel4,969
-mns_common-1.5.1.0.dist-info/METADATA,sha256=LZl22-oEFAdnlTofLKZnGkkmd-Qsbpna0oSwS7Ji48c,61
-mns_common-1.5.1.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-mns_common-1.5.1.0.dist-info/top_level.txt,sha256=ZC58kAR-8Hvc6U2xhYNBNLAh3mb6sZazbdj5nZpvEkQ,11
-mns_common-1.5.1.0.dist-info/RECORD,,
+mns_common-1.5.1.2.dist-info/METADATA,sha256=DyC9VPrDz4P1fCi07rp1zSndJ1FCgbfnoHKYQUXuN8M,61
+mns_common-1.5.1.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+mns_common-1.5.1.2.dist-info/top_level.txt,sha256=ZC58kAR-8Hvc6U2xhYNBNLAh3mb6sZazbdj5nZpvEkQ,11
+mns_common-1.5.1.2.dist-info/RECORD,,