fix name collision

capsulflask/__init__.py

@@ -40,12 +40,12 @@ from capsulflask import db
 
 db.init_app(app)
 
-from capsulflask import auth, landing, console, stripe
+from capsulflask import auth, landing, console, payment_stripe
 
 app.register_blueprint(landing.bp)
 app.register_blueprint(auth.bp)
 app.register_blueprint(console.bp)
-app.register_blueprint(stripe.bp)
+app.register_blueprint(payment_stripe.bp)
 
 app.add_url_rule("/", endpoint="index")
 

capsulflask/payment_stripe.py

@@ -0,0 +1,116 @@
+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")
+