diff --git a/src/ants_hub/urls.py b/src/ants_hub/urls.py index 80630d8..a01dcd6 100644 --- a/src/ants_hub/urls.py +++ b/src/ants_hub/urls.py @@ -52,5 +52,5 @@ urlpatterns = [ if 'mellon' in settings.INSTALLED_APPS: urlpatterns += [ - path('saml/', include('mellon.urls')), + path('saml/', urls_utils.decorate_admin(include('mellon.urls'))), ] diff --git a/src/ants_hub/urls_utils.py b/src/ants_hub/urls_utils.py index 7f127cf..db72b84 100644 --- a/src/ants_hub/urls_utils.py +++ b/src/ants_hub/urls_utils.py @@ -27,7 +27,7 @@ class DecoratedURLResolver(URLResolver): def decorated_includes(func, includes, *args, **kwargs): urlconf_module, app_name, namespace = includes - for item in urlconf_module: + for item in getattr(urlconf_module, 'urlpatterns', urlconf_module): if isinstance(item, URLResolver): item.__class__ = DecoratedURLResolver else: