liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

处理图片下载流程

Browse Source
This commit is contained in:
liuwei
2025-04-28 13:19:53 +08:00
parent 426ef878f8
commit 9e39f7f0b8
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
robot.py
View File

@@ -227,7 +227,7 @@ class Robot(Job):
# # 聊天记录入库动作： # # 聊天记录入库动作：
try: try:
self.message_storage.archive_message(msg) self.message_storage.archive_message(msg)
# 单独处理图片消息 # 单独处理图片消息 后续写定时任务自动完成下载。延时处理。
if msg.msg_type == MessageType.IMAGE: # 图片消息类型 if msg.msg_type == MessageType.IMAGE: # 图片消息类型
self.message_storage.process_image(msg) self.message_storage.process_image(msg)
except Exception as e: except Exception as e:

5
utils/wechat/message_to_db.py
View File

@@ -1,3 +1,4 @@
import time
from datetime import datetime, timedelta from datetime import datetime, timedelta
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
import logging import logging
@@ -121,7 +122,9 @@ class MessageStorage:
# 确保目标目录存在 # 确保目标目录存在
if not os.path.exists(target_dir): if not os.path.exists(target_dir):
os.makedirs(target_dir, exist_ok=True) os.makedirs(target_dir, exist_ok=True)
# 延时10秒下载试试
logger.info(f"延时10秒下载图片{msg.content.xml_content}")
time.sleep(10)
# 尝试使用wcf下载图片到分组后的目录 # 尝试使用wcf下载图片到分组后的目录
json = self.client.download_image(msg.appid, msg.content.xml_content, 1) json = self.client.download_image(msg.appid, msg.content.xml_content, 1)
# { # {