uncloud-mravi/uncloud/uncloud_pay/stripe.py

import stripe
import stripe.error
import logging

from django.conf import settings

# Static stripe configuration used below.
CURRENCY = 'chf'

# Register stripe (secret) API key from config.
stripe.api_key = settings.STRIPE_API_KEY

# Helper (decorator) used to catch errors raised by stripe logic.
def handle_stripe_error(f):
    def handle_problems(*args, **kwargs):
        response = {
            'paid': False,
            'response_object': None,
            'error': None
        }

        common_message = "Currently it's not possible to make payments."
        try:
            response_object = f(*args, **kwargs)
            response = {
                'response_object': response_object,
                'error': None
            }
            return response
        except stripe.error.CardError as e:
            # Since it's a decline, stripe.error.CardError will be caught
            body = e.json_body
            err = body['error']
            response.update({'error': err['message']})
            logging.error(str(e))
            return response
        except stripe.error.RateLimitError:
            response.update(
                {'error': "Too many requests made to the API too quickly"})
            return response
        except stripe.error.InvalidRequestError as e:
            logging.error(str(e))
            response.update({'error': "Invalid parameters"})
            return response
        except stripe.error.AuthenticationError as e:
            # Authentication with Stripe's API failed
            # (maybe you changed API keys recently)
            logging.error(str(e))
            response.update({'error': common_message})
            return response
        except stripe.error.APIConnectionError as e:
            logging.error(str(e))
            response.update({'error': common_message})
            return response
        except stripe.error.StripeError as e:
            # maybe send email
            logging.error(str(e))
            response.update({'error': common_message})
            return response
        except Exception as e:
            # maybe send email
            logging.error(str(e))
            response.update({'error': common_message})
            return response

    return handle_problems

# Convenience CC container, also used for serialization.
class CreditCard():
    number = None
    exp_year = None
    exp_month = None
    cvc = None

    def __init__(self, number, exp_month, exp_year, cvc):
        self.number=number
        self.exp_year = exp_year
        self.exp_month = exp_month
        self.cvc = cvc

# Actual Stripe logic.

@handle_stripe_error
def create_card(customer_id, credit_card):
    # Test settings
    credit_card.number = "5555555555554444"

    return stripe.Customer.create_source(
            customer_id,
            card={
                'number': credit_card.number,
                'exp_month': credit_card.exp_month,
                'exp_year': credit_card.exp_year,
                'cvc': credit_card.cvc
                })

@handle_stripe_error
def get_card(customer_id, card_id):
    return stripe.Card.retrieve_source(customer_id, card_id)

@handle_stripe_error
def charge_customer(amount, source):
    return stripe.Charge.create(
            amount=amount,
            currenty=CURRENCY,
            source=source)

@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)