48 lines
1.3 KiB
Python
48 lines
1.3 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.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_bp.route("/become", methods=["GET", "POST"])
|
|
def become():
|
|
form = BecomePartnerForm(request.form)
|
|
|
|
if form.validate_on_submit():
|
|
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)
|