From ef691e58e24e644ef45db67a08d8165d34b7a899 Mon Sep 17 00:00:00 2001 From: liuwei Date: Tue, 1 Apr 2025 17:00:53 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E5=85=A5=E5=9B=BE=E7=89=87=E4=B8=8B?= =?UTF-8?q?=E8=BD=BD=E9=80=BB=E8=BE=91=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- message_storage/message_to_db.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/message_storage/message_to_db.py b/message_storage/message_to_db.py index bb17f2a..18de517 100644 --- a/message_storage/message_to_db.py +++ b/message_storage/message_to_db.py @@ -114,8 +114,14 @@ class MessageStorage: try: # 使用wcf下载图片，确保图片存在 if self.wcf and msg.id: - # 尝试使用wcf下载图片到我们的图片目录 - download_path = self.wcf.download_image(msg.id, msg.extra, self.image_dir) + # 创建按群ID或个人wxid分割的目录 + target_dir = os.path.join(self.image_dir, msg.roomid if msg.roomid else msg.sender) + # 确保目标目录存在 + if not os.path.exists(target_dir): + os.makedirs(target_dir, exist_ok=True) + + # 尝试使用wcf下载图片到分组后的目录 + download_path = self.wcf.download_image(msg.id, msg.extra, target_dir) if download_path: logger.info(f"使用wcf下载图片成功: {msg.id} -> {download_path}")