Remove legacy test script and enhance main application with WeChat server and menu management features. Added commands for starting the WeChat callback server and managing WeChat menus, along with necessary imports and error handling. Updated requirements to include Flask for the WeChat server functionality.

2026-01-08 15:16:42 +08:00
parent 317dbf3c58
commit a8108d9afd
25 changed files with 3089 additions and 152 deletions
@@ -0,0 +1,357 @@
+"""
+企业微信消息处理器
+处理用户消息和事件，实现菜单功能
+"""
+
+import time
+import json
+from typing import Optional, Dict, Any, List
+from datetime import datetime
+
+try:
+    from ..core.config_manager import get_config
+    from ..core.logger import get_logger
+    from ..notification.wechat import send_system_notification
+    from ..storage.postgresql import save_all_announcements_by_source_to_storage
+    from ..storage.md_generator import generate_onu_md
+    from ..core.models import Announcement
+except ImportError:
+    try:
+        from core.config_manager import get_config
+        from core.logger import get_logger
+        from notification.wechat import send_system_notification
+        from storage.postgresql import save_all_announcements_by_source_to_storage
+        from storage.md_generator import generate_onu_md
+        from core.models import Announcement
+    except ImportError as e:
+        raise ImportError(f"消息处理器导入失败: {e}")
+
+
+logger = get_logger(__name__)
+
+
+class WeChatMessageHandler:
+    """企业微信消息处理器"""
+
+    def __init__(self):
+        self.config = get_config()
+        self.monitor_app = None
+
+        # 菜单配置
+        self.menu_config = {
+            "crawl": {
+                "key": "crawl_now",
+                "name": "立即爬取",
+                "description": "立即执行一次公告爬取"
+            },
+            "today_summary": {
+                "key": "today_summary",
+                "name": "今日总结",
+                "description": "查看今日公告统计"
+            },
+            "custom_crawl": {
+                "key": "custom_crawl",
+                "name": "自定义爬取",
+                "description": "输入关键词进行爬取"
+            }
+        }
+
+        logger.info("企业微信消息处理器初始化完成")
+
+    def _get_monitor_app(self):
+        """获取监控应用实例"""
+        if self.monitor_app is None:
+            # 动态导入避免循环导入
+            try:
+                from ..main import GXGPMonitorApp
+                self.monitor_app = GXGPMonitorApp()
+                # 初始化但不启动服务器
+                if not self.monitor_app.initialize():
+                    logger.error("监控应用初始化失败")
+                    return None
+            except ImportError:
+                logger.error("无法导入监控应用")
+                return None
+        return self.monitor_app
+
+    def handle_event(self, event: str, event_key: Optional[str], from_user: str) -> Optional[str]:
+        """处理事件消息"""
+        try:
+            logger.info(f"处理事件: {event}, key: {event_key}, user: {from_user}")
+
+            if event == 'click':
+                # 菜单点击事件
+                if event_key == 'crawl_now':
+                    return self._handle_crawl_now(from_user)
+                elif event_key == 'today_summary':
+                    return self._handle_today_summary(from_user)
+                elif event_key.startswith('custom_crawl'):
+                    return self._handle_custom_crawl(event_key, from_user)
+                else:
+                    return self._create_text_response("未知菜单项", from_user)
+
+            elif event == 'subscribe':
+                # 关注事件
+                welcome_msg = """欢迎关注广西政府采购网公告监控！
+
+我可以帮您：
+• 自动监控最新采购公告
+• 筛选您关心的关键词信息
+• 及时推送重要更新
+
+点击下方菜单开始使用。"""
+                return self._create_text_response(welcome_msg, from_user)
+
+            elif event == 'unsubscribe':
+                # 取消关注事件
+                logger.info(f"用户 {from_user} 取消关注")
+                return None
+
+            else:
+                logger.info(f"未处理的event类型: {event}")
+                return None
+
+        except Exception as e:
+            logger.error(f"事件处理异常: {str(e)}")
+            return self._create_text_response("处理失败，请稍后重试", from_user)
+
+    def handle_text_message(self, content: str, from_user: str) -> Optional[str]:
+        """处理文本消息"""
+        try:
+            logger.info(f"处理文本消息: {content}, user: {from_user}")
+
+            # 移除前后空格
+            content = content.strip()
+
+            if content == "帮助" or content == "help":
+                return self._handle_help(from_user)
+            elif content.startswith("爬取"):
+                return self._handle_manual_crawl(content, from_user)
+            elif content.startswith("总结"):
+                return self._handle_today_summary(from_user)
+            elif content.startswith("关键词"):
+                return self._handle_keyword_search(content, from_user)
+            else:
+                # 默认当作关键词搜索
+                return self._handle_keyword_search(f"关键词 {content}", from_user)
+
+        except Exception as e:
+            logger.error(f"文本消息处理异常: {str(e)}")
+            return self._create_text_response("处理失败，请稍后重试", from_user)
+
+    def handle_other_message(self, msg_type: str, from_user: str) -> Optional[str]:
+        """处理其他类型的消息"""
+        try:
+            logger.info(f"处理其他消息类型: {msg_type}, user: {from_user}")
+
+            if msg_type == 'image':
+                return self._create_text_response("收到图片消息，但我只能处理文本消息", from_user)
+            elif msg_type == 'voice':
+                return self._create_text_response("收到语音消息，但我只能处理文本消息", from_user)
+            else:
+                return self._create_text_response(f"收到{msg_type}消息，暂不支持此类型", from_user)
+
+        except Exception as e:
+            logger.error(f"其他消息处理异常: {str(e)}")
+            return self._create_text_response("处理失败，请稍后重试", from_user)
+
+    def _handle_crawl_now(self, from_user: str) -> Optional[str]:
+        """处理立即爬取菜单"""
+        try:
+            logger.info(f"用户 {from_user} 触发立即爬取")
+
+            # 获取监控应用
+            app = self._get_monitor_app()
+            if not app:
+                return self._create_text_response("系统初始化失败，请稍后重试", from_user)
+
+            # 执行爬取
+            result = app.run_crawl()
+
+            if result.get("success"):
+                total = result.get("total_crawled", 0)
+                filtered = result.get("filtered", 0)
+                saved = result.get("saved", 0)
+
+                response = f"""✅ 爬取完成！
+
+📊 统计信息：
+• 总共发现: {total} 条公告
+• 关键词筛选: {filtered} 条
+• 已保存: {saved} 条
+
+如有匹配的公告，我会及时推送通知。"""
+            else:
+                error = result.get("error", "未知错误")
+                response = f"❌ 爬取失败: {error}"
+
+            return self._create_text_response(response, from_user)
+
+        except Exception as e:
+            logger.error(f"立即爬取处理异常: {str(e)}")
+            return self._create_text_response("爬取失败，请稍后重试", from_user)
+
+    def _handle_today_summary(self, from_user: str) -> Optional[str]:
+        """处理今日总结菜单"""
+        try:
+            logger.info(f"用户 {from_user} 请求今日总结")
+
+            # 这里可以查询今日的公告统计
+            # 由于数据库查询较为复杂，这里先返回简单的响应
+            response = """📅 今日公告统计
+
+由于系统正在优化中，今日统计功能暂时不可用。
+
+您可以：
+• 点击"立即爬取"获取最新数据
+• 发送关键词进行搜索
+• 发送"帮助"查看更多功能"""
+
+            return self._create_text_response(response, from_user)
+
+        except Exception as e:
+            logger.error(f"今日总结处理异常: {str(e)}")
+            return self._create_text_response("获取统计失败，请稍后重试", from_user)
+
+    def _handle_custom_crawl(self, event_key: str, from_user: str) -> Optional[str]:
+        """处理自定义爬取菜单"""
+        try:
+            logger.info(f"用户 {from_user} 触发自定义爬取")
+
+            response = """🔍 自定义爬取
+
+请回复您想要搜索的关键词，我将为您执行爬取并筛选相关公告。
+
+例如：
+• 大化
+• 信息化
+• 政府采购
+
+发送关键词开始搜索。"""
+
+            return self._create_text_response(response, from_user)
+
+        except Exception as e:
+            logger.error(f"自定义爬取处理异常: {str(e)}")
+            return self._create_text_response("操作失败，请稍后重试", from_user)
+
+    def _handle_help(self, from_user: str) -> Optional[str]:
+        """处理帮助命令"""
+        help_text = """🤖 广西政府采购网公告监控助手
+
+📋 菜单功能：
+• 立即爬取 - 执行一次公告爬取
+• 今日总结 - 查看今日公告统计
+• 自定义爬取 - 输入关键词搜索
+
+💬 文本命令：
+• 发送关键词 - 搜索相关公告
+• "爬取 [关键词]" - 指定关键词爬取
+• "总结" - 查看今日统计
+• "帮助" - 显示此帮助信息
+
+📢 自动推送：
+系统会自动监控最新公告，并推送匹配关键词的信息。
+
+💡 使用提示：
+• 关键词支持多个，用空格分隔
+• 公告按时间倒序显示
+• 点击公告标题可查看详情"""
+
+        return self._create_text_response(help_text, from_user)
+
+    def _handle_manual_crawl(self, content: str, from_user: str) -> Optional[str]:
+        """处理手动爬取命令"""
+        try:
+            # 解析关键词
+            parts = content.split()
+            if len(parts) < 2:
+                return self._create_text_response("请指定爬取关键词，例如：爬取 大化", from_user)
+
+            keywords = parts[1:]
+            logger.info(f"用户 {from_user} 手动爬取关键词: {keywords}")
+
+            # 获取监控应用
+            app = self._get_monitor_app()
+            if not app:
+                return self._create_text_response("系统初始化失败，请稍后重试", from_user)
+
+            # 执行爬取
+            result = app.run_crawl(keywords=keywords)
+
+            if result.get("success"):
+                total = result.get("total_crawled", 0)
+                filtered = result.get("filtered", 0)
+
+                if filtered > 0:
+                    response = f"""✅ 爬取完成！
+
+关键词: {' '.join(keywords)}
+📊 统计信息：
+• 总共发现: {total} 条公告
+• 匹配筛选: {filtered} 条
+
+相关公告已推送，请查收。"""
+                else:
+                    response = f"""✅ 爬取完成！
+
+关键词: {' '.join(keywords)}
+📊 统计信息：
+• 总共发现: {total} 条公告
+• 匹配筛选: 0 条
+
+没有找到匹配的公告。"""
+            else:
+                error = result.get("error", "未知错误")
+                response = f"❌ 爬取失败: {error}"
+
+            return self._create_text_response(response, from_user)
+
+        except Exception as e:
+            logger.error(f"手动爬取处理异常: {str(e)}")
+            return self._create_text_response("爬取失败，请稍后重试", from_user)
+
+    def _handle_keyword_search(self, content: str, from_user: str) -> Optional[str]:
+        """处理关键词搜索"""
+        try:
+            # 解析关键词
+            parts = content.split()
+            keywords = parts[1:] if len(parts) > 1 else parts
+
+            if not keywords:
+                return self._create_text_response("请提供搜索关键词", from_user)
+
+            logger.info(f"用户 {from_user} 关键词搜索: {keywords}")
+
+            # 这里可以实现关键词搜索逻辑
+            # 目前先返回提示信息
+            response = f"""🔍 关键词搜索
+
+搜索关键词: {' '.join(keywords)}
+
+由于系统正在优化中，搜索功能暂时不可用。
+
+您可以：
+• 使用"爬取 [关键词]"执行新的爬取
+• 点击菜单中的"立即爬取"
+• 发送"帮助"查看更多功能"""
+
+            return self._create_text_response(response, from_user)
+
+        except Exception as e:
+            logger.error(f"关键词搜索处理异常: {str(e)}")
+            return self._create_text_response("搜索失败，请稍后重试", from_user)
+
+    def _create_text_response(self, content: str, to_user: str) -> str:
+        """创建文本消息响应"""
+        timestamp = str(int(time.time()))
+
+        response_xml = f"""<xml>
+<ToUserName><![CDATA[{to_user}]]></ToUserName>
+<FromUserName><![CDATA[{self.config.wechat_app.corp_id}]]></FromUserName>
+<CreateTime>{timestamp}</CreateTime>
+<MsgType><![CDATA[text]]></MsgType>
+<Content><![CDATA[{content}]]></Content>
+</xml>"""
+
+        return response_xml