def registration(er=None):
form = RegistrationForm()
if form.validate_on_submit():
users = UsersModel(db.get_connection())
users.insert(
form.username.data,
str(md5(bytes(form.password.data,
encoding='utf-8')).hexdigest())) #add new user
return redirect('/login')
return render_template('registration.html', form=form, error=er)
def register():
if current_user.is_authenticated:
return redirect(url_for('scoreboard'))
form = RegistrationForm()
if form.validate_on_submit():
user = create_user(form.username.data, form.email.data,
form.password.data)
user.save()
flash('Congratulations, you are now a registered user!')
login_user(user, remember=True)
next_page = request.args.get('next')
if not next_page or url_parse(next_page).netloc != '':
next_page = url_for('scoreboard')
return redirect(next_page)
return render_template('register.html', title='Register', form=form)
def registration(er=None):
form = RegistrationForm()
if form.validate_on_submit(): # if password, login, and photo is not empty
users = UsersModel(db.get_connection())
if users.exists_only_by_name(form.username.data):
return redirect(
'/registration/exists'
) # if user is already in db you should do reregistration
else:
users.insert(
form.username.data,
str(
md5(bytes(form.password.data, encoding='utf-8')).hexdigest(
))) # add to db hash of the password for better encryption
form.fileName.data.save(
os.path.join(os.path.join('static', 'avas'),
str(users.get_table_size()) + ".jpg"))
return redirect('/login')
return render_template('registration.html', form=form, error=er)
def registration(er=None):
form = RegistrationForm()
if form.validate_on_submit():
users = UsersModel(db.get_connection())
udm = UserDataModel(db.get_connection())
find = users.find(form.username.data)
if find[0]:
return redirect('/registration/exists')
else:
if form.username.data == 'admin':
users.insert(form.username.data, form.password.data, 'admin')
else:
users.insert(form.username.data, form.password.data,
'user') # add new user
udm.insert(form.username.data, 'Фамилия', 'Имя', 'Отчество',
'Адрес', 'Номер счетчика')
return redirect('/login')
return render_template('registration.html',
form=form,
error=er,
text="Авторизоваться")
def registration():
# Форма регистрации
form = RegistrationForm()
# POST
if form.validate_on_submit():
# Проверка паролей на совпадение
if form.password.data != form.password_again.data:
return render_template('registration.html',
form=form,
message='Пароли не совпадают!')
# Создаём сессию подключения к БД
session = db_session.create_session()
# Проверка почты на уникальность
if session.query(Users).filter(Users.email == form.email.data).first():
return render_template(
'registration.html',
form=form,
message='Пользователь с такой почтой уже существует!')
# Проверка логина на уникальность
if session.query(Users).filter(Users.login == form.login.data).first():
return render_template(
'registration.html',
form=form,
message='Пользователь с таким логином уже существует!')
# Добавление пользователя в БД
user = Users(login=form.login.data, email=form.email.data)
# Генерация хешированного пароля
user.generate_hashed_password(form.password.data)
session.add(user)
# Сохранение пользователя
session.commit()
# Перенаправление на страницу входа
return redirect('/login')
return render_template('registration.html', form=form)