def test_func_email_is_free(self):
business = Business(name='name',
password='******',
email='email',
confirmed=True)
business.registered_on -= timedelta(days=32)
db.session.add(business)
db.session.commit()
self.assertTrue(Business.email_is_free('email1'))
self.assertFalse(Business.email_is_free('email'))
def registration_business():
if session.get('id', '') != '':
if not Business.check_confirmed(session.get('id', '')):
session['id'] = ''
return redirect(url_for('business.business_card', business_id=session.get('id', '')))
session['id_client'] = ''
form = RegistrationBusinessForm(csrf_enabled=False)
if form.validate_on_submit():
free_name = Business.name_is_free(form.name.data)
free_email = Business.email_is_free(form.email.data)
check_pass = (form.password.data == form.check_password.data)
password_is_valid = Business.password_is_valid(form.password.data)
if free_name and check_pass and password_is_valid and free_email:
Business.send_token(form.email.data)
business = Business(name=form.name.data, password=form.password.data,
email=form.email.data, confirmed=False)
Business.save(business)
return redirect(
"/business/feedback/Регистрация прошла успешно. Пройдите по ссылке, \
которую мы отправили вам на почту, чтобы подтвердить её")
elif not free_name:
form.name.errors = ('Название занято', '')
elif not free_email:
form.email.errors = ('Почта занята', '')
elif not password_is_valid:
form.check_password.errors = ('Пароль не может содержать пробелы', '')
elif not check_pass:
form.check_password.errors = ('Пароли не совпадают', '')
return render_template('registration_business.html', title='Регистрация бизнеса', form=form)