mns-scheduler 1.1.8.4__py3-none-any.whl → 1.4.5.7__py3-none-any.whl

mns_scheduler/trade/balance/ths_account_balance_service.py

@@ -0,0 +1,7 @@
+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)

mns_scheduler/trade/sync_position_api.py

@@ -2,23 +2,23 @@ import sys
 import os
 
 file_path = os.path.abspath(__file__)
-end = file_path.index('mns') + 16
+end = file_path.index('mns') + 17
 project_path = file_path[0:end]
 sys.path.append(project_path)
 import mns_common.component.deal.deal_service_api as deal_service_api
 import pandas as pd
 import mns_common.utils.data_frame_util as data_frame_util
 import mns_common.constant.db_name_constant as db_name_constant
-import datetime
-
+from datetime import datetime
+from mns_common.component.deal.terminal_enum import TerminalEnum
 from mns_common.db.MongodbUtil import MongodbUtil
 
 mongodb_util = MongodbUtil('27017')
 
 
 # 同步持仓
-def sync_position():
-    now_date = datetime.datetime.now()
+def sync_position_ths():
+    now_date = datetime.now()
     str_day = now_date.strftime('%Y-%m-%d')
     query_exist = {'str_day': str_day}
     if mongodb_util.exist_data_query(db_name_constant.POSITION_STOCK, query_exist):
@@ -52,5 +52,38 @@ def sync_position():
         mongodb_util.save_mongo(position_df, db_name_constant.POSITION_STOCK)
 
 
+def sync_position_qmt():
+    now_date = datetime.now()
+    str_day = now_date.strftime('%Y-%m-%d')
+    query_exist = {'str_day': str_day}
+    # if mongodb_util.exist_data_query(db_name_constant.POSITION_STOCK, query_exist):
+    #     return None
+    position_list = deal_service_api.get_position(TerminalEnum.QMT.terminal_code)
+    if len(position_list) == 0:
+        return None
+    position_df = pd.DataFrame(position_list)
+    position_df = position_df.rename(columns={
+        "stock_code": "symbol",
+        "avg_price": "cost_price",
+        "profit_loss": "floating_profit_loss",
+        "market_value": "flow_mv",
+        "can_use_volume": "available_position",
+        "frozen_volume": "frozen_position",
+    })
+
+    position_df['cost_price'] = round(position_df['cost_price'], 2)
+    position_df['open_price'] = round(position_df['open_price'], 2)
+
+    position_df['open_position'] = position_df['available_position'] + position_df['frozen_position'] + position_df[
+        'on_road_volume']
+
+    position_df['symbol'] = position_df['symbol'].str.slice(0, 6)
+
+    position_df["_id"] = position_df['symbol'] + '-' + str_day
+    position_df["str_day"] = str_day
+    position_df["valid"] = True
+    mongodb_util.save_mongo(position_df, db_name_constant.POSITION_STOCK)
+
+
 if __name__ == '__main__':
-    sync_position()
+    sync_position_qmt()

mns_scheduler/trade/task/trader_task_service.py

@@ -2,7 +2,7 @@ import sys
 import os
 
 file_path = os.path.abspath(__file__)
-end = file_path.index('mns') + 16
+end = file_path.index('mns') + 17
 project_path = file_path[0:end]
 sys.path.append(project_path)
 import mns_common.utils.cmd_util as cmd_util
@@ -11,26 +11,53 @@ import mns_scheduler.trade.auto_login.trader_auto_service as trader_auto_service
 import mns_common.utils.data_frame_util as data_frame_util
 from loguru import logger
 
-# 交易任务
-TRADER_SERVER_PATH = 'H:\\mns-trader.bat'
+# 交易服务端
+TRADER_SERVER_PATH = 'H:\\mns_trader_server.bat'
 # 实时行情同步任务 python名称
-TRADER_SERVER_NAME = "mns_trader"
+TRADER_SERVER_NAME = "mns_trader_server"
+
+
+# 打开qmt
+def open_qmt_terminal():
+    logger.info('打开 [交易web服务端]')
+    if bool(1 - is_open_trader_server()):
+        cmd_util.open_bat_file(TRADER_SERVER_PATH)
+    else:
+        logger.info('交易服务端已经运行')
+    logger.info('自动登陆qmt')
+    trader_auto_service.qmt_auto_login()
 
 
 # 打开交易客户端
 def open_trader_terminal():
-    # 打开任务进程
-    cmd_util.open_bat_file(TRADER_SERVER_PATH)
-    # 自动登陆
+    # 打开 [交易web服务端]
+    logger.info('打开 [交易web服务端]')
+    if bool(1 - is_open_trader_server()):
+        cmd_util.open_bat_file(TRADER_SERVER_PATH)
+    time.sleep(10)
+    logger.info('自动登陆ths和qmt')
+    # 自动登陆ths和qmt
     trader_auto_service.auto_login()
     time.sleep(5)
-    # 需先打开同花顺终端在开始交易服务
+    # 杀死[交易web服务端]，使用同花顺需要先打开xiadan.exe，在重新打开[交易web服务端]
+    logger.info('杀死[交易web服务端]')
     kill_server()
     time.sleep(5)
-    # 打开任务进程
+    logger.info('重新打开 [交易web服务端]')
+    # 重新打开 [交易web服务端]
     cmd_util.open_bat_file(TRADER_SERVER_PATH)
 
 
+def is_open_trader_server():
+    all_cmd_processes = cmd_util.get_all_process()
+    all_cmd_processes_trader = all_cmd_processes.loc[
+        (all_cmd_processes['total_info'].str.contains(TRADER_SERVER_NAME, case=False, na=False))]
+    if data_frame_util.is_not_empty(all_cmd_processes_trader):
+        return True
+    else:
+        return False
+
+
 def kill_server():
     all_cmd_processes = cmd_util.get_all_process()
     all_cmd_processes_trader = all_cmd_processes.loc[
@@ -45,4 +72,5 @@ def kill_server():
 
 
 if __name__ == '__main__':
-    open_trader_terminal()
+    # kill_server()
+    open_qmt_terminal()

mns_scheduler/trade/tfp/__init__.py

@@ -0,0 +1,7 @@
+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)

mns_scheduler/trade/tfp/stock_tfp_info_sync.py

@@ -0,0 +1,56 @@
+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.tfp.stock_tfp_api as stock_tfp_api
+import mns_common.constant.db_name_constant as db_name_constant
+import mns_common.utils.data_frame_util as data_frame_util
+from mns_common.db.MongodbUtil import MongodbUtil
+import mns_common.component.common_service_fun_api as common_service_fun_api
+
+mongodb_util = MongodbUtil('27017')
+
+
+def sync_stock_tfp(str_day):
+    stock_tfp_df = stock_tfp_api.get_stock_tfp_by_day(str_day)
+    if data_frame_util.is_not_empty(stock_tfp_df):
+        stock_tfp_df['sus_begin_time'] = stock_tfp_df['sus_begin_time'].fillna('无')
+        stock_tfp_df['sus_end_time'] = stock_tfp_df['sus_end_time'].fillna('无')
+        stock_tfp_df['resume_time'] = stock_tfp_df['resume_time'].fillna('无')
+        stock_tfp_df['_id'] = stock_tfp_df['symbol'] + "_" + str_day
+        # 排除name列中值包含'B'的行
+        stock_tfp_df = stock_tfp_df[~stock_tfp_df['name'].str.contains('B', na=False)]
+
+        stock_tfp_df['sus_begin_time'] = stock_tfp_df['sus_begin_time'].astype(str)
+        stock_tfp_df['sus_end_time'] = stock_tfp_df['sus_end_time'].astype(str)
+        stock_tfp_df['resume_time'] = stock_tfp_df['resume_time'].astype(str)
+        stock_tfp_df['sus_begin_date'] = stock_tfp_df['sus_begin_date'].astype(str)
+        stock_tfp_df = common_service_fun_api.exclude_st_symbol(stock_tfp_df)
+        # 根据条件设置新列'type'的值
+
+        # 初始化类型值
+        stock_tfp_df['type'] = '0'
+        stock_tfp_df.loc[stock_tfp_df['sus_reason'] == '盘中停牌', 'type'] = '0'
+        stock_tfp_df.loc[stock_tfp_df['sus_reason'] == '停牌一天', 'type'] = '1'
+        stock_tfp_df.loc[stock_tfp_df['sus_reason'] == '连续停牌', 'type'] = '2'
+        stock_tfp_df.loc[stock_tfp_df['sus_reason'] == '刊登重要公告', 'type'] = '3'
+        stock_tfp_df.loc[stock_tfp_df['sus_reason'] == '拟筹划重大资产重组', 'type'] = '4'
+        stock_tfp_df['valid'] = True
+        stock_tfp_df['str_day'] = str_day
+        stock_tfp_df.loc[stock_tfp_df['type'] == '2', 'valid'] = False
+        stock_tfp_df.loc[stock_tfp_df['type'] == '1', 'valid'] = False
+
+        stock_tfp_df = stock_tfp_df.fillna(0)
+
+        mongodb_util.save_mongo(stock_tfp_df, db_name_constant.STOCK_TFP_INFO)
+
+    return stock_tfp_df
+
+
+if __name__ == '__main__':
+    df = sync_stock_tfp('2025-12-25')
+    print(df)

mns_scheduler/zb/stock_zb_pool_sync.py

@@ -27,19 +27,4 @@ def sync_stock_zb_pool(str_now_day):
 
 
 if __name__ == '__main__':
-    sync_stock_zb_pool('2024-09-05')
-    sync_date = date_handle_util.add_date_day('20231214', 1)
-
-    now_date = datetime.now()
-
-    str_now_day_01 = sync_date.strftime('%Y-%m-%d')
-
-    while now_date > sync_date:
-        stock_zb_pool_df = sync_stock_zb_pool(str_now_day_01)
-        if stock_zb_pool_df is None:
-            sync_date = date_handle_util.add_date_day(date_handle_util.no_slash_date(str_now_day_01), 1)
-            str_now_day_01 = sync_date.strftime('%Y-%m-%d')
-            continue
-        mongodb_util.save_mongo(stock_zb_pool_df, 'stock_zb_pool')
-        sync_date = date_handle_util.add_date_day(date_handle_util.no_slash_date(str_now_day_01), 1)
-        str_now_day_01 = sync_date.strftime('%Y-%m-%d')
+    sync_stock_zb_pool('2025-12-11')

mns_scheduler/zt/high_chg/sync_high_chg_pool_service.py

@@ -9,7 +9,7 @@ import pandas as pd
 from loguru import logger
 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.em.east_money_stock_api as east_money_stock_api
+import mns_common.component.em.em_stock_info_api as em_stock_info_api
 import mns_common.utils.date_handle_util as date_handle_util
 import mns_common.component.company.company_common_service_api as company_common_service_api
 from mns_common.db.MongodbUtil import MongodbUtil
@@ -138,7 +138,7 @@ def sync_stock_high_chg_pool_list(str_day, symbol_list):
     now_date = datetime.now()
     now_day = now_date.strftime('%Y-%m-%d')
 
-    real_time_quotes_now_init = east_money_stock_api.get_real_time_quotes_all_stocks()
+    real_time_quotes_now_init = em_stock_info_api.get_a_stock_info()
 
     last_trade_zt_pool = zt_common_service_api.get_last_trade_day_zt(str_day)
 

mns_scheduler/zt/high_chg/sync_high_chg_real_time_quotes_service.py

@@ -142,4 +142,4 @@ def save_realtime_quotes_now_zt_data(realtime_quotes_now_zt, str_day, symbol):
 
 
 if __name__ == '__main__':
-    sync_high_chg_real_time_quotes('2024-06-24')
+    sync_high_chg_real_time_quotes('2024-12-26')

mns_scheduler/zt/script/__init__.py

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

mns_scheduler/zt/script/fix_error_deal_day.py

@@ -0,0 +1,41 @@
+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)
+from mns_common.db.MongodbUtil import MongodbUtil
+from loguru import logger
+import mns_common.utils.date_handle_util as date_handle_util
+mongodb_util = MongodbUtil('27017')
+
+from mns_scheduler.db.script.sync.remote_mongo_util import RemoteMongodbUtil
+
+
+remote_mongodb_util = RemoteMongodbUtil('27017')
+
+
+def fix_error_deal_days(db_name):
+    realtime_quotes_now_zt_new_kc_open_df = mongodb_util.find_query_data(db_name, {})
+    realtime_quotes_now_zt_new_kc_open_df['id_key'] = realtime_quotes_now_zt_new_kc_open_df['_id']
+    for stock_one in realtime_quotes_now_zt_new_kc_open_df.itertuples():
+        try:
+            symbol = stock_one.symbol
+            str_day = stock_one.str_day
+            query = {'symbol': symbol, 'date': {"$lt": date_handle_util.no_slash_date(str_day)}}
+            deal_days = mongodb_util.count(query, 'stock_qfq_daily')
+            new_values = {"$set": {"deal_days": deal_days}}
+            id_key = stock_one.id_key
+            update_query = {'_id': id_key}
+            mongodb_util.update_many(update_query, new_values, db_name)
+            logger.info("更新到:{},{}", symbol, str_day)
+        except BaseException as e:
+            logger.error("出现异常:{},{},{}", symbol, str_day, e)
+
+
+if __name__ == '__main__':
+    # db_name_0 = 'realtime_quotes_now_zt_new_kc_open'
+    db_name_01 = 'stock_high_chg_pool'
+    # fix_error_deal_days(db_name_0)
+    fix_error_deal_days(db_name_01)

mns_scheduler/zt/script/kcx_high_chg_open_his_data_handle.py

@@ -2,7 +2,7 @@ import sys
 import os
 
 file_path = os.path.abspath(__file__)
-end = file_path.index('mns') + 16
+end = file_path.index('mns') + 17
 project_path = file_path[0:end]
 sys.path.append(project_path)
 import mns_scheduler.zt.open_data.kcx_high_chg_open_data_sync as kcx_high_chg_open_data_sync
@@ -83,7 +83,7 @@ def fix_miss_data(str_end):
 
 
 if __name__ == '__main__':
-    kcx_high_chg_open_data_sync.sync_all_kc_zt_data('2024-06-25', None)
+    kcx_high_chg_open_data_sync.sync_all_kc_zt_data('2025-06-30', None)
     # sync_all_kc_zt_data('2023-08-16')
     # sync_all_kc_zt_data('2023-07-07')
     # realtime_quotes_now_zt_new_kc_open_sync()

mns_scheduler/zt/script/sync_high_chg_pool_his_data.py

@@ -2,7 +2,7 @@ import sys
 import os
 
 file_path = os.path.abspath(__file__)
-end = file_path.index('mns') + 16
+end = file_path.index('mns') + 17
 project_path = file_path[0:end]
 sys.path.append(project_path)
 import mns_common.utils.date_handle_util as date_handle_util
@@ -36,4 +36,4 @@ def sync_his_zt_pool_data(begin_day, end_day):
 
 
 if __name__ == '__main__':
-    sync_high_chg_pool_service.sync_stock_high_chg_pool_list('2024-07-22', None)
+    sync_high_chg_pool_service.sync_stock_high_chg_pool_list('2025-05-23', None)

mns_scheduler/zt/script/sync_now_higt_chg_zt.py

@@ -2,12 +2,12 @@ import sys
 import os
 
 file_path = os.path.abspath(__file__)
-end = file_path.index('mns') + 16
+end = file_path.index('mns') + 17
 project_path = file_path[0:end]
 sys.path.append(project_path)
-import mns_common.api.em.east_money_stock_api as east_money_stock_api
+import mns_common.component.em.em_stock_info_api as em_stock_info_api
 import mns_common.component.common_service_fun_api as common_service_fun_api
-import mns_scheduler.k_line.sync.daily_week_month_line_sync as daily_week_month_line_sync
+import mns_scheduler.k_line.month_week_daily.daily_week_month_line_sync as daily_week_month_line_sync
 from datetime import datetime
 import mns_scheduler.zt.zt_pool.em_zt_pool_sync_api as em_zt_pool_sync_api
 import mns_scheduler.zt.high_chg.sync_high_chg_pool_service as sync_high_chg_pool_service
@@ -16,16 +16,17 @@ import mns_scheduler.zt.high_chg.sync_high_chg_real_time_quotes_service as sync_
 
 
 def sync_now_day_high_chg():
-    real_time_quotes_now_es = east_money_stock_api.get_real_time_quotes_all_stocks()
+    real_time_quotes_now_es = em_stock_info_api.get_a_stock_info()
     real_time_quotes_now_es_high_chg = real_time_quotes_now_es.loc[
         real_time_quotes_now_es['chg'] > common_service_fun_api.ZT_CHG]
+    now_date = datetime.now()
+
+    str_now_day = now_date.strftime('%Y-%m-%d')
     # 同步qfq k线
-    daily_week_month_line_sync.sync_all_daily_data('daily', 'qfq', 'stock_qfq_daily', None,
+    daily_week_month_line_sync.sync_all_daily_data('daily', 'qfq', 'stock_qfq_daily', str_now_day,
                                                    list(real_time_quotes_now_es_high_chg['symbol']))
 
-    now_date = datetime.now()
 
-    str_now_day = now_date.strftime('%Y-%m-%d')
     # 同步当前涨停
     em_zt_pool_sync_api.save_zt_info(str_now_day)