def signup():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
user = User.query.filter_by(username=username).first()
if user:
return redirect('/signup')
else:
User.add(username=username, password=password)
auth.login(username, password)
session['username'] = username
return redirect(session['last_page'])
return render_template('signup.html', title='Регистрация')
def registration():
has_error = False
form = UserCreateForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
user = User.query.filter_by(username=username).first()
if user:
has_error = True
else:
User.add(username=username, password=password)
auth.login(username, password)
return redirect('/')
return render_template('registration.html',
title='Зарегистрироваться',
form=form,
has_error=has_error)
def login():
if request.method == 'POST':
username = request.form['username']
if auth.login(username, request.form['password']):
session['username'] = username
return redirect(session['last_page'])
else:
return redirect('/login')
return render_template('login.html', title='Вход')
def login():
has_error = False
login = ''
if request.method == 'POST':
username = request.form['username']
if auth.login(username, request.form['password']):
return redirect('/')
else:
has_error = True
return render_template('login.html',
title='Вход',
login=login,
has_error=has_error)
