Expose Image.collection in the API

Author: laymonage

bakerydemo/base/models.py

@@ -21,6 +21,7 @@
     register_setting,
 )
 from wagtail.fields import RichTextField, StreamField
+from wagtail.images.models import Image
 from wagtail.models import (
     Collection,
     DraftStateMixin,
@@ -37,6 +38,9 @@
 
 from .blocks import BaseStreamBlock
 
+# Allow filtering by collection
+Image.api_fields = [APIField("collection")]
+
 
 class Person(
     WorkflowMixin,