兼容企业微信

yj

2025-09-03 f28ac0166536a2a4b68cac685a41ea667f60f7e9

 app/services/ecloud_client.py

@@ -3,7 +3,7 @@
"""

import requests
from typing import Optional, Dict, Any
from typing import Optional, Dict, Any, List
from loguru import logger
from config import settings

@@ -20,16 +20,40 @@
        self.session = requests.Session()
        self.session.headers.update(self.headers)

    def _filter_keywords(self, content: str) -> str:
        """
        过滤消息内容中的关键词

        Args:
            content: 原始消息内容

        Returns:
            过滤后的消息内容
        """
        if not settings.keyword_filter_enabled or not settings.keyword_filter_keywords:
            return content

        filtered_content = content
        for keyword in settings.keyword_filter_keywords:
            if keyword in filtered_content:
                filtered_content = filtered_content.replace(keyword, "")
                logger.info(f"过滤关键词: {keyword}")
        # 去除前后空格
        filtered_content = filtered_content.strip()
        return filtered_content

    def get_contact_info(self, w_id: str, wc_id: str) -> Optional[Dict[str, Any]]:
        """
        获取联系人信息

        Args:
            w_id: 登录实例标识
            wc_id: 好友微信id/群id
            wc_id: 好友微信id/群id，多个使用英文逗号分隔

        Returns:
            联系人信息字典，失败返回None
            联系人信息字典或列表，失败返回None
            - 单个wcId时返回字典
            - 多个wcId时返回列表
        """
        try:
            url = f"{self.base_url}/getContact"
@@ -45,8 +69,14 @@
            if result.get("code") == "1000":
                contact_data = result.get("data", [])
                if contact_data and len(contact_data) > 0:
                    logger.info(f"成功获取联系人信息: wcId={wc_id}")
                    return contact_data[0]  # 返回第一个联系人信息
                    # 如果是单个wcId，返回第一个联系人信息
                    if "," not in wc_id:
                        logger.info(f"成功获取联系人信息: wcId={wc_id}")
                        return contact_data[0]
                    else:
                        # 如果是多个wcId，返回完整列表
                        logger.info(f"成功获取批量联系人信息: count={len(contact_data)}")
                        return contact_data
                else:
                    logger.warning(f"联系人信息为空: wcId={wc_id}")
                    return None
@@ -63,7 +93,7 @@
            logger.error(f"获取联系人信息异常: wcId={wc_id}, error={str(e)}")
            return None

    def send_text_message(self, w_id: str, wc_id: str, content: str) -> bool:
    def send_text_message(self, w_id: str, wc_id: str, content: str, max_retries: int = None) -> bool:
        """
        发送文本消息

@@ -71,38 +101,58 @@
            w_id: 登录实例标识
            wc_id: 接收人微信id/群id
            content: 文本内容消息
            max_retries: 最大重试次数

        Returns:
            发送成功返回True，失败返回False
        """
        try:
            url = f"{self.base_url}/sendText"
            payload = {"wId": w_id, "wcId": wc_id, "content": content}
        # 过滤关键词
        filtered_content = self._filter_keywords(content)

            logger.info(
                f"发送文本消息: wId={w_id}, wcId={wc_id}, content_length={len(content)}"
            )
        if max_retries is None:
            from config import settings
            max_retries = settings.max_retry_count

            response = self.session.post(url, json=payload, timeout=30)
            response.raise_for_status()
        retry_count = 0
        while retry_count <= max_retries:
            try:
                url = f"{self.base_url}/sendText"
                payload = {"wId": w_id, "wcId": wc_id, "content": filtered_content}

            result = response.json()

            if result.get("code") == "1000":
                logger.info(f"文本消息发送成功: wcId={wc_id}")
                return True
            else:
                logger.error(
                    f"文本消息发送失败: wcId={wc_id}, code={result.get('code')}, message={result.get('message')}"
                logger.info(
                    f"发送文本消息: wId={w_id}, wcId={wc_id}, content_length={len(filtered_content)}, retry={retry_count}"
                )
                return False

        except requests.exceptions.RequestException as e:
            logger.error(f"发送文本消息网络错误: wcId={wc_id}, error={str(e)}")
            return False
        except Exception as e:
            logger.error(f"发送文本消息异常: wcId={wc_id}, error={str(e)}")
            return False
                response = self.session.post(url, json=payload, timeout=30)
                response.raise_for_status()

                result = response.json()

                if result.get("code") == "1000":
                    logger.info(f"文本消息发送成功: wcId={wc_id}")
                    return True
                else:
                    logger.error(
                        f"文本消息发送失败: wcId={wc_id}, code={result.get('code')}, message={result.get('message')}"
                    )

            except requests.exceptions.RequestException as e:
                logger.error(f"发送文本消息网络错误: wcId={wc_id}, retry={retry_count}, error={str(e)}")
            except Exception as e:
                logger.error(f"发送文本消息异常: wcId={wc_id}, retry={retry_count}, error={str(e)}")
            
            retry_count += 1
            if retry_count <= max_retries:
                from config import settings
                wait_time = settings.retry_delay * retry_count
                logger.info(f"等待重试: wcId={wc_id}, wait_time={wait_time}s")
                import time
                time.sleep(wait_time)

        logger.error(
            f"文本消息发送失败，已达最大重试次数: wcId={wc_id}, max_retries={max_retries}"
        )
        return False

    def send_group_message(self, w_id: str, group_id: str, content: str) -> bool:
        """
@@ -116,7 +166,200 @@
        Returns:
            发送成功返回True，失败返回False
        """
        return self.send_text_message(w_id, group_id, content)
        # 过滤关键词
        filtered_content = self._filter_keywords(content)
        return self.send_text_message(w_id, group_id, filtered_content)

    def init_address_list(self, w_id: str) -> bool:
        """
        初始化通讯录列表

        Args:
            w_id: 登录实例标识

        Returns:
            初始化成功返回True，失败返回False
        """
        try:
            url = f"{self.base_url}/initAddressList"
            payload = {"wId": w_id}

            logger.info(f"初始化通讯录列表: wId={w_id}")

            response = self.session.post(url, json=payload, timeout=30)
            response.raise_for_status()

            result = response.json()

            if result.get("code") == "1000":
                logger.info(f"初始化通讯录列表成功: wId={w_id}")
                return True
            else:
                logger.error(
                    f"初始化通讯录列表失败: wId={w_id}, code={result.get('code')}, message={result.get('message')}"
                )
                return False

        except requests.exceptions.RequestException as e:
            logger.error(f"初始化通讯录列表网络错误: wId={w_id}, error={str(e)}")
            return False
        except Exception as e:
            logger.error(f"初始化通讯录列表异常: wId={w_id}, error={str(e)}")
            return False

    def get_address_list(self, w_id: str) -> Optional[Dict[str, Any]]:
        """
        获取通讯录列表

        Args:
            w_id: 登录实例标识

        Returns:
            通讯录数据字典，失败返回None
            返回格式: {
                "chatrooms": [...],  # 群组列表
                "friends": [...],    # 好友列表
                "ghs": [...],        # 公众号列表
                "others": [...]      # 其他
            }
        """
        try:
            url = f"{self.base_url}/getAddressList"
            payload = {"wId": w_id}

            logger.info(f"获取通讯录列表: wId={w_id}")

            response = self.session.post(url, json=payload, timeout=30)
            response.raise_for_status()

            result = response.json()

            if result.get("code") == "1000":
                address_data = result.get("data", {})
                logger.info(f"成功获取通讯录列表: wId={w_id}, friends_count={len(address_data.get('friends', []))}")
                return address_data
            else:
                logger.error(
                    f"获取通讯录列表失败: wId={w_id}, code={result.get('code')}, message={result.get('message')}"
                )
                return None

        except requests.exceptions.RequestException as e:
            logger.error(f"获取通讯录列表网络错误: wId={w_id}, error={str(e)}")
            return None
        except Exception as e:
            logger.error(f"获取通讯录列表异常: wId={w_id}, error={str(e)}")
            return None

    def send_group_at_message(self, w_id: str, wc_id: str, content: str, at_wc_ids: List[str], max_retries: int = None) -> bool:
        """
        发送群聊@消息

        Args:
            w_id: 登录实例标识
            wc_id: 接收方群id
            content: 文本内容消息（@的微信昵称需要自己拼接，必须拼接艾特符号，不然不生效）
            at_wc_ids: 艾特的微信id列表
            max_retries: 最大重试次数

        Returns:
            发送成功返回True，失败返回False
        """
        # 过滤关键词
        filtered_content = self._filter_keywords(content)

        if max_retries is None:
            from config import settings
            max_retries = settings.max_retry_count

        retry_count = 0
        while retry_count <= max_retries:
            try:
                url = f"{self.base_url}/sendText"
                # 将at_wc_ids列表用逗号拼接
                at_str = ",".join(at_wc_ids)
                payload = {
                    "wId": w_id,
                    "wcId": wc_id,
                    "content": filtered_content,
                    "at": at_str
                }

                logger.info(
                    f"发送群聊@消息: wId={w_id}, wcId={wc_id}, at={at_str}, content_length={len(filtered_content)}, retry={retry_count}"
                )

                response = self.session.post(url, json=payload, timeout=30)
                response.raise_for_status()

                result = response.json()

                if result.get("code") == "1000":
                    logger.info(f"群聊@消息发送成功: wcId={wc_id}, at={at_str}")
                    return True
                else:
                    logger.error(
                        f"群聊@消息发送失败: wcId={wc_id}, at={at_str}, code={result.get('code')}, message={result.get('message')}"
                    )

            except requests.exceptions.RequestException as e:
                logger.error(f"发送群聊@消息网络错误: wcId={wc_id}, at={at_str}, retry={retry_count}, error={str(e)}")
            except Exception as e:
                logger.error(f"发送群聊@消息异常: wcId={wc_id}, at={at_str}, retry={retry_count}, error={str(e)}")

            retry_count += 1
            if retry_count <= max_retries:
                from config import settings
                wait_time = settings.retry_delay * retry_count
                logger.info(f"等待重试: wcId={wc_id}, wait_time={wait_time}s")
                import time
                time.sleep(wait_time)

        logger.error(
            f"群聊@消息发送失败，已达最大重试次数: wcId={wc_id}, at={at_str}, max_retries={max_retries}"
        )
        return False

    def query_online_wechat_list(self) -> Optional[List[Dict[str, str]]]:
        """
        查询账号中在线的微信列表

        Returns:
            在线微信列表，每个元素包含wcId和wId，失败返回None
            返回格式: [
                {
                    "wcId": "wxid_i6qsbbjenju2",
                    "wId": "72223018-7f2a-4f4f-bfa3-26e47dbd61"
                }
            ]
        """
        try:
            url = f"{self.base_url}/queryLoginWx"
            payload = {}

            logger.info("查询在线微信列表")

            response = self.session.post(url, json=payload, timeout=30)
            response.raise_for_status()

            result = response.json()

            if result.get("code") == "1000":
                online_list = result.get("data", [])
                logger.info(f"成功查询在线微信列表: count={len(online_list)}")
                return online_list
            else:
                logger.warning(
                    f"查询在线微信列表失败: code={result.get('code')}, message={result.get('message')}"
                )
                return []

        except requests.exceptions.RequestException as e:
            logger.error(f"查询在线微信列表网络错误: error={str(e)}")
            return None
        except Exception as e:
            logger.error(f"查询在线微信列表异常: error={str(e)}")
            return None


# 全局E云管家客户端实例