diff --git a/setup.py b/setup.py index 73849d4880..bd8fb70fc8 100644 --- a/setup.py +++ b/setup.py @@ -28,6 +28,7 @@ install_requires = [ "django-modelcluster>=0.6", "django-taggit>=0.13.0", "django-treebeard==3.0", + "djangorestframework==3.1.3", "Pillow>=2.6.1", "beautifulsoup4>=4.3.2", "html5lib==0.999", diff --git a/wagtail/contrib/wagtailapi/endpoints.py b/wagtail/contrib/wagtailapi/endpoints.py index 318c2c3c47..c50d7eed69 100644 --- a/wagtail/contrib/wagtailapi/endpoints.py +++ b/wagtail/contrib/wagtailapi/endpoints.py @@ -10,6 +10,7 @@ from django.utils.encoding import force_text from django.shortcuts import get_object_or_404 from django.conf.urls import url from django.conf import settings +from rest_framework.viewsets import ViewSet from wagtail.wagtailcore.models import Page from wagtail.wagtailimages.models import get_image_model @@ -94,7 +95,7 @@ def get_api_data(obj, fields): continue -class BaseAPIEndpoint(object): +class BaseAPIEndpoint(ViewSet): known_query_parameters = frozenset([ 'limit', 'offset',