From 6015c850435208d6c0fe5dd7b02dad0abfaee4ab Mon Sep 17 00:00:00 2001
From: shengyudong <shengyudong@yunqueai.net>
Date: Tue, 3 Feb 2026 10:45:17 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E5=9B=BE=E7=89=87=E6=8C=82?=
 =?UTF-8?q?=E8=BD=BD=E9=80=BB=E8=BE=91=EF=BC=9A=E7=AD=9B=E9=80=89generate?=
 =?UTF-8?q?=E7=8A=B6=E6=80=81=E5=9B=BE=E7=89=87=EF=BC=8C=E6=8C=82=E8=BD=BD?=
 =?UTF-8?q?=E6=88=90=E5=8A=9F=E5=90=8E=E6=9B=B4=E6=96=B0=E6=AC=A1=E6=95=B0?=
 =?UTF-8?q?=E5=92=8C=E7=8A=B6=E6=80=81=E4=B8=BApublished?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 article_auto_image_matching.py | 60 ++++++++++++++++++++++++----------
 1 file changed, 43 insertions(+), 17 deletions(-)

diff --git a/article_auto_image_matching.py b/article_auto_image_matching.py
index 4794a8f..731cbee 100644
--- a/article_auto_image_matching.py
+++ b/article_auto_image_matching.py
@@ -128,7 +128,7 @@ class ArticleImageMatcher:
     
     def get_available_images_with_tags(self) -> List[Dict]:
         """
-        从ai_image_tags表获取可用的图片及其标签
+        从ai_image_tags表获取可用的图片及其标签（状态为generate且挂载次数<5）
         
         Returns:
             包含图片ID、标签等信息的列表
@@ -137,24 +137,26 @@ class ArticleImageMatcher:
             connection = self.db_manager.get_connection()
             try:
                 with connection.cursor(pymysql.cursors.DictCursor) as cursor:
-                    # 查询附加文章数量小于5的图片
+                    # 查询状态为generate且附加文章数量小于5的图片
                     sql = """
                     SELECT 
-                        id,
-                        image_id,
-                        image_name,
-                        image_url,
-                        image_thumb_url,
-                        tag_id,
-                        tag_name,
-                        keywords_id,
-                        keywords_name,
-                        department_id,
-                        department_name,
-                        image_attached_article_count
-                    FROM ai_image_tags
-                    WHERE image_attached_article_count < 5
-                    ORDER BY image_attached_article_count ASC, id DESC
+                        it.id,
+                        it.image_id,
+                        it.image_name,
+                        it.image_url,
+                        it.image_thumb_url,
+                        it.tag_id,
+                        it.tag_name,
+                        it.keywords_id,
+                        it.keywords_name,
+                        it.department_id,
+                        it.department_name,
+                        it.image_attached_article_count
+                    FROM ai_image_tags it
+                    INNER JOIN ai_images i ON it.image_id = i.id
+                    WHERE it.image_attached_article_count < 5
+                    AND i.status = 'generate'
+                    ORDER BY it.image_attached_article_count ASC, it.id DESC
                     """
                     cursor.execute(sql)
                     results = cursor.fetchall()
@@ -352,6 +354,14 @@ class ArticleImageMatcher:
                     """
                     cursor.execute(update_sql, (image_data['id'],))
                     
+                    # 更新图片状态为published
+                    update_image_status_sql = """
+                    UPDATE ai_images
+                    SET status = 'published'
+                    WHERE id = %s
+                    """
+                    cursor.execute(update_image_status_sql, (image_data['image_id'],))
+                    
                     connection.commit()
                     logger.info(f"成功插入文章图片关联 - 文章ID: {article_id}, 图片ID: {image_data['image_id']}, 分数: {match_score}")
                     return True
@@ -652,6 +662,22 @@ class ArticleImageMatcher:
                     article_image_id = cursor.lastrowid
                     logger.info(f"文章图片关联信息已插入ai_article_images表，id: {article_image_id}")
                     
+                    # 更新图片附加文章计数
+                    update_count_sql = """
+                    UPDATE ai_image_tags 
+                    SET image_attached_article_count = image_attached_article_count + 1
+                    WHERE id = %s
+                    """
+                    cursor.execute(update_count_sql, (tag_image_id,))
+                    
+                    # 更新图片状态为published
+                    update_image_status_sql = """
+                    UPDATE ai_images
+                    SET status = 'published'
+                    WHERE id = %s
+                    """
+                    cursor.execute(update_image_status_sql, (image_id,))
+                    
                     connection.commit()
                     return article_image_id
             finally: