升级Wagtail至v4后如何重新填充后台Image Usage引用？

Wagtail 4.x新增的Image Usage功能默认只会追踪升级后新增的图片引用，升级前的历史引用不会自动统计，需要手动触发扫描来填充数据。以下是具体解决方法：

方法：自定义Django管理命令重新扫描所有图片引用 #

1. 创建管理命令文件
   在你的Django项目任意app下，创建management/commands/refresh_image_usage.py文件，写入以下代码：

from django.core.management.base import BaseCommand
from wagtail.images.models import Image, ImageUsage
from wagtail.models import Page
from wagtail.fields import RichTextField
import re

class Command(BaseCommand):
    help = 'Refresh image usage statistics for all existing pages and models'

    def handle(self, *args, **options):
        # 清空现有记录（可选，避免重复统计）
        ImageUsage.objects.all().delete()

        # 处理所有Wagtail页面
        self.process_pages()

        # 处理自定义模型（如有非页面模型使用图片，自行扩展此方法）
        # self.process_custom_models()

        self.stdout.write(self.style.SUCCESS('Image usage statistics refreshed successfully'))

    def process_pages(self):
        # 遍历所有页面实例
        for page in Page.objects.all().specific():
            # 扫描页面中的ImageField字段
            for field_name, field in page.__class__.__dict__.items():
                if hasattr(field, 'related_model') and field.related_model == Image:
                    image = getattr(page, field_name, None)
                    if image:
                        ImageUsage.objects.get_or_create(image=image, content_object=page)

            # 扫描页面中RichTextField里嵌入的图片
            for field_name, field in page.__class__.__dict__.items():
                if isinstance(field, RichTextField):
                    html_content = getattr(page, field_name, '')
                    # 提取富文本中的图片ID
                    image_ids = re.findall(r'embed-(\d+)"', html_content)
                    for image_id in image_ids:
                        try:
                            image = Image.objects.get(id=image_id)
                            ImageUsage.objects.get_or_create(image=image, content_object=page)
                        except Image.DoesNotExist:
                            continue

    def process_custom_models(self):
        # 示例：替换为你的自定义模型和图片字段
        # from myapp.models import Product
        # for product in Product.objects.all():
        #     if product.featured_image:
        #         ImageUsage.objects.get_or_create(image=product.featured_image, content_object=product)
        pass

2. 运行命令
   在项目根目录下执行：

python manage.py refresh_image_usage

注意事项 #

• 执行前建议备份数据库，避免数据异常。
• 若站点包含大量页面或模型，可修改脚本加入分批处理逻辑，防止执行超时。
• 非页面模型的图片引用需自行扩展process_custom_models方法，适配你的业务模型。

内容的提问来源于stack exchange，提问作者user21889468