78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
from flask import request, render_template, redirect, url_for, flash
|
|
from flask_login import current_user
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import StringField
|
|
|
|
from app.config import PARTNER_CODES
|
|
from app.email_utils import notify_admin
|
|
from app.extensions import db
|
|
from app.models import Partner
|
|
from app.partner.base import partner_bp
|
|
|
|
|
|
class BecomePartnerForm(FlaskForm):
|
|
email = StringField("Email")
|
|
name = StringField("Name")
|
|
website = StringField("Website")
|
|
additional_information = StringField("Additional Information")
|
|
partner_code = StringField("Partner Code")
|
|
|
|
|
|
@partner_bp.route("/become", methods=["GET", "POST"])
|
|
def become():
|
|
form = BecomePartnerForm(request.form)
|
|
|
|
if form.validate_on_submit():
|
|
# bypass the application
|
|
if form.partner_code.data:
|
|
if not current_user.is_authenticated:
|
|
raise Exception("only authenticated user can enter partner code")
|
|
|
|
if form.partner_code.data in PARTNER_CODES:
|
|
notify_admin(
|
|
f"User {current_user.name} has become partner!",
|
|
{current_user.email},
|
|
)
|
|
|
|
current_user.is_developer = True
|
|
db.session.commit()
|
|
|
|
flash(
|
|
"Congratulations, you are now a SimpleLogin partner! "
|
|
"You will have access to tech resources on SimpleLogin.",
|
|
"success",
|
|
)
|
|
|
|
return redirect(url_for("developer.index"))
|
|
else:
|
|
error = (
|
|
"The partner code is unknown. Are you sure this is the right code?"
|
|
)
|
|
return render_template("partner/become.html", form=form, error=error)
|
|
else:
|
|
partner = Partner.create(
|
|
email=form.email.data,
|
|
name=form.name.data,
|
|
website=form.website.data,
|
|
additional_information=form.additional_information.data,
|
|
)
|
|
|
|
if current_user.is_authenticated:
|
|
partner.user_id = current_user.id
|
|
|
|
db.session.commit()
|
|
|
|
notify_admin(
|
|
f"New partner {partner.name} {partner.email} has signed up!",
|
|
partner.website,
|
|
)
|
|
|
|
flash(
|
|
"Your request has been submitted, we'll come back to you asap!",
|
|
"success",
|
|
)
|
|
|
|
return redirect(url_for("partner.become"))
|
|
|
|
return render_template("partner/become.html", form=form)
|