from django.apps import AppConfig from django.db.models import ForeignKey class WagtailMediaAppConfig(AppConfig): default_auto_field = "django.db.models.AutoField" name = "wagtailmedia" label = "wagtailmedia" verbose_name = "Wagtail media" def ready(self): from wagtail.admin.compare import register_comparison_class from .edit_handlers import MediaFieldComparison from .models import get_media_model from .signal_handlers import register_signal_handlers register_signal_handlers() # Set up image ForeignKeys to use ImageFieldComparison as the comparison class # when comparing page revisions register_comparison_class( ForeignKey, to=get_media_model(), comparison_class=MediaFieldComparison )