Import coreapi from compat

This commit is contained in:
Tom Christie 2016-06-09 14:04:58 +01:00
parent 80c595edba
commit 47c776596c

View File

@ -18,13 +18,13 @@ from __future__ import unicode_literals
import itertools import itertools
from collections import OrderedDict, namedtuple from collections import OrderedDict, namedtuple
import coreapi
import uritemplate import uritemplate
from django.conf.urls import url from django.conf.urls import url
from django.core.exceptions import ImproperlyConfigured from django.core.exceptions import ImproperlyConfigured
from django.core.urlresolvers import NoReverseMatch from django.core.urlresolvers import NoReverseMatch
from rest_framework import renderers, views from rest_framework import renderers, views
from rest_framework.compat import coreapi
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.reverse import reverse from rest_framework.reverse import reverse
from rest_framework.settings import api_settings from rest_framework.settings import api_settings
@ -335,6 +335,7 @@ class DefaultRouter(SimpleRouter):
view_renderers = api_settings.DEFAULT_RENDERER_CLASSES view_renderers = api_settings.DEFAULT_RENDERER_CLASSES
if self.schema_title: if self.schema_title:
assert coreapi is not None, '`coreapi` must be installed for schema support.'
content = self.get_links() content = self.get_links()
schema = coreapi.Document(title=self.schema_title, content=content) schema = coreapi.Document(title=self.schema_title, content=content)
view_renderers += [renderers.CoreJSONRenderer] view_renderers += [renderers.CoreJSONRenderer]