mns-scheduler 1.0.8.7__py3-none-any.whl → 1.4.3.2__py3-none-any.whl

mns_scheduler/__init__.py

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

mns_scheduler/company_info/announce/company_announce_sync_service.py

@@ -0,0 +1,65 @@
+import sys
+import os
+
+file_path = os.path.abspath(__file__)
+end = file_path.index('mns') + 17
+project_path = file_path[0:end]
+sys.path.append(project_path)
+import mns_common.component.company.company_common_service_new_api as company_common_service_new_api
+import mns_common.api.ths.company.ths_company_announce_api as ths_company_announce_api
+from loguru import logger
+from mns_common.db.MongodbUtil import MongodbUtil
+import mns_common.constant.db_name_constant as db_name_constant
+
+mongodb_util = MongodbUtil('27017')
+
+
+# 同步最新公告
+# eq-f1001 业绩预告 eq-f1002 重大事项   eq-f1003 股份变动公告
+
+def sync_company_announce(symbol_list):
+    page_size = 100
+    announce_type_list = ['all', 'eq-f1003', 'eq-f1001', 'eq-f1002']
+    for announce_type_one in announce_type_list:
+        try:
+            get_company_announce(announce_type_one, page_size, symbol_list)
+        except BaseException as e:
+            logger.error("更新公告出现异常:{}", e)
+    logger.info("同步到公告信息完成")
+
+
+def get_company_announce(announce_type, page_size, symbol_list):
+    company_all_df = company_common_service_new_api.get_company_all_info_info()
+    de_list_company = company_common_service_new_api.get_de_list_company()
+    company_all_df = company_all_df.loc[~(company_all_df['symbol'].isin(de_list_company))]
+
+    if symbol_list is not None:
+        company_all_df = company_all_df.loc[(company_all_df['symbol'].isin(symbol_list))]
+
+    for stock_one in company_all_df.itertuples():
+        try:
+            symbol = stock_one.symbol
+            market_id = stock_one.market_id
+            # 公告应该不多 只更新一页的数据 页码设置100已经是最大
+            page_number = 1
+            try:
+                ths_company_announce_result = ths_company_announce_api.get_company_announce_info(symbol, market_id,
+                                                                                                 announce_type,
+                                                                                                 page_size,
+                                                                                                 page_number)
+            except BaseException as e:
+                logger.error("更新公告出现异常:{}.{}", e, symbol)
+                continue
+            ths_company_announce_result['type'] = announce_type
+            ths_company_announce_result['symbol'] = symbol
+            ths_company_announce_result['_id'] = ths_company_announce_result['guid'] + '_' + \
+                                                 ths_company_announce_result['seq'] + "_" + announce_type
+            mongodb_util.save_mongo(ths_company_announce_result, db_name_constant.COMPANY_ANNOUNCE_INFO)
+            logger.info("更新公告完成:{},{}", symbol, stock_one.name)
+
+        except BaseException as e:
+            logger.error("更新公告出现异常:{}", e)
+
+
+if __name__ == '__main__':
+    sync_company_announce(None)