mns-scheduler 1.1.5.9__tar.gz → 1.1.6.0__tar.gz

{mns_scheduler-1.1.5.9 → mns_scheduler-1.1.6.0}/PKG-INFO

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

{mns_scheduler-1.1.5.9 → mns_scheduler-1.1.6.0}/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-1.1.6.0/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-1.1.5.9 → mns_scheduler-1.1.6.0}/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 → mns_scheduler-1.1.6.0}/mns_scheduler.egg-info/PKG-INFO

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

{mns_scheduler-1.1.5.9 → mns_scheduler-1.1.6.0}/mns_scheduler.egg-info/SOURCES.txt

@@ -134,5 +134,6 @@ mns_scheduler/zt/script/sync_now_higt_chg_zt.py
 mns_scheduler/zt/zt_pool/__init__.py
 mns_scheduler/zt/zt_pool/em_zt_pool_sync_api.py
 mns_scheduler/zt/zt_pool/ths_zt_pool_sync_api.py
+mns_scheduler/zt/zt_pool/update_null_zt_reason_api.py
 mns_scheduler/zz_task/__init__.py
 mns_scheduler/zz_task/data_sync_task.py

{mns_scheduler-1.1.5.9 → mns_scheduler-1.1.6.0}/setup.py

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