diff --git a/utils/wechat/message_to_db.py b/utils/wechat/message_to_db.py index 36dbf13..38ffc75 100644 --- a/utils/wechat/message_to_db.py +++ b/utils/wechat/message_to_db.py @@ -10,7 +10,7 @@ from db.connection import DBConnectionManager from db.message_storage import MessageStorageDB # 导入积分系统 from db.points_db import PointsDBOperator, PointSource -from gewechat.call_back_message.message import WxMessage +from gewechat.call_back_message.message import WxMessage, MessageType # 配置日志 logging.basicConfig( @@ -98,7 +98,7 @@ class MessageStorage: def process_image(self, msg: WxMessage): """异步处理图片消息,与消息存档分离""" - if msg.msg_type != 3 or not self.client: # 不是图片消息或没有client实例 + if msg.msg_type != MessageType.IMAGE or not self.client: # 不是图片消息或没有client实例 return False # 提交任务到图片处理线程池