mns-common 1.3.9.2__py3-none-any.whl → 1.6.1.4__py3-none-any.whl

mns_common/component/cookie/cookie_info_service.py

@@ -6,28 +6,38 @@ end = file_path.index('mns') + 16
 project_path = file_path[0:end]
 sys.path.append(project_path)
 from mns_common.db.MongodbUtil import MongodbUtil
+from mns_common.component.cookie.cookie_enum import CookieEnum
 
 mongodb_util = MongodbUtil('27017')
 
-from functools import lru_cache
 
-
-@lru_cache(maxsize=None)
+# ths cookie
 def get_ths_cookie():
-    stock_account_info = mongodb_util.find_query_data('stock_account_info', {"type": "ths_cookie", })
+    stock_account_info = mongodb_util.find_query_data('stock_account_info',
+                                                      {"type": CookieEnum.THS_COOKIE.cookie_code})
     ths_cookie = list(stock_account_info['cookie'])[0]
     return ths_cookie
 
 
-@lru_cache(maxsize=None)
+# 东财 cookie
 def get_em_cookie():
-    stock_account_info = mongodb_util.find_query_data('stock_account_info', {"type": "em_cookie", })
+    stock_account_info = mongodb_util.find_query_data('stock_account_info',
+                                                      {"type": CookieEnum.EM_COOKIE.cookie_code})
     em_cookie = list(stock_account_info['cookie'])[0]
     return em_cookie
 
 
-@lru_cache(maxsize=None)
+# 雪球 cookie
 def get_xue_qiu_cookie():
-    stock_account_info = mongodb_util.find_query_data('stock_account_info', {"type": "xue_qiu", })
+    stock_account_info = mongodb_util.find_query_data('stock_account_info',
+                                                      {"type": CookieEnum.XUE_QIU_COOKIE.cookie_code })
+    cookie = list(stock_account_info['cookie'])[0]
+    return cookie
+
+
+# 开盘啦 token
+def get_kpl_cookie():
+    stock_account_info = mongodb_util.find_query_data('stock_account_info',
+                                                      {"type": CookieEnum.KPL_COOKIE.cookie_code, })
     cookie = list(stock_account_info['cookie'])[0]
     return cookie

mns_common/component/data/data_init_api.py

@@ -60,16 +60,21 @@ def calculate_parameter_factor(real_time_quotes_now):
     real_time_quotes_now['large_inflow_multiple'] = round(
         (real_time_quotes_now['large_order_net_inflow'] / real_time_quotes_now['reference_main_inflow']), 2)
 
-    real_time_quotes_now.loc[:, 'real_disk_diff_amount_exchange'] = round(
-        (real_time_quotes_now['disk_diff_amount_exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+    if 'real_disk_diff_amount_exchange' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_disk_diff_amount_exchange'] = round(
+            (real_time_quotes_now['disk_diff_amount_exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
 
-    real_time_quotes_now.loc[:, 'real_main_inflow_multiple'] = round(
-        (real_time_quotes_now['main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+    if 'real_main_inflow_multiple' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_main_inflow_multiple'] = round(
+            (real_time_quotes_now['main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
 
-    real_time_quotes_now.loc[:, 'real_super_main_inflow_multiple'] = round(
-        (real_time_quotes_now['super_main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
-    real_time_quotes_now.loc[:, 'real_exchange'] = round(
-        (real_time_quotes_now['exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+    if 'real_super_main_inflow_multiple' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_super_main_inflow_multiple'] = round(
+            (real_time_quotes_now['super_main_inflow_multiple'] / real_time_quotes_now['mv_circulation_ratio']), 2)
+
+    if 'real_exchange' not in real_time_quotes_now.columns:
+        real_time_quotes_now.loc[:, 'real_exchange'] = round(
+            (real_time_quotes_now['exchange'] / real_time_quotes_now['mv_circulation_ratio']), 2)
 
     real_time_quotes_now.loc[:, 'max_real_main_inflow_multiple'] = real_time_quotes_now[
         ['real_main_inflow_multiple', 'real_super_main_inflow_multiple']].max(axis=1)

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,12 @@ 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['list_date'].fillna(19890604, inplace=True)
+    em_a_stock_info['list_date'] = em_a_stock_info['list_date'].replace(99990909, 19890604)
+    em_a_stock_info['list_date'] = em_a_stock_info['list_date'].replace(0, 19890604)
+    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 +33,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/exception/ExceptionMonitor.py

@@ -0,0 +1,86 @@
+import threading
+from functools import wraps
+from typing import Optional, Callable
+
+class ExceptionMonitor:
+    """支持动态配置的异常监控器"""
+    def __init__(
+        self,
+        threshold: int = 1000,
+        alert_handler: Optional[Callable[[int, int], None]] = None,
+        auto_reset: bool = True
+    ):
+        """
+        :param threshold: 报警阈值，默认1000次
+        :param alert_handler: 自定义报警函数，格式 func(current_count, threshold)
+        :param auto_reset: 触发报警后是否自动重置计数器
+        """
+        self.counter = 0
+        self._threshold = threshold
+        self.alert_handler = alert_handler
+        self.auto_reset = auto_reset
+        self.lock = threading.Lock()
+
+    @property
+    def threshold(self) -> int:
+        """当前报警阈值"""
+        return self._threshold
+
+    @threshold.setter
+    def threshold(self, value: int):
+        """动态设置报警阈值"""
+        with self.lock:
+            self._threshold = value
+
+    def increment(self):
+        """线程安全的计数器递增"""
+        with self.lock:
+            self.counter += 1
+            if self.counter >= self.threshold:
+                self.trigger_alert()
+                if self.auto_reset:
+                    self.reset_counter()
+
+    def reset_counter(self):
+        """手动重置计数器"""
+        with self.lock:
+            self.counter = 0
+
+    def trigger_alert(self):
+        """触发报警（支持自定义处理逻辑）"""
+        if self.alert_handler:
+            self.alert_handler(self.counter, self.threshold)
+        else:
+            print(f"[ALERT] Exception count {self.counter} exceeded threshold {self.threshold}")
+
+def exception_counter(
+    threshold: int = 1000,
+    alert_handler: Optional[Callable[[int, int], None]] = None,
+    auto_reset: bool = True
+):
+    """
+    异常计数装饰器工厂
+    :param threshold: 报警阈值
+    :param alert_handler: 自定义报警处理函数
+    :param auto_reset: 是否自动重置计数器
+    """
+    def decorator(func):
+        # 为每个被装饰函数创建独立监控实例
+        monitor = ExceptionMonitor(
+            threshold=threshold,
+            alert_handler=alert_handler,
+            auto_reset=auto_reset
+        )
+
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            try:
+                return func(*args, **kwargs)
+            except Exception as e:
+                monitor.increment()
+                raise  # 保持原始异常栈
+
+        # 暴露监控器以便外部访问
+        wrapper.monitor = monitor
+        return wrapper
+    return decorator

mns_common/component/exception/__init__.py

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

mns_common/component/main_line/__init__.py

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