mns-common 1.3.9.1__tar.gz → 1.3.9.3__tar.gz

{mns_common-1.3.9.1 → mns_common-1.3.9.3}/PKG-INFO

@@ -1,3 +1,3 @@
 Metadata-Version: 2.1
 Name: mns_common
-Version: 1.3.9.1
+Version: 1.3.9.3

{mns_common-1.3.9.1 → mns_common-1.3.9.3}/mns_common/component/common_service_fun_api.py

@@ -39,6 +39,20 @@ def classify_symbol(real_time_quotes_now_df):
     return real_time_quotes_now_df
 
 
+# 增加前缀
+def add_pre_prefix(real_time_quotes_now_df):
+    real_time_quotes_now_df['symbol_prefix'] = real_time_quotes_now_df['symbol'].apply(
+        lambda symbol: add_pre_prefix_one(symbol))
+    return real_time_quotes_now_df
+
+
+# 增加后缀
+def add_after_prefix(real_time_quotes_now_df):
+    real_time_quotes_now_df['symbol_prefix'] = real_time_quotes_now_df['symbol'].apply(
+        lambda symbol: add_after_prefix_one(symbol))
+    return real_time_quotes_now_df
+
+
 # 单个股票分类
 def classify_symbol_one(symbol):
     if symbol.startswith('3'):
@@ -54,6 +68,27 @@ def classify_symbol_one(symbol):
         return 'X'
 
 
+# 添加前缀
+def add_pre_prefix_one(symbol):
+    symbol_simple = symbol[0:6]
+    if symbol_simple.startswith('6'):
+        return 'SH' + symbol
+    elif symbol_simple.startswith('0') or symbol_simple.startswith('3'):
+        return 'SZ' + symbol
+    else:
+        return 'BJ' + symbol
+
+# 添加后缀
+def add_after_prefix_one(symbol):
+    symbol_simple = symbol[0:6]
+    if symbol_simple.startswith('6'):
+        return symbol + '.SH'
+    elif symbol_simple.startswith('0') or symbol_simple.startswith('3'):
+        return symbol + '.SZ'
+    else:
+        return symbol + '.BJ'
+
+
 def symbol_amount_simple(real_time_quotes_now_df):
     real_time_quotes_now_df['amount_level'] = round(real_time_quotes_now_df['amount'] / HUNDRED_MILLION, 2)
     return real_time_quotes_now_df

mns_common-1.3.9.3/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-1.3.9.3/mns_common/db/v2/__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-1.3.9.1 → mns_common-1.3.9.3}/mns_common.egg-info/PKG-INFO

@@ -1,3 +1,3 @@
 Metadata-Version: 2.1
 Name: mns-common
-Version: 1.3.9.1
+Version: 1.3.9.3

{mns_common-1.3.9.1 → mns_common-1.3.9.3}/mns_common.egg-info/SOURCES.txt

@@ -101,6 +101,8 @@ mns_common/component/deal/terminal_enum.py
 mns_common/component/em/__init__.py
 mns_common/component/em/em_real_time_quotes_api.py
 mns_common/component/em/em_stock_info_api.py
+mns_common/component/exception/ExceptionMonitor.py
+mns_common/component/exception/__init__.py
 mns_common/component/hk/__init__.py
 mns_common/component/hk/company_hk_service_api.py
 mns_common/component/industry/__init__.py

{mns_common-1.3.9.1 → mns_common-1.3.9.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='mns_common',
-    version='1.3.9.1',
+    version='1.3.9.3',
     # packages=["mns_common", "mns_common.api",
     #           "mns_common.api.akshare",
     #           "mns_common.api.em",