def test_verify_confirmation_token(self):
u = User(name='ose', password='******', email='*****@*****.**')
u2 = User(name='john', password='******', email='*****@*****.**')
token = u.generate_confirmation_token()
token2 = u2.generate_confirmation_token()
obj = u.validate_confirmation_token(token)
obj2 = u.validate_confirmation_token(token2)
self.assertTrue(obj)
def create_user():
request_data = request.get_json()
print(request_data)
user = User(email=request_data['email'],
username=request_data['username'],
password=request_data['password'])
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token()
return jsonify({'token': g.current_user.generate_auth_token(
expiration=3600), 'expiration': 3600})
def new_user():
the_form = UserForm()
if the_form.validate_on_submit():
user = User(login=the_form.login.data, password=the_form.password.data, email=the_form.email.data)
modules['db'].session.add(user)
modules['db'].session.commit()
flash('Ok, user '+user.login+' registered successfully! Confirmation email has been sent to '+user.email)
token = user.generate_confirmation_token()
send_email(user.email, "Account confirmation", 'auth/email/confirm', user=user, token=token)
return redirect(url_for('.index'))
return render_template('new_user.html', new_user_form=the_form)
def register():
form = RegistrationForm()
if form.validate_on_submit():
user = User(email=form.email.data,
username=form.username.data,
password=form.password.data)
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token()
send_email(user.email, 'Confirm Your Account',
'auth/email/confirm', user=user, token=token)
flash('A confirmation email has been sent to you by email.')
return redirect(url_for('auth.login'))
return render_template('auth/register.html', form=form)
def register():
"""
This function allows visitor to register
:return: if the user registers successful, prompt the user to activate account
"""
form = RegistrationForm()
if form.validate_on_submit():
user = User(email=form.email.data,
username=form.username.data,
password=form.password.data,
login_type='website')
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token()
send_email(user.email,
'Confirm Your Account',
'auth/email/confirm',
user=user,
token=token)
flash(
'A confirmation email has been sent to you by email, please check your mail'
)
return redirect(url_for('auth.login', type='index'))
return render_template('auth/login/register.html', form=form)
def test_generate_confirmation_token(self):
u = User(name='ose', password='******', email='*****@*****.**')
self.assertTrue(u.generate_confirmation_token())
self.assertTrue(type(u.generate_confirmation_token()) == str)
