diff --git a/hc/payments/views.py b/hc/payments/views.py index 08f6475e..9316115e 100644 --- a/hc/payments/views.py +++ b/hc/payments/views.py @@ -1,4 +1,4 @@ -import braintree +from django.conf import settings from django.contrib import messages from django.contrib.auth.decorators import login_required from django.http import (HttpResponseBadRequest, HttpResponseForbidden, @@ -9,6 +9,13 @@ from hc.accounts.models import Profile from .models import Subscription +if settings.USE_PAYMENTS: + import braintree +else: + # hc.payments tests mock this object, so tests should + # still be able to run: + braintree = None + @login_required def get_client_token(request):