From ef9d5b852d49d0963c9eb9f3863ff066fe494092 Mon Sep 17 00:00:00 2001
From: Tom Christie <tom@tomchristie.com>
Date: Mon, 20 Jul 2015 11:46:46 +0100
Subject: [PATCH] Use ViewSet as base class for API endpoints.

---
 setup.py                                | 1 +
 wagtail/contrib/wagtailapi/endpoints.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

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',