Move django-based uncloud to top-level

uncloud_pay/stripe.py

@@ -0,0 +1,114 @@
+import stripe
+import stripe.error
+import logging
+
+from django.core.exceptions import ObjectDoesNotExist
+from django.conf import settings
+
+import uncloud_pay.models
+
+# 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
+
+# Helper (decorator) used to catch errors raised by stripe logic.
+# 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
+        }
+
+        common_message = "Currently it is not possible to make payments. Please try agin later."
+        try:
+            response_object = f(*args, **kwargs)
+            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))
+
+            raise e # For error handling.
+        except stripe.error.RateLimitError:
+            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))
+            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))
+            raise Exception(common_message)
+        except stripe.error.APIConnectionError as e:
+            logging.error(str(e))
+            raise Exception(common_message)
+        except stripe.error.StripeError as e:
+            # XXX: maybe send email
+            logging.error(str(e))
+            raise Exception(common_message)
+        except Exception as e:
+            # maybe send email
+            logging.error(str(e))
+            raise Exception(common_message)
+
+    return handle_problems
+
+# 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.
+        try:
+            customer = create_customer(user.username, user.email)
+            uncloud_stripe_mapping = uncloud_pay.models.StripeCustomer.objects.create(
+                    owner=user, stripe_id=customer.id)
+            return uncloud_stripe_mapping.stripe_id
+        except Exception as e:
+            return None
+
+@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)
+
+@handle_stripe_error
+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
+    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,
+    )
+
+@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)