Skip to content

Commit 0f89e44

Browse files
author
Emmanouil Konstantinidis
committed
Catch both cases of common ROOT_URLCONF
1 parent 86dd021 commit 0f89e44

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

rest_framework_docs/api_docs.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,10 @@ class ApiDocumentation(object):
99
def __init__(self):
1010
self.endpoints = []
1111
root_urlconf = __import__(settings.ROOT_URLCONF)
12-
self.get_all_view_names(root_urlconf.urls.urlpatterns)
12+
if hasattr(root_urlconf, 'urls'):
13+
self.get_all_view_names(root_urlconf.urls.urlpatterns)
14+
else:
15+
self.get_all_view_names(root_urlconf.urlpatterns)
1316

1417
def get_all_view_names(self, urlpatterns, parent_pattern=None):
1518
for pattern in urlpatterns:

0 commit comments

Comments
 (0)