mirror of
https://github.com/simple-login/app.git
synced 2024-11-10 21:27:10 +01:00
372466ab06
- add __tablename__ for all models - use sa and orm instead of db - rollback all changes in tests - remove session in @app.teardown_appcontext
83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
from flask import render_template, flash, redirect, url_for, request
|
|
from flask_login import login_required, current_user
|
|
|
|
from app.config import PADDLE_MONTHLY_PRODUCT_ID, PADDLE_YEARLY_PRODUCT_ID
|
|
from app.dashboard.base import dashboard_bp
|
|
from app.db import Session
|
|
from app.log import LOG
|
|
from app.models import Subscription, PlanEnum
|
|
from app.paddle_utils import cancel_subscription, change_plan
|
|
|
|
|
|
@dashboard_bp.route("/billing", methods=["GET", "POST"])
|
|
@login_required
|
|
def billing():
|
|
# sanity check: make sure this page is only for user who has paddle subscription
|
|
sub: Subscription = current_user.get_subscription()
|
|
|
|
if not sub:
|
|
flash("You don't have any active subscription", "warning")
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
if request.method == "POST":
|
|
if request.form.get("form-name") == "cancel":
|
|
LOG.w(f"User {current_user} cancels their subscription")
|
|
success = cancel_subscription(sub.subscription_id)
|
|
|
|
if success:
|
|
sub.cancelled = True
|
|
Session.commit()
|
|
flash("Your subscription has been canceled successfully", "success")
|
|
else:
|
|
flash(
|
|
"Something went wrong, sorry for the inconvenience. Please retry. "
|
|
"We are already notified and will be on it asap",
|
|
"error",
|
|
)
|
|
|
|
return redirect(url_for("dashboard.billing"))
|
|
elif request.form.get("form-name") == "change-monthly":
|
|
LOG.d(f"User {current_user} changes to monthly plan")
|
|
success, msg = change_plan(
|
|
current_user, sub.subscription_id, PADDLE_MONTHLY_PRODUCT_ID
|
|
)
|
|
|
|
if success:
|
|
sub.plan = PlanEnum.monthly
|
|
Session.commit()
|
|
flash("Your subscription has been updated", "success")
|
|
else:
|
|
if msg:
|
|
flash(msg, "error")
|
|
else:
|
|
flash(
|
|
"Something went wrong, sorry for the inconvenience. Please retry. "
|
|
"We are already notified and will be on it asap",
|
|
"error",
|
|
)
|
|
|
|
return redirect(url_for("dashboard.billing"))
|
|
elif request.form.get("form-name") == "change-yearly":
|
|
LOG.d(f"User {current_user} changes to yearly plan")
|
|
success, msg = change_plan(
|
|
current_user, sub.subscription_id, PADDLE_YEARLY_PRODUCT_ID
|
|
)
|
|
|
|
if success:
|
|
sub.plan = PlanEnum.yearly
|
|
Session.commit()
|
|
flash("Your subscription has been updated", "success")
|
|
else:
|
|
if msg:
|
|
flash(msg, "error")
|
|
else:
|
|
flash(
|
|
"Something went wrong, sorry for the inconvenience. Please retry. "
|
|
"We are already notified and will be on it asap",
|
|
"error",
|
|
)
|
|
|
|
return redirect(url_for("dashboard.billing"))
|
|
|
|
return render_template("dashboard/billing.html", sub=sub, PlanEnum=PlanEnum)
|