uncloud/uncloud_pay/stripe.py

115 lines
3.8 KiB
Python
Raw Normal View History

import stripe
2020-03-02 21:26:40 +00:00
import stripe.error
import logging
from django.core.exceptions import ObjectDoesNotExist
from django.conf import settings
import uncloud_pay.models
2020-03-02 21:26:40 +00:00
# Static stripe configuration used below.
CURRENCY = 'chf'
# README: We use the Payment Intent API as described on
# https://stripe.com/docs/payments/save-and-reuse
# For internal use only.
stripe.api_key = settings.STRIPE_KEY
2020-03-02 21:26:40 +00:00
# Helper (decorator) used to catch errors raised by stripe logic.
2020-03-05 10:03:47 +00:00
# Catch errors that should not be displayed to the end user, raise again.
def handle_stripe_error(f):
def handle_problems(*args, **kwargs):
response = {
'paid': False,
'response_object': None,
'error': None
}
2020-03-05 10:03:47 +00:00
common_message = "Currently it is not possible to make payments. Please try agin later."
try:
response_object = f(*args, **kwargs)
2020-03-05 10:03:47 +00:00
return response_object
except stripe.error.CardError as e:
# Since it's a decline, stripe.error.CardError will be caught
body = e.json_body
logging.error(str(e))
2020-03-05 10:03:47 +00:00
raise e # For error handling.
except stripe.error.RateLimitError:
2020-03-05 10:03:47 +00:00
logging.error("Too many requests made to the API too quickly.")
raise Exception(common_message)
except stripe.error.InvalidRequestError as e:
logging.error(str(e))
2020-03-05 10:03:47 +00:00
raise Exception('Invalid parameters.')
except stripe.error.AuthenticationError as e:
# Authentication with Stripe's API failed
# (maybe you changed API keys recently)
logging.error(str(e))
2020-03-05 10:03:47 +00:00
raise Exception(common_message)
except stripe.error.APIConnectionError as e:
logging.error(str(e))
2020-03-05 10:03:47 +00:00
raise Exception(common_message)
except stripe.error.StripeError as e:
2020-03-05 10:03:47 +00:00
# XXX: maybe send email
logging.error(str(e))
2020-03-05 10:03:47 +00:00
raise Exception(common_message)
except Exception as e:
# maybe send email
logging.error(str(e))
2020-03-05 10:03:47 +00:00
raise Exception(common_message)
return handle_problems
2020-03-02 21:26:40 +00:00
# Actual Stripe logic.
def public_api_key():
return settings.STRIPE_PUBLIC_KEY
def get_customer_id_for(user):
try:
# .get() raise if there is no matching entry.
return uncloud_pay.models.StripeCustomer.objects.get(owner=user).stripe_id
except ObjectDoesNotExist:
# No entry yet - making a new one.
2020-03-05 10:03:47 +00:00
try:
customer = create_customer(user.username, user.email)
uncloud_stripe_mapping = uncloud_pay.models.StripeCustomer.objects.create(
owner=user, stripe_id=customer.id)
2020-03-05 10:03:47 +00:00
return uncloud_stripe_mapping.stripe_id
except Exception as e:
return None
2020-03-02 21:26:40 +00:00
@handle_stripe_error
def create_setup_intent(customer_id):
return stripe.SetupIntent.create(customer=customer_id)
@handle_stripe_error
def get_setup_intent(setup_intent_id):
return stripe.SetupIntent.retrieve(setup_intent_id)
def get_payment_method(payment_method_id):
return stripe.PaymentMethod.retrieve(payment_method_id)
2020-03-02 21:26:40 +00:00
@handle_stripe_error
2020-03-03 17:16:25 +00:00
def charge_customer(amount, customer_id, card_id):
# Amount is in CHF but stripes requires smallest possible unit.
# https://stripe.com/docs/api/payment_intents/create#create_payment_intent-amount
2020-03-03 17:16:25 +00:00
adjusted_amount = int(amount * 100)
return stripe.PaymentIntent.create(
amount=adjusted_amount,
currency=CURRENCY,
customer=customer_id,
payment_method=card_id,
off_session=True,
confirm=True,
)
2020-03-02 21:26:40 +00:00
@handle_stripe_error
def create_customer(name, email):
return stripe.Customer.create(name=name, email=email)
@handle_stripe_error
def get_customer(customer_id):
return stripe.Customer.retrieve(customer_id)