diff --git a/rest_invitations/urls.py b/rest_invitations/urls.py index 36585ce..545118f 100644 --- a/rest_invitations/urls.py +++ b/rest_invitations/urls.py @@ -2,7 +2,24 @@ from rest_framework import routers from .app_settings import ACCEPT_INVITE_URL, API_BASE_URL -from .views import InvitationViewSet, accept_invitation +from .views import (InvitationViewSet, accept_invitation) +from .utils import import_callable +from django.conf import settings + +InvitationViewSet = import_callable( + getattr( + settings, + 'INVITATION_VIEW', + InvitationViewSet + ) +) +accept_invitation = import_callable( + getattr( + settings, + 'INVITATION_ACCEPT_INVATION', + accept_invitation + ) +) router = routers.SimpleRouter() router.register(r'{0}'.format(API_BASE_URL), InvitationViewSet) diff --git a/rest_invitations/utils.py b/rest_invitations/utils.py index d5d38d5..7e24ba2 100644 --- a/rest_invitations/utils.py +++ b/rest_invitations/utils.py @@ -6,5 +6,4 @@ def import_callable(path): return path else: module, attr_name = path.rsplit('.', 1) - print(module, attr_name) return getattr(import_module(module), attr_name)