mirror of
https://github.com/simple-login/app.git
synced 2024-11-13 23:51:13 +01:00
121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
import arrow
|
|
from flask import render_template, flash, redirect, url_for, request
|
|
from flask_login import login_required, current_user
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import StringField, validators
|
|
|
|
from app import parallel_limiter
|
|
from app.config import PADDLE_VENDOR_ID, PADDLE_COUPON_ID
|
|
from app.dashboard.base import dashboard_bp
|
|
from app.db import Session
|
|
from app.log import LOG
|
|
from app.models import (
|
|
ManualSubscription,
|
|
Coupon,
|
|
Subscription,
|
|
AppleSubscription,
|
|
CoinbaseSubscription,
|
|
LifetimeCoupon,
|
|
)
|
|
|
|
|
|
class CouponForm(FlaskForm):
|
|
code = StringField("Coupon Code", validators=[validators.DataRequired()])
|
|
|
|
|
|
@dashboard_bp.route("/coupon", methods=["GET", "POST"])
|
|
@login_required
|
|
@parallel_limiter.lock()
|
|
def coupon_route():
|
|
coupon_form = CouponForm()
|
|
|
|
if coupon_form.validate_on_submit():
|
|
code = coupon_form.code.data
|
|
if LifetimeCoupon.get_by(code=code):
|
|
LOG.d("redirect %s to lifetime page instead", current_user)
|
|
flash("Redirect to the lifetime coupon page instead", "success")
|
|
return redirect(url_for("dashboard.lifetime_licence"))
|
|
|
|
# handle case user already has an active subscription via another channel (Paddle, Apple, etc)
|
|
can_use_coupon = True
|
|
|
|
if current_user.lifetime:
|
|
can_use_coupon = False
|
|
|
|
sub: Subscription = current_user.get_paddle_subscription()
|
|
if sub:
|
|
can_use_coupon = False
|
|
|
|
apple_sub: AppleSubscription = AppleSubscription.get_by(user_id=current_user.id)
|
|
if apple_sub and apple_sub.is_valid():
|
|
can_use_coupon = False
|
|
|
|
coinbase_subscription: CoinbaseSubscription = CoinbaseSubscription.get_by(
|
|
user_id=current_user.id
|
|
)
|
|
if coinbase_subscription and coinbase_subscription.is_active():
|
|
can_use_coupon = False
|
|
|
|
if coupon_form.validate_on_submit():
|
|
code = coupon_form.code.data
|
|
|
|
coupon: Coupon = Coupon.get_by(code=code)
|
|
if coupon and not coupon.used:
|
|
if coupon.expires_date and coupon.expires_date < arrow.now():
|
|
flash(
|
|
f"The coupon was expired on {coupon.expires_date.humanize()}",
|
|
"error",
|
|
)
|
|
return redirect(request.url)
|
|
|
|
updated = (
|
|
Session.query(Coupon)
|
|
.filter_by(code=code, used=False)
|
|
.update({"used_by_user_id": current_user.id, "used": True})
|
|
)
|
|
if updated != 1:
|
|
flash("Coupon is not valid", "error")
|
|
return redirect(request.url)
|
|
|
|
manual_sub: ManualSubscription = ManualSubscription.get_by(
|
|
user_id=current_user.id
|
|
)
|
|
if manual_sub:
|
|
# renew existing subscription
|
|
if manual_sub.end_at > arrow.now():
|
|
manual_sub.end_at = manual_sub.end_at.shift(years=coupon.nb_year)
|
|
else:
|
|
manual_sub.end_at = arrow.now().shift(years=coupon.nb_year, days=1)
|
|
Session.commit()
|
|
flash(
|
|
f"Your current subscription is extended to {manual_sub.end_at.humanize()}",
|
|
"success",
|
|
)
|
|
else:
|
|
ManualSubscription.create(
|
|
user_id=current_user.id,
|
|
end_at=arrow.now().shift(years=coupon.nb_year, days=1),
|
|
comment="using coupon code",
|
|
is_giveaway=coupon.is_giveaway,
|
|
commit=True,
|
|
)
|
|
flash(
|
|
"Your account has been upgraded to Premium, thanks for your support!",
|
|
"success",
|
|
)
|
|
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
else:
|
|
flash(f"Code *{code}* expired or invalid", "warning")
|
|
|
|
return render_template(
|
|
"dashboard/coupon.html",
|
|
coupon_form=coupon_form,
|
|
PADDLE_VENDOR_ID=PADDLE_VENDOR_ID,
|
|
PADDLE_COUPON_ID=PADDLE_COUPON_ID,
|
|
can_use_coupon=can_use_coupon,
|
|
# a coupon is only valid until this date
|
|
# this is to avoid using the coupon to renew an account forever
|
|
max_coupon_date=arrow.now().shift(years=1, days=-1),
|
|
)
|