From c6920795de6bff9bfdcdd70dec0080856b5a0272 Mon Sep 17 00:00:00 2001 From: ambivalentno Date: Tue, 22 Mar 2016 15:23:26 +0200 Subject: [PATCH] Update api_docs.py --- rest_framework_docs/api_docs.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/rest_framework_docs/api_docs.py b/rest_framework_docs/api_docs.py index 01e913d..c51c205 100644 --- a/rest_framework_docs/api_docs.py +++ b/rest_framework_docs/api_docs.py @@ -1,3 +1,4 @@ +from importlib import import_module from django.conf import settings from django.core.urlresolvers import RegexURLResolver, RegexURLPattern from django.utils.module_loading import import_string @@ -9,7 +10,11 @@ class ApiDocumentation(object): def __init__(self): self.endpoints = [] - root_urlconf = import_string(settings.ROOT_URLCONF) + try: + root_urlconf = import_string(settings.ROOT_URLCONF) + except ImportError: + # Handle a case when there's no dot in ROOT_URLCONF + root_urlconf = import_module(settings.ROOT_URLCONF) if hasattr(root_urlconf, 'urls'): self.get_all_view_names(root_urlconf.urls.urlpatterns) else: