diff --git a/uncloud/uncloud_pay/views.py b/uncloud/uncloud_pay/views.py index a6066b4..38d1aa4 100644 --- a/uncloud/uncloud_pay/views.py +++ b/uncloud/uncloud_pay/views.py @@ -10,6 +10,7 @@ import json from .models import * from .serializers import * from datetime import datetime +import uncloud_pay.stripe as uncloud_stripe ### # Standard user views: @@ -79,15 +80,15 @@ class PaymentMethodViewSet(viewsets.ModelViewSet): serializer.is_valid(raise_exception=True) # Retrieve Stripe customer ID for user. - customer_id = stripe.get_customer_id_for(request.user) + customer_id = uncloud_stripe.get_customer_id_for(request.user) if customer_id == None: return Response( {'error': 'Could not resolve customer stripe ID.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) # Register card under stripe customer. - credit_card = stripe.CreditCard(**serializer.validated_data.pop('credit_card')) - card_request = stripe.create_card(customer_id, credit_card) + credit_card = uncloud_stripe.CreditCard(**serializer.validated_data.pop('credit_card')) + card_request = uncloud_stripe.create_card(customer_id, credit_card) if card_request['error']: return Response({'stripe_error': card_request['error']}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) card_id = card_request['response_object']['id']