mirror of
https://github.com/simple-login/app.git
synced 2024-09-04 01:21:30 +02:00
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
from flask import render_template, flash, redirect, url_for
|
|
from flask_login import login_required, current_user
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import StringField, validators
|
|
|
|
from app.config import (
|
|
PADDLE_VENDOR_ID,
|
|
PADDLE_MONTHLY_PRODUCT_ID,
|
|
PADDLE_YEARLY_PRODUCT_ID,
|
|
URL,
|
|
ADMIN_EMAIL,
|
|
)
|
|
from app.dashboard.base import dashboard_bp
|
|
from app.email_utils import send_email
|
|
from app.extensions import db
|
|
from app.models import LifetimeCoupon
|
|
|
|
|
|
class CouponForm(FlaskForm):
|
|
code = StringField("Coupon Code", validators=[validators.DataRequired()])
|
|
|
|
|
|
@dashboard_bp.route("/pricing", methods=["GET", "POST"])
|
|
@login_required
|
|
def pricing():
|
|
# sanity check: make sure this page is only for free user
|
|
if current_user.is_premium():
|
|
flash("You are already a premium user", "warning")
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
coupon_form = CouponForm()
|
|
|
|
if coupon_form.validate_on_submit():
|
|
code = coupon_form.code.data
|
|
|
|
coupon = LifetimeCoupon.get_by(code=code)
|
|
|
|
if coupon and coupon.nb_used > 0:
|
|
coupon.nb_used -= 1
|
|
current_user.lifetime = True
|
|
db.session.commit()
|
|
|
|
# notify admin
|
|
send_email(
|
|
ADMIN_EMAIL,
|
|
subject=f"User {current_user.id} used lifetime coupon. Coupon nb_used: {coupon.nb_used}",
|
|
plaintext="",
|
|
html="",
|
|
)
|
|
|
|
flash("You are upgraded to lifetime premium!", "success")
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
else:
|
|
flash(f"Coupon *{code}* expired or invalid", "warning")
|
|
|
|
return render_template(
|
|
"dashboard/pricing.html",
|
|
PADDLE_VENDOR_ID=PADDLE_VENDOR_ID,
|
|
PADDLE_MONTHLY_PRODUCT_ID=PADDLE_MONTHLY_PRODUCT_ID,
|
|
PADDLE_YEARLY_PRODUCT_ID=PADDLE_YEARLY_PRODUCT_ID,
|
|
success_url=URL + "/dashboard/subscription_success",
|
|
coupon_form=coupon_form,
|
|
)
|
|
|
|
|
|
@dashboard_bp.route("/subscription_success")
|
|
@login_required
|
|
def subscription_success():
|
|
flash("Thanks so much for supporting SimpleLogin!", "success")
|
|
return redirect(url_for("dashboard.index"))
|