91 lines
3.4 KiB
Python
91 lines
3.4 KiB
Python
import stripe
|
|
from flask import render_template, request, flash, redirect, url_for
|
|
from flask_login import login_required, current_user
|
|
from stripe.error import CardError
|
|
|
|
from app.config import STRIPE_API, STRIPE_MONTHLY_PLAN, STRIPE_YEARLY_PLAN
|
|
from app.dashboard.base import dashboard_bp
|
|
from app.email_utils import notify_admin
|
|
from app.extensions import db
|
|
from app.log import LOG
|
|
from app.models import PlanEnum
|
|
|
|
|
|
@dashboard_bp.route("/pricing", methods=["GET", "POST"])
|
|
@login_required
|
|
def pricing():
|
|
# sanity check: make sure this page is only for free user that has never subscribed before
|
|
# case user unsubscribe and re-subscribe will be handled later
|
|
if current_user.is_premium():
|
|
flash("You are already a premium user", "warning")
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
if (
|
|
current_user.stripe_customer_id
|
|
or current_user.stripe_card_token
|
|
or current_user.stripe_subscription_id
|
|
):
|
|
raise Exception("only user not exist on stripe can view this page")
|
|
|
|
if stripe.Customer.list(email=current_user.email):
|
|
raise Exception("user email is already used on stripe!")
|
|
|
|
if request.method == "POST":
|
|
plan_str = request.form.get("plan") # either monthly or yearly
|
|
if plan_str == "monthly":
|
|
plan = PlanEnum.monthly
|
|
elif plan_str == "yearly":
|
|
plan = PlanEnum.yearly
|
|
else:
|
|
raise Exception("Plan must be either yearly or monthly")
|
|
|
|
stripe_token = request.form.get("stripeToken")
|
|
LOG.d("stripe card token %s for plan %s", stripe_token, plan)
|
|
current_user.stripe_card_token = stripe_token
|
|
|
|
try:
|
|
customer = stripe.Customer.create(
|
|
source=stripe_token,
|
|
email=current_user.email,
|
|
metadata={"id": current_user.id},
|
|
name=current_user.name,
|
|
)
|
|
except CardError as e:
|
|
LOG.exception("payment problem, code:%s", e.code)
|
|
flash(
|
|
"Payment refused with error {e.message}. Could you re-try with another card please?",
|
|
"danger",
|
|
)
|
|
else:
|
|
LOG.d("stripe customer %s", customer)
|
|
current_user.stripe_customer_id = customer.id
|
|
|
|
stripe_plan = (
|
|
STRIPE_MONTHLY_PLAN if plan == PlanEnum.monthly else STRIPE_YEARLY_PLAN
|
|
)
|
|
subscription = stripe.Subscription.create(
|
|
customer=current_user.stripe_customer_id,
|
|
items=[{"plan": stripe_plan}],
|
|
expand=["latest_invoice.payment_intent"],
|
|
)
|
|
|
|
LOG.d("stripe subscription %s", subscription)
|
|
|
|
current_user.stripe_subscription_id = subscription.id
|
|
|
|
db.session.commit()
|
|
|
|
if subscription.latest_invoice.payment_intent.status == "succeeded":
|
|
LOG.d("payment successful for user %s", current_user)
|
|
current_user.plan = plan
|
|
current_user.plan_expiration = None
|
|
db.session.commit()
|
|
flash("Thanks for your subscription!", "success")
|
|
notify_admin(
|
|
f"user {current_user.email} has finished subscription",
|
|
f"plan: {plan}",
|
|
)
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
return render_template("dashboard/pricing.html", stripe_api=STRIPE_API)
|