def deliveryman_login():
user = User(name='deliveryman', role='deliveryman')
session = create_session()
session.add(user)
session.commit()
login_user(user)
session.close()
return 'ok'
def cook_login():
user = User(name='cook', role='cook')
session = create_session()
session.add(user)
session.commit()
login_user(user)
session.close()
return 'ok'
def deliveryman_login():
form = WorkerLoginForm()
if form.validate_on_submit():
session = db_session.create_session()
user = session.query(User).filter(
User.role == 'deliveryman',
User.phone_number == form.phone_number.data).first()
if user and user.check_password(form.password.data):
login_user(user, remember=True)
return redirect('/deliveryman/')
return render_template('worker_login.html',
message='Неправильный логин или пароль',
form=form)
return render_template('worker_login.html', title='Авторизация', form=form)
def test_login_legal_user_not_from_training_set(self, mock_record):
if not self.check_files_exist():
prepare()
users = get_users()
speaker_file = join(users[0], listdir(users[0])[-1])
mock_record.return_value = wavfile.read(speaker_file)
self.assertTrue(login_user())
def test_login_illegal_user_not_from_training_set(self, mock_record):
if not self.check_files_exist():
prepare()
users = get_users()
illegal_user = '******'
self.assertFalse(illegal_user in users)
mock_record.return_value = wavfile.read(join(illegal_user, 'ru_0014.wav'))
self.assertFalse(login_user())
def test_login_unknown_illegal_user(self, mock_record):
if not self.check_files_exist():
prepare()
users = get_users()
illegal_user = '******'
self.assertFalse(illegal_user in users)
mock_record.return_value = wavfile.read(join(illegal_user, 'ru_0042.wav'))
self.assertFalse(login_user())
def client_login():
form = ClientLoginForm()
if form.validate_on_submit():
session = db_session.create_session()
user = session.query(User).filter(User.role == 'client').filter(
User.phone_number == form.phone_number.data).first()
if user:
login_user(user, remember=True)
return redirect('/')
user = User(phone_number=form.phone_number.data, role='client')
cart = Order(client=user)
cart.update_total_cost()
session.add(user)
session.add(cart)
session.commit()
login_user(user, remember=True)
return redirect('/')
return render_template('login.html', title='Авторизация', form=form)