You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
1.0 KiB
29 lines
1.0 KiB
from flask import render_template, session, redirect, url_for, current_app
|
|
from .. import db
|
|
from ..models import User
|
|
from ..email import send_email
|
|
from . import main
|
|
from .forms import NameForm
|
|
|
|
|
|
@main.route('/', methods=['GET', 'POST'])
|
|
def index():
|
|
form = NameForm()
|
|
if form.validate_on_submit():
|
|
user = User.query.filter_by(username=form.name.data).first()
|
|
if user is None:
|
|
user = User(username=form.name.data)
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
session['known'] = False
|
|
if current_app.config['FLASKY_ADMIN']:
|
|
send_email(current_app.config['FLASKY_ADMIN'], 'New user',
|
|
'mail/new_user', user=user)
|
|
else:
|
|
session['known'] = True
|
|
session['name'] = form.name.data
|
|
form.name.data = ''
|
|
return redirect(url_for('.index'))
|
|
return render_template('index.html',
|
|
form=form, name=session.get('name'),
|
|
known=session.get('known', False))
|