def registration():
form = RegisterForm()
if form.validate_on_submit():
# Проверка правильности повторного ввода пароля
if form.password.data != form.password_again.data:
return render_template('register.html',
title='Регистрация',
form=form,
message="Пароли не совпадают")
# Проверка наличия в системе такого пользователя
session = db_session.create_session()
if session.query(User).filter(User.email == form.email.data).first():
return render_template('register.html',
title='Регистрация',
form=form,
message="Такой пользователь уже есть")
# Создание пользователя
user = User(surname=form.surname.data,
name=form.name.data,
age=form.age.data,
email=form.email.data)
# Добавление пользователя в БД
user.set_password(form.password.data)
session.add(user)
session.commit()
return redirect('/login')
return render_template('register.html', title='Регистрация', form=form)
def login():
form = LoginForm()
if form.validate_on_submit():
if form.submit_res.data:
session = db_session.create_session()
if session.query(User).filter(
User.email == form.mail.data).first():
return render_template(
"login.html",
form=form,
message="Пользователь с такой почтой уже есть")
user = User()
user.name = form.name.data
user.surname = form.surname.data
user.email = form.mail.data
user.set_password(form.password.data)
session.add(user)
session.commit()
try:
send_code(user=user)
except:
session.delete(user)
session.commit()
return render_template("login.html",
form=form,
message="Почта введена некорректно")
#login_user(user)
return redirect(url_for('.check', user=user.id))
if form.submit_sign.data:
return redirect('/signin')
return render_template('login.html', form=form)
from data import db_session
from data.models import User, Detection
db_session.global_init("test.sqlite")
session = db_session.create_session()
user_1 = User()
user_1.name = "Пользоватеasdf§ль 1"
user_1.surname = "asdfasdf§"
user_1.email = "user_§[email protected]"
user_1.set_password('s§hadfasdfit')
#damn = Detection()
#damn.number_of_people = 10
#damn.distance_violation = 2
#user_1.detections.append(damn)
session.add(user_1)
session.commit()
print('!')
for i in session.query(User):
#print(i)
session.delete(i)
session.commit()
from data import db_session
from data.models import User, Detection
db_session.global_init("damn.sqlite")
session = db_session.create_session()
user_1 = User()
user_1.name = "Пользователь 1"
user_1.surname = "11"
user_1.email = "*****@*****.**"
user_1.set_password('shit')
session.add(user_1)
session.commit()
print('Запрос одного (произвольного) пользователя:')
user = session.query(User).first()
print(user.name)
