mns-common 1.4.1.8__py3-none-any.whl → 1.5.7.2__py3-none-any.whl

mns_common/component/deal/deal_service_api.py

@@ -7,6 +7,7 @@ project_path = file_path[0:end]
 sys.path.append(project_path)
 import json
 import requests
+from loguru import logger
 
 '''
 
@@ -89,7 +90,12 @@ def get_position(terminal):
     if response.status_code != 200:
         result = {"message": '获取持仓失败'}
     else:
-        result = response.json()
+        try:
+            result = response.json()
+        except BaseException as e:
+            result_test = response.text
+            result = json.loads(result_test)
+
     return result
 
 
@@ -126,7 +132,24 @@ def auto_login(terminal):
     if response.status_code != 200:
         result = {"message": '获取持仓失败'}
     else:
+
+        result = response.json()
+
+    return result
+
+
+# 查询订单
+def query_orders(terminal):
+    param_order = {
+        'terminal': terminal}
+    param_json = json.dumps(param_order)
+    response = request_trader('/order', param_json)
+    if response.status_code != 200:
+        result = {"message": '查询订单失败'}
+    else:
+
         result = response.json()
+
     return result
 
 
@@ -138,18 +161,57 @@ def request_trader(url, param):
     return requests.post(total_url, data=param, headers=headers)
 
 
+# 获取交易价格
+def get_trade_price(terminal, symbol, price_code, limit_chg):
+    param_position = {
+        'symbol': symbol,
+        'terminal': terminal,
+        'price_code': price_code,
+        'limit_chg': limit_chg}
+
+    param_json = json.dumps(param_position)
+    response = request_trader('/trade/price', param_json)
+    if response.status_code != 200:
+        result = {"message": '获取行情失败'}
+    else:
+        result = response.json()
+    return result
+
+
+# 获取qmt 行情
+def get_qmt_real_time_quotes_detail(terminal, symbol_list):
+    param_position = {
+        'symbol_list': symbol_list,
+        'terminal': terminal}
+
+    param_json = json.dumps(param_position)
+    response = request_trader('/qmt/real/time/quotes/detail', param_json)
+    if response.status_code != 200:
+        result = {"message": '获取行情失败'}
+    else:
+        result = response.json()
+    return result
+
+
+from mns_common.component.deal.terminal_enum import TerminalEnum
+
 if __name__ == '__main__':
-    auto_login('qmt')
-    get_position('qmt')
+    terminal_test = TerminalEnum.QMT.terminal_code
+    get_position(terminal_test)
+    symbol_one_test = ['301181.SZ']
+    result_json = get_qmt_real_time_quotes_detail(terminal_test, symbol_one_test)
+    print(result_json)
+    # auto_login('qmt')
+    # get_position('qmt')
     # terminal_test = 'easy_trader'
     # order_cancel('251145121', terminal_test)
     # get_position(terminal_test)
     # auto_ipo_buy(terminal_test)
-    trade_buy(
-        '301314.SZ',
-        35.77,
-        1000000,
-        'qmt')
+    # trade_buy(
+    #     '301314.SZ',
+    #     35.77,
+    #     1000000,
+    #     'qmt')
     # trade_sell(
     #     '301314',
     #     35.77,

mns_common/component/deal/deal_service_v2_api.py

@@ -0,0 +1,167 @@
+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)
+import json
+import requests
+
+'''
+
+'''
+
+
+def trade_buy(symbol, buy_price, buy_volume, terminal):
+    '''
+    买入
+    :param symbol:
+    :param buy_price:
+    :param buy_volume:
+    :param terminal:
+    :return:
+    '''
+    param_buy = {
+        'symbol': symbol,
+        'buy_price': buy_price,
+        'buy_volume': buy_volume,
+        'terminal': terminal}
+    param_json = json.dumps(param_buy)
+    response = request_trader_post('/buy', param_json)
+    if response.status_code != 200:
+        buy_result = {"message": '买入失败'}
+    else:
+        buy_result = response.json()
+    return buy_result
+
+
+def trade_sell(symbol, sell_price, sell_volume, terminal):
+    '''
+    卖出
+    :param symbol:
+    :param sell_price:
+    :param sell_volume:
+    :param terminal:
+    :return:
+    '''
+    param_sell = {
+        'symbol': symbol,
+        'sell_price': sell_price,
+        'sell_volume': sell_volume,
+        'terminal': terminal}
+    param_json = json.dumps(param_sell)
+    response = request_trader_post('/sell', param_json)
+    if response.status_code != 200:
+        sell_result = {"message": '卖出失败'}
+    else:
+        sell_result = response.json()
+    return sell_result
+
+
+def auto_ipo_buy(terminal):
+    '''
+    自动打新
+    :param terminal:
+    :return:
+    '''
+    param_auto_ipo = {
+        'terminal': terminal}
+    param_json = json.dumps(param_auto_ipo)
+    response = request_trader_get('/auto/ipo/buy', param_json)
+    if response.status_code != 200:
+        result = {"message": '自动打新失败'}
+    else:
+        result = response.json()
+    return result
+
+
+def get_position(terminal):
+    '''
+    获取持仓
+    :param terminal:
+    :return:
+    '''
+    param_position = {
+        'terminal': terminal}
+    param_json = json.dumps(param_position)
+    response = request_trader_get('/position', param_json)
+    if response.status_code != 200:
+        result = {"message": '获取持仓失败'}
+    else:
+        result = response.json()
+    return result
+
+
+def order_cancel(entrust_no, terminal):
+    '''
+    撤单
+       :param entrust_no:
+    :param terminal:
+    :return:
+    '''
+    param_cancel = {
+        "entrust_no": entrust_no,
+        'terminal': terminal}
+    param_json = json.dumps(param_cancel)
+    response = request_trader_post('/cancel', param_json)
+    if response.status_code != 200:
+        result = {"message": '撤单失败'}
+    else:
+        result = response.json()
+    return result
+
+
+# 自动登陆接口
+def auto_login(terminal):
+    '''
+      自动登陆客户端
+      :param terminal:
+      :return:
+      '''
+    param_position = {
+        'terminal': terminal}
+    param_json = json.dumps(param_position)
+    response = request_trader_post('/auto/login', param_json)
+    if response.status_code != 200:
+        result = {"message": '获取持仓失败'}
+    else:
+        result = response.json()
+    return result
+
+
+def request_trader_post(url, param):
+    total_url = "http://127.0.0.1:5002/api/trade" + url
+    headers = {
+        "Content-Type": "application/json"
+    }
+    return requests.post(total_url, data=param, headers=headers)
+
+
+def request_trader_get(url, param):
+    total_url = "http://127.0.0.1:5002/api/trade" + url
+    headers = {
+        "Content-Type": "application/json"
+    }
+    return requests.get(total_url, data=param, headers=headers)
+
+
+if __name__ == '__main__':
+    # auto_login('qmt')
+    # get_position('qmt')
+    # terminal_test = 'easy_trader'
+    # order_cancel('251145121', terminal_test)
+    # get_position(terminal_test)
+    # auto_ipo_buy(terminal_test)
+    buy_result_test = trade_buy(
+        '688693.SH',
+        36.39,
+        1000,
+        'ths')
+    entrust_no_test = buy_result_test['entrust_no']
+    order_cancel(entrust_no_test, 'ths')
+    # trade_sell(
+    #     '301314',
+    #     35.77,
+    #     100,
+    #     'easy_trader')

mns_common/component/em/em_stock_info_api.py

@@ -13,7 +13,9 @@ mongodb_util = MongodbUtil('27017')
 
 # 获取东方财富A股全部信息
 def get_a_stock_info():
-    return mongodb_util.find_all_data(extra_income_db_name.EM_A_STOCK_INFO)
+    em_a_stock_info = mongodb_util.find_all_data(extra_income_db_name.EM_A_STOCK_INFO)
+    em_a_stock_info = em_a_stock_info[~em_a_stock_info['symbol'].str.startswith(('8', '4'))]
+    return em_a_stock_info
 
 
 # 获取东方财富ETF全部信息
@@ -28,15 +30,19 @@ def get_kzz_info():
 
 # 获取东方财富美股全部信息
 def get_us_stock_info():
-    return mongodb_util.find_all_data(extra_income_db_name.EM_US_STOCK_INFO)
+    return mongodb_util.find_all_data(extra_income_db_name.US_STOCK_INFO_EM)
 
 
 # 获取东方财富美股 eft全部信息
 def get_us_etf_info():
-    return mongodb_util.find_all_data(extra_income_db_name.EM_US_ETF_INFO)
+    return mongodb_util.find_all_data(extra_income_db_name.US_ETF_INFO_EM)
 
 
 # 获取东方财富港股全部信息
 
 def get_hk_stock_info():
     return mongodb_util.find_all_data(extra_income_db_name.EM_HK_STOCK_INFO)
+
+
+if __name__ == '__main__':
+    get_a_stock_info()

mns_common/component/main_line/main_line_zt_reason_service.py

@@ -0,0 +1,237 @@
+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)
+import mns_common.constant.db_name_constant as db_name_constant
+from mns_common.db.MongodbUtil import MongodbUtil
+import mns_common.api.ths.zt.ths_stock_zt_pool_api as ths_stock_zt_pool_api
+from mns_common.utils.async_fun import async_fun
+import mns_common.component.trade_date.trade_date_common_service_api as trade_date_common_service_api
+
+mongodb_util = MongodbUtil('27017')
+import mns_common.utils.data_frame_util as data_frame_util
+import time
+from loguru import logger
+import pandas as pd
+from datetime import datetime
+
+
+# 添加主线和涨停分析临时数据
+def merge_main_line_info(str_day, data_df):
+    # 保证数据完整性
+    if 'main_line' not in data_df.columns:
+        data_df['main_line'] = ''
+    else:
+        data_df.fillna({'main_line': ''}, inplace=True)
+
+    if 'sub_main_line' not in data_df.columns:
+        data_df['sub_main_line'] = ''
+    else:
+        data_df.fillna({'sub_main_line': ''}, inplace=True)
+
+    if 'zt_analysis' not in data_df.columns:
+        data_df['zt_analysis'] = ''
+
+    else:
+        data_df.fillna({'zt_analysis': ''}, inplace=True)
+
+    if 'zt_reason' not in data_df.columns:
+        data_df['zt_reason'] = ''
+    else:
+        data_df.fillna({'zt_reason': ''}, inplace=True)
+
+    if 'main_line_choose_source' not in data_df.columns:
+        data_df['main_line_choose_source'] = 'now_zt'
+    else:
+        data_df.fillna({'main_line_choose_source': 'now_zt'}, inplace=True)
+
+    if 'main_line_grade' not in data_df.columns:
+        data_df['main_line_grade'] = 1
+    else:
+        data_df.fillna({'main_line_grade': 1}, inplace=True)
+
+    query_zt_now = {'symbol': {"$in": list(data_df['symbol'])}, 'str_day': str_day}
+    # merge 主线 涨停详情
+    main_line_detail_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL, query_zt_now)
+    if data_frame_util.is_not_empty(main_line_detail_df):
+        symbol_mapping_zt_reason_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['zt_reason']))
+
+        symbol_mapping_zt_analysis_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['zt_analysis']))
+
+        symbol_mapping_main_line_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['main_line']))
+
+        symbol_mapping_sub_main_line_now = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['sub_main_line']))
+
+        symbol_mapping_main_line_choose_source = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['main_line_choose_source']))
+
+        symbol_mapping_main_line_grade = dict(
+            zip(main_line_detail_df['symbol'], main_line_detail_df['main_line_grade']))
+
+        data_df['main_line_grade'] = data_df['symbol'].map(
+            symbol_mapping_main_line_grade).fillna(
+            data_df['main_line_grade'])
+
+        data_df['main_line'] = data_df['symbol'].map(
+            symbol_mapping_main_line_now).fillna(
+            data_df['main_line'])
+
+        data_df['sub_main_line'] = data_df['symbol'].map(
+            symbol_mapping_sub_main_line_now).fillna(
+            data_df['sub_main_line'])
+        data_df['zt_reason'] = data_df['symbol'].map(
+            symbol_mapping_zt_reason_now).fillna(
+            data_df['zt_reason'])
+        data_df['zt_analysis'] = data_df['symbol'].map(
+            symbol_mapping_zt_analysis_now).fillna(
+            data_df['zt_analysis'])
+
+        data_df['main_line_choose_source'] = data_df['symbol'].map(
+            symbol_mapping_main_line_choose_source).fillna(
+            data_df['main_line_choose_source'])
+
+    return data_df
+
+
+# merge涨停分析 原因
+def merge_zt_reason_info(str_day, data_df):
+    if 'zt_analysis' not in data_df.columns:
+        data_df['zt_analysis'] = ''
+
+    else:
+        data_df.fillna({'zt_analysis': ''}, inplace=True)
+
+    if 'zt_reason' not in data_df.columns:
+        data_df['zt_reason'] = ''
+    else:
+        data_df.fillna({'zt_reason': ''}, inplace=True)
+
+    query_zt_now = {'symbol': {"$in": list(data_df['symbol'])}, 'str_day': str_day}
+    # merge 主线 涨停详情
+    zt_reason_analysis_df = mongodb_util.find_query_data(db_name_constant.ZT_REASON_ANALYSIS, query_zt_now)
+    if data_frame_util.is_not_empty(zt_reason_analysis_df):
+        symbol_mapping_zt_reason_now = dict(
+            zip(zt_reason_analysis_df['symbol'], zt_reason_analysis_df['zt_reason']))
+
+        symbol_mapping_zt_analysis_now = dict(
+            zip(zt_reason_analysis_df['symbol'], zt_reason_analysis_df['zt_analysis']))
+
+        data_df['zt_reason'] = data_df['symbol'].map(
+            symbol_mapping_zt_reason_now).fillna(
+            data_df['zt_reason'])
+        data_df['zt_analysis'] = data_df['symbol'].map(
+            symbol_mapping_zt_analysis_now).fillna(
+            data_df['zt_analysis'])
+
+    return data_df
+
+
+def update_zt_reason_analysis(symbol, str_day, name, need_update):
+    try:
+        key_id = symbol + "_" + str_day
+        query_zt = {"_id": key_id}
+
+        # 已经存在的数据
+        zt_reason_analysis_exists_df = mongodb_util.find_query_data(db_name_constant.ZT_REASON_ANALYSIS,
+                                                                    query_zt)
+
+        if data_frame_util.is_not_empty(zt_reason_analysis_exists_df):
+            zt_analysis = list(zt_reason_analysis_exists_df['zt_analysis'])[0]
+            zt_reason = list(zt_reason_analysis_exists_df['zt_reason'])[0]
+            # 需要更新数据
+            if data_frame_util.is_string_empty(zt_analysis) or data_frame_util.is_string_empty(
+                    zt_reason) or need_update:
+                try:
+                    zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(symbol)
+                    zt_analysis = zt_result_dict['zt_analyse_detail']
+                    zt_reason = zt_result_dict['zt_reason']
+                    time.sleep(1)
+                except BaseException as e:
+                    time.sleep(1)
+                    zt_analysis = ''
+                    zt_reason = ''
+
+                zt_reason_analysis_exists_df['zt_analysis'] = zt_analysis
+                zt_reason_analysis_exists_df['zt_reason'] = zt_reason
+                now_date = datetime.now()
+                str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+                zt_reason_analysis_exists_df['str_now_date'] = str_now_date
+                mongodb_util.save_mongo(zt_reason_analysis_exists_df, db_name_constant.ZT_REASON_ANALYSIS)
+        else:
+            # 不存在临时主线数据
+            try:
+                zt_result_dict = ths_stock_zt_pool_api.zt_analyse_reason(symbol)
+                zt_analysis = zt_result_dict['zt_analyse_detail']
+                zt_reason = zt_result_dict['zt_reason']
+                time.sleep(1)
+            except BaseException as e:
+                time.sleep(1)
+                zt_analysis = ''
+                zt_reason = ''
+            now_date = datetime.now()
+            str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+            reason_dict = {'_id': key_id,
+                           'symbol': symbol,
+                           'name': name,
+                           'zt_analysis': zt_analysis,
+                           'zt_reason': zt_reason,
+                           'str_day': str_day,
+                           'update_time': str_now_date,
+                           }
+            reason_df = pd.DataFrame(reason_dict, index=[1])
+            mongodb_util.save_mongo(reason_df, db_name_constant.ZT_REASON_ANALYSIS)
+    except BaseException as e:
+        logger.error("添加涨停原因详情异常:{},{}", e, name)
+
+
+@async_fun
+def update_symbol_list_zt_reason_analysis(data_df, need_update):
+    for stock_one in data_df.itertuples():
+        str_day = stock_one.str_day
+        symbol = stock_one.symbol
+        name = stock_one.name
+        update_zt_reason_analysis(symbol, str_day, name, need_update)
+        time.sleep(1)
+
+
+# 保存连板股票主线
+@async_fun
+def save_last_trade_day_main_line(str_day, stock_em_zt_pool_df_data):
+    last_trade_day = trade_date_common_service_api.get_last_trade_day(str_day)
+    stock_em_zt_pool_connected_df = stock_em_zt_pool_df_data.loc[
+        stock_em_zt_pool_df_data['connected_boards_numbers'] > 1]
+    if data_frame_util.is_empty(stock_em_zt_pool_connected_df):
+        return
+    else:
+        query = {'str_day': last_trade_day, 'symbol': {"$in": list(stock_em_zt_pool_connected_df['symbol'])}}
+        last_trade_day_main_line_detail_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL, query)
+        if data_frame_util.is_empty(last_trade_day_main_line_detail_df):
+            return
+        else:
+            last_trade_day_main_line_detail_df['_id'] = last_trade_day_main_line_detail_df['symbol'] + '_' + str_day
+            last_trade_day_main_line_detail_df['str_day'] = str_day
+            last_trade_day_main_line_detail_df['connected_boards_numbers'] = last_trade_day_main_line_detail_df[
+                                                                                 'connected_boards_numbers'] + 1
+            now_date = datetime.now()
+            str_now_date = now_date.strftime('%Y-%m-%d %H:%M:%S')
+            last_trade_day_main_line_detail_df['update_time'] = str_now_date
+            today_exist_main_line_df = mongodb_util.find_query_data(db_name_constant.MAIN_LINE_DETAIL,
+                                                                    {'str_day': str_day, 'symbol': {"$in": list(
+                                                                        last_trade_day_main_line_detail_df['symbol'])}})
+            if data_frame_util.is_not_empty(today_exist_main_line_df):
+                today_new_main_line_df = last_trade_day_main_line_detail_df.loc[~
+                last_trade_day_main_line_detail_df['symbol'].isin(list(today_exist_main_line_df['symbol']))]
+            else:
+                today_new_main_line_df = last_trade_day_main_line_detail_df.copy()
+            mongodb_util.save_mongo(today_new_main_line_df, db_name_constant.MAIN_LINE_DETAIL)
+
+
+if __name__ == '__main__':
+    update_zt_reason_analysis('600362', '2025-12-26', '江西铜业', True)