sycommon-python-lib 0.1.56b2__py3-none-any.whl → 0.1.56b3__py3-none-any.whl

sycommon/notice/uvicorn_monitor.py

@@ -0,0 +1,195 @@
+from datetime import datetime
+import sys
+import traceback
+import aiohttp
+import asyncio
+import json
+from typing import Optional
+from urllib.parse import urlparse, parse_qs, urlencode, urlunparse
+from sycommon.config.Config import Config
+
+
+async def send_wechat_markdown_msg(
+    content: str,
+    webhook: str = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=b9992872-ea66-494f-8683-f43411b7713a"
+) -> Optional[dict]:
+    """
+    异步发送企业微信Markdown格式的WebHook消息
+
+    Args:
+        content: Markdown格式的消息内容（支持企业微信支持的markdown语法）
+        webhook: 完整的企业微信WebHook URL（默认值包含key）
+
+    Returns:
+        接口返回的JSON数据（dict），失败返回None
+    """
+    # 设置请求头
+    headers = {
+        "Content-Type": "application/json; charset=utf-8"
+    }
+
+    # 构造请求体（Markdown格式）
+    payload = {
+        "msgtype": "markdown",
+        "markdown": {
+            "content": content
+        }
+    }
+
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                url=webhook,
+                data=json.dumps(payload, ensure_ascii=False),
+                headers=headers,
+                timeout=aiohttp.ClientTimeout(total=10)
+            ) as response:
+                status = response.status
+                response_text = await response.text()
+                response_data = json.loads(
+                    response_text) if response_text else {}
+
+                if status == 200 and response_data.get("errcode") == 0:
+                    print(f"消息发送成功: {response_data}")
+                    return response_data
+                else:
+                    print(f"消息发送失败 - 状态码: {status}, 响应: {response_data}")
+                    return None
+    except Exception as e:
+        print(f"错误：未知异常 - {str(e)}")
+        return None
+
+
+async def send_webhook(error_info: dict = None, webhook: str = None):
+    """
+    发送服务启动结果的企业微信通知
+    Args:
+        error_info: 错误信息字典（启动失败时必填），包含：error_type, error_msg, stack_trace, elapsed_time
+        webhook: 完整的企业微信WebHook URL（覆盖默认值）
+    """
+    # 获取服务名和环境（兼容配置读取失败）
+    try:
+        service_name = Config().config.get('Name', "未知服务")
+        env = Config().config.get('Nacos', {}).get('namespaceId', '未知环境')
+    except Exception as e:
+        service_name = "未知服务"
+        env = "未知环境"
+        print(f"读取配置失败: {str(e)}")
+
+    start_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+
+    # 启动失败的通知内容（包含详细错误信息）
+    error_type = error_info.get("error_type", "未知错误")
+    error_msg = error_info.get("error_msg", "无错误信息")
+    stack_trace = error_info.get("stack_trace", "无堆栈信息")[:500]  # 限制长度避免超限
+    elapsed_time = error_info.get("elapsed_time", 0)
+
+    markdown_content = f"""### {service_name}服务启动失败告警 ⚠️
+> 环境: <font color="warning">{env}</font>
+> 启动时间: <font color="comment">{start_time}</font>
+> 耗时: <font color="comment">{elapsed_time:.2f}秒</font>
+> 错误类型: <font color="danger">{error_type}</font>
+> 错误信息: <font color="danger">{error_msg}</font>
+> 错误堆栈: {stack_trace}"""
+
+    # 发送消息（优先使用传入的webhook，否则用默认值）
+    default_webhook = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=b9992872-ea66-494f-8683-f43411b7713a"
+    result = await send_wechat_markdown_msg(
+        content=markdown_content,
+        webhook=webhook if webhook else default_webhook
+    )
+    print(f"通知发送结果: {result}")
+
+
+def run(*args, webhook: str = None, **kwargs):
+    """
+    带企业微信告警的Uvicorn启动监控
+    调用方式1（默认配置）：uvicorn_monitor.run("app:app", **app.state.config)
+    调用方式2（自定义webhook）：uvicorn_monitor.run("app:app", webhook="完整URL", **app.state.config)
+
+    Args:
+        *args: 传递给uvicorn.run的位置参数（如"app:app"）
+        webhook: 完整的企业微信WebHook URL（可选，覆盖默认值）
+        **kwargs: 传递给uvicorn.run的关键字参数（如app.state.config）
+    """
+    # 判断环境
+    env = Config().config.get('Nacos', {}).get('namespaceId', '未知环境')
+    if env == "prod":
+        import uvicorn
+        uvicorn.run(*args, **kwargs)
+        return
+
+    # 记录启动开始时间
+    start_time = datetime.now()
+
+    if webhook:
+        # 脱敏展示webhook（隐藏key的后半部分）
+        parsed = urlparse(webhook)
+        query = parse_qs(parsed.query)
+        if 'key' in query and query['key'][0]:
+            key = query['key'][0]
+            masked_key = key[:8] + "****" if len(key) > 8 else key + "****"
+            query['key'] = [masked_key]
+            masked_query = urlencode(query, doseq=True)
+            masked_webhook = urlunparse(
+                (parsed.scheme, parsed.netloc, parsed.path, parsed.params, masked_query, parsed.fragment))
+            print(f"自定义企业微信WebHook: {masked_webhook}")
+
+    # 初始化错误信息
+    error_info = None
+
+    try:
+        import uvicorn
+        # 执行启动（如果启动成功，此方法会阻塞，不会执行后续except）
+        uvicorn.run(*args, **kwargs)
+
+    except KeyboardInterrupt:
+        # 处理用户手动中断（不算启动失败）
+        elapsed = (datetime.now() - start_time).total_seconds()
+        print(f"\n{'='*50}")
+        print(f"ℹ️  应用被用户手动中断")
+        print(f"启动耗时: {elapsed:.2f} 秒")
+        print(f"{'='*50}\n")
+        sys.exit(0)
+
+    except Exception as e:
+        # 捕获启动失败异常，收集错误信息
+        elapsed = (datetime.now() - start_time).total_seconds()
+        # 捕获堆栈信息
+        stack_trace = traceback.format_exc()
+
+        # 构造错误信息字典
+        error_info = {
+            "error_type": type(e).__name__,
+            "error_msg": str(e),
+            "stack_trace": stack_trace,
+            "elapsed_time": elapsed
+        }
+
+        # 打印错误信息
+        print(f"\n{'='*50}")
+        print(f"🚨 应用启动失败！")
+        print(f"失败时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+        print(f"错误类型: {type(e).__name__}")
+        print(f"错误信息: {str(e)}")
+        print(f"\n📝 错误堆栈（关键）:")
+        print(f"-"*50)
+        traceback.print_exc(file=sys.stdout)
+        print(f"\n⏱️  启动耗时: {elapsed:.2f} 秒")
+        print(f"{'='*50}\n")
+
+    finally:
+        # 运行异步通知函数，传递自定义的webhook参数
+        try:
+            asyncio.run(send_webhook(
+                error_info=error_info,
+                webhook=webhook
+            ))
+        except Exception as e:
+            print(f"错误：异步通知失败 - {str(e)}")
+        # 启动失败时退出程序
+        sys.exit(1)
+
+
+# 兼容旧调用方式（可选）
+run_uvicorn_with_monitor = run

sycommon/synacos/nacos_service.py

@@ -1,4 +1,3 @@
-import logging
 import threading
 import json
 from typing import Callable, Dict, List, Optional

{sycommon_python_lib-0.1.56b2.dist-info → sycommon_python_lib-0.1.56b3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sycommon-python-lib
-Version: 0.1.56b2
+Version: 0.1.56b3
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{sycommon_python_lib-0.1.56b2.dist-info → sycommon_python_lib-0.1.56b3.dist-info}/RECORD

@@ -43,6 +43,8 @@ sycommon/models/mqlistener_config.py,sha256=vXp2uMmd0XQ5B9noSRXWHewTy-juQ2y7IsWt
 sycommon/models/mqmsg_model.py,sha256=cxn0M5b0utQK6crMYmL-1waeGYHvK3AlGaRy23clqTE,277
 sycommon/models/mqsend_config.py,sha256=NQX9dc8PpuquMG36GCVhJe8omAW1KVXXqr6lSRU6D7I,268
 sycommon/models/sso_user.py,sha256=i1WAN6k5sPcPApQEdtjpWDy7VrzWLpOrOQewGLGoGIw,2702
+sycommon/notice/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sycommon/notice/uvicorn_monitor.py,sha256=NRQVWs3jzZq9IY0M55s7LvOKtacsNpfCGGzH5071dGw,7181
 sycommon/rabbitmq/rabbitmq_client.py,sha256=GkuYILMZJnvgZs4ID46I-w_UGzzI28YUydKgkTIDhIs,20226
 sycommon/rabbitmq/rabbitmq_pool.py,sha256=QtUcK4HuepRqRmy5XkUQo8gDgj74fr77CX7T5rN0y4I,15640
 sycommon/rabbitmq/rabbitmq_service.py,sha256=wpEJynP0gzbnmMyB02sfR9nTWB4hfTTP00xQxDJu644,38119
@@ -54,15 +56,15 @@ sycommon/synacos/example.py,sha256=61XL03tU8WTNOo3FUduf93F2fAwah1S0lbH1ufhRhRk,5
 sycommon/synacos/example2.py,sha256=adUaru3Hy482KrOA17DfaC4nwvLj8etIDS_KrWLWmCU,4811
 sycommon/synacos/feign.py,sha256=frB3D5LeFDtT3pJLFOwFzEOrNAJKeQNGk-BzUg9T3WM,8295
 sycommon/synacos/feign_client.py,sha256=ExO7Pd5B3eFKDjXqBRc260K1jkI49IYguLwJJaD2R-o,16166
-sycommon/synacos/nacos_service.py,sha256=cvmjd9dJIzaOFm6IXGjy2BCq1gHTHRxLYdgt5kBguUw,35976
+sycommon/synacos/nacos_service.py,sha256=QPdCRFPXWUUoHL72VCjBAImdIo175q2plDO1T0xrPAg,35961
 sycommon/synacos/param.py,sha256=KcfSkxnXOa0TGmCjY8hdzU9pzUsA8-4PeyBKWI2-568,1765
 sycommon/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sycommon/tools/docs.py,sha256=OPj2ETheuWjXLyaXtaZPbwmJKfJaYXV5s4XMVAUNrms,1607
 sycommon/tools/merge_headers.py,sha256=HV_i52Q-9se3SP8qh7ZGYl8bP7Fxtal4CGVkyMwEdM8,4373
 sycommon/tools/snowflake.py,sha256=lVEe5mNCOgz5OqGQpf5_nXaGnRJlI2STX2s-ppTtanA,11947
 sycommon/tools/timing.py,sha256=OiiE7P07lRoMzX9kzb8sZU9cDb0zNnqIlY5pWqHcnkY,2064
-sycommon_python_lib-0.1.56b2.dist-info/METADATA,sha256=HzNJOWrTsJrz8BkzZwJRiq6r9ugYormfdmCEFPM_e_g,7226
-sycommon_python_lib-0.1.56b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sycommon_python_lib-0.1.56b2.dist-info/entry_points.txt,sha256=q_h2nbvhhmdnsOUZEIwpuoDjaNfBF9XqppDEmQn9d_A,46
-sycommon_python_lib-0.1.56b2.dist-info/top_level.txt,sha256=98CJ-cyM2WIKxLz-Pf0AitWLhJyrfXvyY8slwjTXNuc,17
-sycommon_python_lib-0.1.56b2.dist-info/RECORD,,
+sycommon_python_lib-0.1.56b3.dist-info/METADATA,sha256=J7AWcuKaJxgpLHHVQs9qYPokB_-oge0eOD6q8tGIrXo,7226
+sycommon_python_lib-0.1.56b3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sycommon_python_lib-0.1.56b3.dist-info/entry_points.txt,sha256=q_h2nbvhhmdnsOUZEIwpuoDjaNfBF9XqppDEmQn9d_A,46
+sycommon_python_lib-0.1.56b3.dist-info/top_level.txt,sha256=98CJ-cyM2WIKxLz-Pf0AitWLhJyrfXvyY8slwjTXNuc,17
+sycommon_python_lib-0.1.56b3.dist-info/RECORD,,