pixelarraylib 1.2.2__tar.gz → 1.2.4__tar.gz

{pixelarraylib-1.2.2/pixelarraylib.egg-info → pixelarraylib-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pixelarraylib
-Version: 1.2.2
+Version: 1.2.4
 Summary: PixelArray Python开发工具库 - 包含阿里云服务、数据库工具、装饰器、监控等功能
 Author-email: Lu qi <qi.lu@pixelarrayai.com>
 License-Expression: MIT

{pixelarraylib-1.2.2 → pixelarraylib-1.2.4}/pixelarraylib/__init__.py

@@ -20,7 +20,7 @@ PixelArray Python开发工具库
     from pixelarraylib.gitlab import pypi_package_manager
 """
 
-__version__ = "1.2.2"
+__version__ = "1.2.4"
 __author__ = "PixelArray"
 __email__ = "qi.lu@pixelarrayai.com"
 

{pixelarraylib-1.2.2 → pixelarraylib-1.2.4}/pixelarraylib/monitor/feishu.py

@@ -20,6 +20,8 @@ class Feishu:
         "picturetransform服务报警": "https://open.feishu.cn/open-apis/bot/v2/hook/e975aa0a-acef-4e3f-bee4-6dc507b87ebd",
         "cloudstorage服务报警": "https://open.feishu.cn/open-apis/bot/v2/hook/a632d0bc-e400-40ce-a3bb-1b9b9ee634f4",
         "deployengine服务报警": "https://open.feishu.cn/open-apis/bot/v2/hook/9a347a63-58fb-4e10-9a3d-ff4918ddb3a9",
+        "browseruse服务报警": "https://open.feishu.cn/open-apis/bot/v2/hook/a7cf22c5-f50d-4031-b17f-a75f92a3aa91",
+        "openpaper_manage_console服务报警": "https://open.feishu.cn/open-apis/bot/v2/hook/2829b285-03ae-4282-9ae5-c93cec4d62a0",
     }
 
     def __init__(
@@ -571,3 +573,224 @@ class FeishuDocumentManager:
             return result, True
         except requests.RequestException as e:
             return {"code": -1, "msg": f"请求失败: {str(e)}"}, False
+
+
+class FeishuUserManager:
+    """
+    飞书用户管理器，用于管理飞书用户的发送消息和查询
+    """
+
+    def __init__(
+        self,
+        app_id: Optional[str] = None,
+        app_secret: Optional[str] = None,
+    ):
+        """
+        description:
+            初始化飞书用户管理器
+        parameters:
+            app_id(Optional[str]): 飞书应用ID，如果不提供则从settings中读取
+            app_secret(Optional[str]): 飞书应用密钥，如果不提供则从settings中读取
+        """
+        try:
+            from settings import FEISHU_APP_ID, FEISHU_APP_SECRET
+            self.app_id = app_id or FEISHU_APP_ID
+            self.app_secret = app_secret or FEISHU_APP_SECRET
+        except ImportError:
+            if not app_id or not app_secret:
+                raise ValueError(
+                    "必须提供app_id和app_secret，或者在settings中配置FEISHU_APP_ID和FEISHU_APP_SECRET"
+                )
+            self.app_id = app_id
+            self.app_secret = app_secret
+
+        self.base_url = "https://open.feishu.cn/open-apis"
+        self._tenant_access_token = None
+        self._token_expire_time = 0
+
+    def _get_tenant_access_token(self) -> str:
+        """
+        description:
+            获取飞书应用的租户访问令牌（tenant_access_token，带缓存机制）
+        return:
+            str: 租户访问令牌
+        """
+        import time
+
+        # 如果token未过期，直接返回缓存的token
+        if self._tenant_access_token and time.time() < self._token_expire_time:
+            return self._tenant_access_token
+
+        url = f"{self.base_url}/auth/v3/tenant_access_token/internal"
+        headers = {"Content-Type": "application/json; charset=utf-8"}
+        data = {
+            "app_id": self.app_id,
+            "app_secret": self.app_secret,
+        }
+
+        try:
+            response = requests.post(url, headers=headers, json=data)
+            response.raise_for_status()
+            result = response.json()
+
+            if result.get("code") != 0:
+                raise Exception(
+                    f"获取tenant_access_token失败: {result.get('msg', 'unknown error')}"
+                )
+
+            self._tenant_access_token = result.get("tenant_access_token")
+            # token有效期默认7200秒，提前300秒刷新
+            expire = result.get("expire", 7200)
+            self._token_expire_time = time.time() + expire - 300
+
+            return self._tenant_access_token
+        except requests.RequestException as e:
+            raise Exception(f"请求tenant_access_token时发生错误: {str(e)}")
+
+    def _get_headers(self) -> dict:
+        """
+        description:
+            获取请求头，包含tenant_access_token
+        return:
+            dict: 请求头字典
+        """
+        return {
+            "Authorization": f"Bearer {self._get_tenant_access_token()}",
+            "Content-Type": "application/json; charset=utf-8",
+        }
+
+    def send_message_to_user(
+        self,
+        receive_id: str,
+        text: str,
+        receive_id_type: str = "open_id",
+    ) -> tuple[dict, bool]:
+        """
+        description:
+            发送消息给指定的飞书用户
+        parameters:
+            receive_id(str): 接收者ID（用户ID）
+            text(str): 要发送的文本内容
+            receive_id_type(str): 接收者ID类型，默认为"open_id"，可选值："open_id"、"user_id"、"union_id"、"email"、"chat_id"
+        return:
+            result(dict): API返回的结果
+            success(bool): 是否成功
+        """
+        url = f"{self.base_url}/im/v1/messages"
+        headers = self._get_headers()
+        # receive_id 和 receive_id_type 作为查询参数
+        params = {
+            "receive_id_type": receive_id_type,
+        }
+        # 请求体只包含消息类型和内容
+        data = {
+            "receive_id": receive_id,
+            "msg_type": "text",
+            "content": json.dumps({"text": text}),
+        }
+
+        try:
+            response = requests.post(url, headers=headers, params=params, json=data)
+            response.raise_for_status()
+            result = response.json()
+
+            if result.get("code") != 0:
+                return result, False
+
+            return result, True
+        except requests.RequestException as e:
+            # 尝试获取更详细的错误信息
+            error_detail = {}
+            if hasattr(e, 'response') and e.response is not None:
+                try:
+                    error_detail = e.response.json()
+                except:
+                    error_detail = {"text": e.response.text if hasattr(e.response, 'text') else str(e)}
+            return {"code": -1, "msg": f"请求失败: {str(e)}", "detail": error_detail}, False
+
+    async def send_message_to_user_async(
+        self,
+        receive_id: str,
+        text: str,
+        receive_id_type: str = "open_id",
+    ) -> tuple[dict, bool]:
+        """
+        description:
+            异步发送消息给指定的飞书用户
+        parameters:
+            receive_id(str): 接收者ID（用户ID）
+            text(str): 要发送的文本内容
+            receive_id_type(str): 接收者ID类型，默认为"open_id"
+        return:
+            result(dict): API返回的结果
+            success(bool): 是否成功
+        """
+        return await asyncio.to_thread(
+            self.send_message_to_user, receive_id, text, receive_id_type
+        )
+
+    def list_users(
+        self,
+        department_id: Optional[str] = None,
+        page_size: int = 50,
+        page_token: Optional[str] = None,
+        user_id_type: str = "open_id",
+    ) -> tuple[dict, bool]:
+        """
+        description:
+            列出飞书中的所有用户（支持按部门筛选）
+        parameters:
+            department_id(Optional[str]): 部门ID，如果提供则只返回该部门的用户
+            page_size(int): 每页数量，默认50，最大100
+            page_token(Optional[str]): 分页token，用于获取下一页
+            user_id_type(str): 用户ID类型，默认为"open_id"，可选值："open_id"、"user_id"、"union_id"
+        return:
+            result(dict): API返回的结果，包含用户列表
+            success(bool): 是否成功
+        """
+        url = f"{self.base_url}/contact/v3/users"
+        headers = self._get_headers()
+        params = {
+            "page_size": min(page_size, 100),  # 限制最大值为100
+            "user_id_type": user_id_type,
+        }
+
+        if department_id:
+            params["department_id"] = department_id
+        if page_token:
+            params["page_token"] = page_token
+
+        try:
+            response = requests.get(url, headers=headers, params=params)
+            response.raise_for_status()
+            result = response.json()
+
+            if result.get("code") != 0:
+                return result, False
+
+            return result, True
+        except requests.RequestException as e:
+            return {"code": -1, "msg": f"请求失败: {str(e)}"}, False
+
+    async def list_users_async(
+        self,
+        department_id: Optional[str] = None,
+        page_size: int = 50,
+        page_token: Optional[str] = None,
+        user_id_type: str = "open_id",
+    ) -> tuple[dict, bool]:
+        """
+        description:
+            异步列出飞书中的所有用户
+        parameters:
+            department_id(Optional[str]): 部门ID，如果提供则只返回该部门的用户
+            page_size(int): 每页数量，默认50
+            page_token(Optional[str]): 分页token，用于获取下一页
+            user_id_type(str): 用户ID类型，默认为"open_id"
+        return:
+            result(dict): API返回的结果，包含用户列表
+            success(bool): 是否成功
+        """
+        return await asyncio.to_thread(
+            self.list_users, department_id, page_size, page_token, user_id_type
+        )