capsul-flask/capsulflask/stripe.py

import stripe
import json
import decimal


from flask import Blueprint
from flask import request
from flask import current_app
from flask import session
from flask import redirect
from flask import url_for
from flask import jsonify
from flask import flash
from flask import render_template
from werkzeug.exceptions import abort

from capsulflask.auth import account_required

from capsulflask.db import get_model

bp = Blueprint("stripe", __name__, url_prefix="/stripe")

@bp.route("/", methods=("GET", "POST"))
@account_required
def index():

  stripe_checkout_session_id=None

  if request.method == "POST":
    errors = list()
    if "dollars" not in request.form:
      errors.append("dollars is required")
    elif decimal.Decimal(request.form["dollars"]) < decimal.Decimal(1):
      errors.append("dollars must be >= 1")

    if len(errors) == 0:

      print(f"creating stripe checkout session for {session['account']}, ${request.form['dollars']}")

      checkout_session = stripe.checkout.Session.create(
        success_url=current_app.config['BASE_URL'] + "/stripe/success?session_id={CHECKOUT_SESSION_ID}",
        cancel_url=current_app.config['BASE_URL'] + "/stripe",
        payment_method_types=["card"],
        customer_email=session["account"],
        line_items=[
          {
            "name": "Capsul Cloud Compute",
            "images": [current_app.config['BASE_URL']+"/static/capsul-product-image.png"],
            "quantity": 1,
            "currency": "usd",
            "amount": request.form["dollars"]
          }
        ]
      )
      stripe_checkout_session_id = checkout_session['id']

      print(f"stripe_checkout_session_id={stripe_checkout_session_id} ( {session['account']}, ${request.form['dollars']} )")

      get_model().create_stripe_checkout_session(stripe_checkout_session_id, session["account"], request.form["dollars"])

  for error in errors:
    flash(error)

  return render_template(
    "stripe.html", 
    stripe_checkout_session_id=stripe_checkout_session_id,
    stripe_public_key=current_app.config["STRIPE_PUBLISHABLE_KEY"]
  )

@bp.route("/success", methods=("GET",))
def success():
  stripe_checkout_session_id = request.args.get('session_id')
  if not stripe_checkout_session_id:
    print("/stripe/success returned 400: missing required URL parameter session_id")
    abort(400, "missing required URL parameter session_id")
  else:
    checkout_session = stripe.checkout.Session.retrieve(stripe_checkout_session_id)
    if checkout_session and 'display_items' in checkout_session:
      dollars = checkout_session['display_items'][0]['amount']

      #consume_stripe_checkout_session deletes the checkout session row and inserts a payment row
      # its ok to call consume_stripe_checkout_session more than once because it only takes an action if the session exists
      success_account = get_model().consume_stripe_checkout_session(stripe_checkout_session_id, dollars)
      if success_account:
        print(f"{success_account} paid ${dollars} successfully (stripe_checkout_session_id={stripe_checkout_session_id})")

    return redirect(url_for("console.account_balance"))

# I don't think the webhook is needed
# @bp.route("/webhook", methods=("POST",))
# def webhook():

#   request_data = json.loads(request.data)
#   signature = request.headers.get('stripe-signature')
#   try:
#     event = stripe.Webhook.construct_event(
#       payload=request_data, 
#       sig_header=signature, 
#       secret=current_app.config['STRIPE_WEBHOOK_SECRET']
#     )
#     if event['type'] == 'checkout.session.completed':
#       dollars = event['data']['object']['display_items'][0]['amount']
#       stripe_checkout_session_id = event['data']['object']['id']

#       #consume_stripe_checkout_session deletes the checkout session row and inserts a payment row
#       # its ok to call consume_stripe_checkout_session more than once because it only takes an action if the session exists
#       get_model().consume_stripe_checkout_session(stripe_checkout_session_id, dollars)
      
#     return jsonify({'status': 'success'})
#   except ValueError as e:
#     print("/stripe/webhook returned 400: bad request", e)
#     abort(400, "bad request")
#   except stripe.error.SignatureVerificationError:
#     print("/stripe/webhook returned 400: invalid signature")
#     abort(400, "invalid signature")