diff --git a/rest_framework_docs/__init__.py b/rest_framework_docs/__init__.py index 156d6f9..eead319 100644 --- a/rest_framework_docs/__init__.py +++ b/rest_framework_docs/__init__.py @@ -1 +1 @@ -__version__ = '0.0.4' +__version__ = '0.0.5' diff --git a/rest_framework_docs/api_docs.py b/rest_framework_docs/api_docs.py index 844a37d..6e9028a 100644 --- a/rest_framework_docs/api_docs.py +++ b/rest_framework_docs/api_docs.py @@ -9,7 +9,10 @@ class ApiDocumentation(object): def __init__(self): self.endpoints = [] root_urlconf = __import__(settings.ROOT_URLCONF) - self.get_all_view_names(root_urlconf.urls.urlpatterns) + if hasattr(root_urlconf, 'urls'): + self.get_all_view_names(root_urlconf.urls.urlpatterns) + else: + self.get_all_view_names(root_urlconf.urlpatterns) def get_all_view_names(self, urlpatterns, parent_pattern=None): for pattern in urlpatterns: