mns-scheduler 1.1.7.9__tar.gz → 1.1.8.2__tar.gz

{mns_scheduler-1.1.7.9 → mns_scheduler-1.1.8.2}/PKG-INFO

@@ -1,3 +1,3 @@
 Metadata-Version: 2.1
 Name: mns-scheduler
-Version: 1.1.7.9
+Version: 1.1.8.2

{mns_scheduler-1.1.7.9 → mns_scheduler-1.1.8.2}/mns_scheduler/concept/ths/common/ths_concept_sync_common_api.py

@@ -175,7 +175,7 @@ def update_long_short(new_concept_symbol_df):
     for new_concept_one in new_concept_symbol_df.itertuples():
         try:
             query = {'symbol': new_concept_one.symbol, 'concept_code': new_concept_one.concept_code}
-            new_values = {'long': new_concept_one.long, "short": new_concept_one.short}
+            new_values = {"$set":{'long': new_concept_one.long, "short": new_concept_one.short}}
             mongodb_util.update_many(query, new_values, db_name_constant.THS_STOCK_CONCEPT_DETAIL)
             mongodb_util.update_many(query, new_values, db_name_constant.THS_STOCK_CONCEPT_DETAIL_APP)
         except BaseException as e:

{mns_scheduler-1.1.7.9 → mns_scheduler-1.1.8.2}/mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py

@@ -16,6 +16,9 @@ import mns_common.utils.data_frame_util as data_frame_util
 import mns_common.component.common_service_fun_api as common_service_fun_api
 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_pool_v2_api as ths_stock_zt_pool_v2_api
+import mns_common.component.zt.zt_common_service_api as zt_common_service_api
+import mns_common.api.em.east_money_stock_api as east_money_stock_api
+from datetime import datetime
 
 '''
 东方财富涨停池
@@ -50,6 +53,9 @@ def save_zt_info(str_day):
     stock_em_zt_pool_df_data = stock_zt_pool_api.stock_em_zt_pool_df(
         date_handle_util.no_slash_date(str_day))
 
+    # fix 涨停池没有的股票
+    stock_em_zt_pool_df_data = sync_miss_zt_data(stock_em_zt_pool_df_data, str_day)
+
     stock_em_zt_pool_df_data = common_service_fun_api.total_mv_classification(stock_em_zt_pool_df_data)
 
     stock_em_zt_pool_df_data = common_service_fun_api.classify_symbol(stock_em_zt_pool_df_data)
@@ -57,6 +63,10 @@ def save_zt_info(str_day):
     stock_em_zt_pool_df_data = common_service_fun_api.symbol_amount_simple(stock_em_zt_pool_df_data)
 
     stock_em_zt_pool_df_data = company_common_service_api.amendment_industry(stock_em_zt_pool_df_data)
+
+    # 上个交易交易日涨停股票
+    last_trade_day_zt_df = zt_common_service_api.get_last_trade_day_zt(str_day)
+
     try:
         # 同花顺问财涨停池
         ths_zt_pool_df_data = ths_stock_zt_pool_v2_api.get_ths_stock_zt_reason_with_cache(str_day)
@@ -69,6 +79,10 @@ def save_zt_info(str_day):
                 stock_em_zt_pool_df_data = ths_concept_common_service_api.set_last_ths_concept(stock_one.symbol,
                                                                                                stock_em_zt_pool_df_data,
                                                                                                str_day)
+                # 设置连板数目
+                stock_em_zt_pool_df_data = set_connected_boards_numbers(stock_em_zt_pool_df_data,
+                                                                        stock_one.symbol, last_trade_day_zt_df)
+
                 ths_zt_pool_one_df = ths_zt_pool_df_data.loc[ths_zt_pool_df_data['symbol'] == stock_one.symbol]
                 if data_frame_util.is_empty(ths_zt_pool_one_df):
                     stock_em_zt_pool_df_data['zt_reason'] = '0'
@@ -121,8 +135,61 @@ def save_zt_info(str_day):
     return stock_em_zt_pool_df_data
 
 
+# 设置连板数目
+def set_connected_boards_numbers(stock_em_zt_pool_df_data, symbol, last_trade_day_zt_df):
+    connected_boards_df = 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_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):
+        connected_boards_df_one.loc[connected_boards_df_one['symbol'] == symbol, 'connected_boards_numbers'] = \
+            list(connected_boards_df_one['connected_boards_numbers'])[0]
+
+    return stock_em_zt_pool_df_data
+
+
+def sync_miss_zt_data(stock_em_zt_pool_df_data, str_day):
+    now_date = datetime.now()
+    now_day = now_date.strftime('%Y-%m-%d')
+    if now_day == str_day:
+        real_time_quotes_all_stocks_df = east_money_stock_api.get_real_time_quotes_all_stocks()
+        real_time_quotes_all_stocks_df = real_time_quotes_all_stocks_df.loc[
+            (real_time_quotes_all_stocks_df['wei_bi'] == 100) & (real_time_quotes_all_stocks_df['chg'] >= 9)]
+        miss_zt_data_df = real_time_quotes_all_stocks_df.loc[~(
+            real_time_quotes_all_stocks_df['symbol'].isin(stock_em_zt_pool_df_data['symbol']))]
+        if data_frame_util.is_not_empty(miss_zt_data_df):
+            miss_zt_data_df['buy_1_num'] = miss_zt_data_df['buy_1_num'].astype(float)
+            miss_zt_data_df['now_price'] = miss_zt_data_df['now_price'].astype(float)
+            miss_zt_data_df['closure_funds'] = round(miss_zt_data_df['buy_1_num'] * 100 * miss_zt_data_df['now_price'],
+                                                     2)
+            miss_zt_data_df = miss_zt_data_df[[
+                'symbol',
+                'name',
+                'chg',
+                'now_price',
+                'amount',
+                'flow_mv',
+                'total_mv',
+                'exchange',
+                'industry',
+                'closure_funds'
+
+            ]]
+            miss_zt_data_df['index'] = 10000
+            miss_zt_data_df['first_closure_time'] = '150000'
+            miss_zt_data_df['last_closure_time'] = '150000'
+            miss_zt_data_df['statistics'] = '1/1'
+            miss_zt_data_df['frying_plates_numbers'] = 0
+            miss_zt_data_df['connected_boards_numbers'] = 0
+
+            stock_em_zt_pool_df_data = pd.concat([miss_zt_data_df, stock_em_zt_pool_df_data])
+        return stock_em_zt_pool_df_data
+
+
 if __name__ == '__main__':
-    save_zt_info('2024-10-18')
+    save_zt_info('2024-10-30')
 # from datetime import datetime
 #
 # if __name__ == '__main__':

{mns_scheduler-1.1.7.9 → mns_scheduler-1.1.8.2}/mns_scheduler.egg-info/PKG-INFO

@@ -1,3 +1,3 @@
 Metadata-Version: 2.1
 Name: mns-scheduler
-Version: 1.1.7.9
+Version: 1.1.8.2

{mns_scheduler-1.1.7.9 → mns_scheduler-1.1.8.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='mns-scheduler',
-    version='1.1.7.9',
+    version='1.1.8.2',
     packages=find_packages(),
     install_requires=[],  # 如果有依赖项，可以在这里列出
 )