39 lines
1 KiB
Python
39 lines
1 KiB
Python
|
from wagtail.api.v2.filters import FieldsFilter, OrderingFilter, SearchFilter
|
||
|
|
from wagtail.api.v2.views import BaseAPIViewSet
|
||
|
|
|
||
|
|
from ..models import get_media_model
|
||
|
|
from .serializers import MediaItemSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class MediaAPIViewSet(BaseAPIViewSet):
|
||
|
|
base_serializer_class = MediaItemSerializer
|
||
|
|
filter_backends = [FieldsFilter, OrderingFilter, SearchFilter]
|
||
|
|
body_fields = BaseAPIViewSet.body_fields + [
|
||
|
|
"title",
|
||
|
|
"width",
|
||
|
|
"height",
|
||
|
|
"media_type",
|
||
|
|
"collection",
|
||
|
|
]
|
||
|
|
meta_fields = BaseAPIViewSet.meta_fields + [
|
||
|
|
"tags",
|
||
|
|
"download_url",
|
||
|
|
]
|
||
|
|
listing_default_fields = BaseAPIViewSet.listing_default_fields + [
|
||
|
|
"media_type",
|
||
|
|
"title",
|
||
|
|
"width",
|
||
|
|
"height",
|
||
|
|
"tags",
|
||
|
|
"collection",
|
||
|
|
"thumbnail",
|
||
|
|
"download_url",
|
||
|
|
]
|
||
|
|
nested_default_fields = BaseAPIViewSet.nested_default_fields + [
|
||
|
|
"title",
|
||
|
|
"collection",
|
||
|
|
"download_url",
|
||
|
|
]
|
||
|
|
name = "media"
|
||
|
|
model = get_media_model()
|