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)