mns-scheduler 1.1.5.9__py3-none-any.whl → 1.1.6.0__py3-none-any.whl

mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py

@@ -7,7 +7,6 @@ project_path = file_path[0:end]
 sys.path.append(project_path)
 import pandas as pd
 import mns_common.api.akshare.stock_zt_pool_api as stock_zt_pool_api
-import mns_common.api.ths.zt.ths_stock_zt_pool_api as ths_stock_zt_pool_api
 import mns_common.utils.date_handle_util as date_handle_util
 import mns_common.component.concept.ths_concept_common_service_api as ths_concept_common_service_api
 from mns_common.db.MongodbUtil import MongodbUtil
@@ -72,19 +71,15 @@ def save_zt_info(str_day):
                                                                                                str_day)
                 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'
                     continue
                 stock_em_zt_pool_df_data.loc[stock_em_zt_pool_df_data['symbol'] == stock_one.symbol, 'zt_reason'] = \
                     list(ths_zt_pool_one_df['zt_reason'])[0]
             except BaseException as e:
+                stock_em_zt_pool_df_data['zt_reason'] = '0'
                 logger.error("出现异常:{}", e)
     except BaseException as e:
-        stock_em_zt_pool_df_data['zt_reason'] = ''
-        stock_em_zt_pool_df_data['quantity_ratio'] = 0
-        stock_em_zt_pool_df_data['high'] = 0
-        stock_em_zt_pool_df_data['low'] = 0
-        stock_em_zt_pool_df_data['open'] = 0
-        stock_em_zt_pool_df_data['wei_bi'] = 0
-        stock_em_zt_pool_df_data['buy_1_num'] = 0
+        stock_em_zt_pool_df_data['zt_reason'] = '0'
         logger.error("出现异常:{}", e)
 
     stock_em_zt_pool_df_data['list_date'] = stock_em_zt_pool_df_data['list_date'].apply(
@@ -112,12 +107,22 @@ def save_zt_info(str_day):
     stock_em_zt_pool_df_data['_id'] = stock_em_zt_pool_df_data['symbol'] + "_" + str_day
 
     stock_em_zt_pool_df_data = stock_em_zt_pool_df_data[ZT_FIELD]
-    mongodb_util.save_mongo(stock_em_zt_pool_df_data, 'stock_zt_pool')
+    exist_zt_pool_today = mongodb_util.find_query_data('stock_zt_pool', {'str_day': str_day})
+    if data_frame_util.is_empty(exist_zt_pool_today):
+        mongodb_util.save_mongo(stock_em_zt_pool_df_data, 'stock_zt_pool')
+    else:
+        exist_zt_pool_today_not_nan = exist_zt_pool_today.loc[exist_zt_pool_today['zt_reason'] != '0']
+
+        stock_em_zt_pool_df_data = stock_em_zt_pool_df_data.loc[:, ~stock_em_zt_pool_df_data.columns.duplicated()]
+        stock_em_zt_pool_df_data_new = stock_em_zt_pool_df_data.loc[~(
+            stock_em_zt_pool_df_data['symbol'].isin(exist_zt_pool_today_not_nan['symbol']))]
+        if data_frame_util.is_not_empty(stock_em_zt_pool_df_data_new):
+            mongodb_util.save_mongo(stock_em_zt_pool_df_data_new, 'stock_zt_pool')
     return stock_em_zt_pool_df_data
 
 
 if __name__ == '__main__':
-    save_zt_info('2024-09-30')
+    save_zt_info('2024-10-08')
 # from datetime import datetime
 #
 # if __name__ == '__main__':

mns_scheduler/zt/zt_pool/update_null_zt_reason_api.py

@@ -0,0 +1,45 @@
+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 mns_common.db.MongodbUtil import MongodbUtil
+
+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
+
+
+def update_null_zt_reason(str_day):
+    query = {"str_day": str_day, "$or": [{"zt_reason": "0", }, {"zt_reason": float('nan')}]}
+    stock_zt_pool_df_null_zt_reason = mongodb_util.find_query_data('stock_zt_pool', query)
+    if data_frame_util.is_empty(stock_zt_pool_df_null_zt_reason):
+        return None
+    no_reason_list = list(stock_zt_pool_df_null_zt_reason['symbol'])
+    repeat_number = 0
+    # 循环10次
+    while len(no_reason_list) > 0 and repeat_number < 10:
+        ths_zt_pool_df_data = ths_stock_zt_pool_api.get_zt_reason(str_day)
+        for stock_zt_one in stock_zt_pool_df_null_zt_reason.itertuples():
+            try:
+                ths_pool_zt_reason_one_df = ths_zt_pool_df_data.loc[
+                    ths_zt_pool_df_data['symbol'] == stock_zt_one.symbol]
+                if data_frame_util.is_not_empty(ths_pool_zt_reason_one_df):
+                    stock_zt_pool_df_one_df = stock_zt_pool_df_null_zt_reason.loc[
+                        stock_zt_pool_df_null_zt_reason['symbol'] == stock_zt_one.symbol]
+                    stock_zt_pool_df_one_df['zt_reason'] = list(ths_pool_zt_reason_one_df['zt_reason'])[0]
+                    mongodb_util.save_mongo(stock_zt_pool_df_one_df, 'stock_zt_pool')
+                    no_reason_list.remove(stock_zt_one.symbol)
+            except BaseException as e:
+                logger.error("出现异常:{},{}", stock_zt_one.symbol, e)
+                continue
+        repeat_number = repeat_number + 1
+
+    return stock_zt_pool_df_null_zt_reason
+
+
+if __name__ == '__main__':
+    update_null_zt_reason('2024-10-08')

mns_scheduler/zz_task/data_sync_task.py

@@ -54,6 +54,7 @@ import mns_scheduler.company_info.remark.company_remark_info_sync as company_rem
 import mns_scheduler.finance.sync_financial_report_service_api as sync_financial_report_service_api
 import mns_scheduler.hk.hk_industry_info_sync_service_api as hk_industry_info_sync_service_api
 import mns_scheduler.hk.hk_company_info_sync_service_api as hk_company_info_sync_service_api
+import mns_scheduler.zt.zt_pool.update_null_zt_reason_api as update_null_zt_reason_api
 
 
 # 同步交易日期任务完成
@@ -384,6 +385,15 @@ def sync_hk_company_industry_info():
     hk_industry_info_sync_service_api.sync_hk_company_industry()
 
 
+# 更新空的涨停原因股票
+def update_null_zt_reason():
+    now_date = datetime.now()
+    str_now_day = now_date.strftime('%Y-%m-%d')
+    if trade_date_common_service_api.is_trade_day(str_now_day):
+        update_null_zt_reason_api.update_null_zt_reason(str_now_day)
+        logger.info("更新空涨停原因信息:{}", str_now_day)
+
+
 # # 定义BlockingScheduler
 blockingScheduler = BlockingScheduler()
 # sync_trade_date 同步交易日期
@@ -444,7 +454,7 @@ blockingScheduler.add_job(sync_all_kpl_plate_info, 'interval', minutes=5, max_in
 # 同步公司基本信息
 blockingScheduler.add_job(update_company_base_info, 'cron', hour='08,18', minute='05')
 # 更新当天涨停股票池
-blockingScheduler.add_job(sync_stock_zt_pool, 'cron', hour='15,19,21,23', minute='05')
+blockingScheduler.add_job(sync_stock_zt_pool, 'cron', hour='15,19,', minute='05')
 
 # 自动打新 打新中签高时间段 10:30-11:30
 blockingScheduler.add_job(auto_ipo_buy, 'cron', hour='10', minute='40,50')
@@ -478,6 +488,8 @@ blockingScheduler.add_job(sync_financial_report_task, 'cron', hour='17', minute=
 
 # 同步港股公司和行业信息
 blockingScheduler.add_job(sync_hk_company_industry_info, 'cron', hour='08,15', minute='10')
+# 更新空的涨停原因股票
+blockingScheduler.add_job(update_null_zt_reason, 'cron', hour='16,17,18,19,20,21,22,23', minute='10')
 
 print('定时任务启动成功')
 blockingScheduler.start()

{mns_scheduler-1.1.5.9.dist-info → mns_scheduler-1.1.6.0.dist-info}/METADATA

@@ -1,4 +1,4 @@
 Metadata-Version: 2.1
 Name: mns-scheduler
-Version: 1.1.5.9
+Version: 1.1.6.0
 

{mns_scheduler-1.1.5.9.dist-info → mns_scheduler-1.1.6.0.dist-info}/RECORD

@@ -126,11 +126,12 @@ mns_scheduler/zt/script/kcx_high_chg_open_his_data_handle.py,sha256=aTrYgshcccoH
 mns_scheduler/zt/script/sync_high_chg_pool_his_data.py,sha256=dtREQdNpoDM4KLTvdXuiEhXmdjpc1dScMDgR3fbEbww,1685
 mns_scheduler/zt/script/sync_now_higt_chg_zt.py,sha256=bhoIGDWTI3w0YKVfIAVNHWpUrYqJYDMGPt-1i3d_Zmw,1850
 mns_scheduler/zt/zt_pool/__init__.py,sha256=Tyvi_iQlv3jz59EdH67Mycnt9CSixcWPQoJwu55bOq0,165
-mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py,sha256=Y2TattsWufh8rSYxN5szSU_F5dnRELEdcA8jgF-h6Zo,6603
+mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py,sha256=t2ZTNDyzEB1dgCBok_UY5du_xJ8uJjTUjEFnZWsNOlI,7070
 mns_scheduler/zt/zt_pool/ths_zt_pool_sync_api.py,sha256=Sy39T-yFwLSIIoSZqQzS-6-W1RlaFWvYpksEXKQVFdI,10456
+mns_scheduler/zt/zt_pool/update_null_zt_reason_api.py,sha256=5bHFXEUE-h6kFxtKshGV8qT7eLevG8-HDJsemUBN39E,2032
 mns_scheduler/zz_task/__init__.py,sha256=QWBdZwBCvQw8aS4hnL9_pg3U3ZiNLUXzlImyy9WhUcI,163
-mns_scheduler/zz_task/data_sync_task.py,sha256=rxW_heXhyFBrkB4SPy1SK0t7zvv5f7Csdt29yoXp-E0,20200
-mns_scheduler-1.1.5.9.dist-info/METADATA,sha256=g3wVhh90TYb-QBwFf5330JrCHFpYm_NFtCzV7HBtCv8,64
-mns_scheduler-1.1.5.9.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-mns_scheduler-1.1.5.9.dist-info/top_level.txt,sha256=PXQDFBGR1pWmsUbH5yiLAh71P5HZODTRED0zJ8CCgOc,14
-mns_scheduler-1.1.5.9.dist-info/RECORD,,
+mns_scheduler/zz_task/data_sync_task.py,sha256=3Ggntt959GPAEPmNfSsgTazqNF-bMd8VtU_ZwQwFs0c,20771
+mns_scheduler-1.1.6.0.dist-info/METADATA,sha256=vWSJRyyuLTYLJHuB6unk9A4D3KoUEAxaVVbjRNVc3U8,64
+mns_scheduler-1.1.6.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+mns_scheduler-1.1.6.0.dist-info/top_level.txt,sha256=PXQDFBGR1pWmsUbH5yiLAh71P5HZODTRED0zJ8CCgOc,14
+mns_scheduler-1.1.6.0.dist-info/RECORD,,