def register_user(login, password, mail, name) -> User:
session = db_session.create_session()
user = User()
auth = Auth()
user.is_approved = False
user.name = name
user.role = 'user'
user.key = geherate_key()
auth.login = login
auth.password = password
auth.email = mail
user.auth.append(auth)
session.add(user)
session.flush()
application = Applications()
application.user_id = user.id
application.login = login
session.add(application)
session.commit()
session.close()
return user
def register_admin(login, password, mail, name) -> User:
session = db_session.create_session()
user = User()
auth = Auth()
user.is_approved = True
user.name = name
user.role = 'admin'
user.key = geherate_key()
auth.login = login
auth.password = password
auth.email = mail
user.auth.append(auth)
session.add(user)
session.commit()
session.close()
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)
