def create_demo_user():
demo_user = User(username='******',
email='*****@*****.**',
password='******',
membership='none',
vip1= 'none',
vip2= 'none',
invitation_code_vip1=generate_verification_code(),
invitation_code_vip2=generate_verification_code()
)
db.session.add(demo_user)
db.session.commit()
def register_cn():
if current_user.is_authenticated:
return redirect(url_for('webapp.welcome'))
form = RegistrationForm()
if form.validate_on_submit():
# hash the original user password
hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
user = User(username = form.username.data, email = form.email.data,
password = hashed_password)
db.session.add(user)
db.session.commit()
flash('我们为您创建了账户! 您现在可以登录了！', 'success')
return redirect(url_for('user.login_cn'))
return render_template('register.html', title = '注册', form = form, get_defaulte_language='cn')
def register_en():
if current_user.is_authenticated:
return redirect(url_for('webapp.welcome'))
form = RegistrationForm()
if form.validate_on_submit():
# hash the original user password
hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
user = User(username = form.username.data, email = form.email.data,
password = hashed_password)
db.session.add(user)
db.session.commit()
flash('Your account has been created! You are now able to login in!', 'success')
return redirect(url_for('user.login_en'))
return render_template('register.html', title = 'Register', form = form, lang='en')
def reset_token_en(token):
if current_user.is_authenticated:
return redirect(url_for('webapp.welcome'))
user = User.verify_reset_token(token)
if user is None:
flash('This is an invalid or expired token!', 'warning')
return redirect(url_for('user.reset_request'))
form = ResetPasswordForm()
if form.validate_on_submit():
# hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
# user.password = hashed_password
user.password = form.password.data
db.session.commit()
flash('Your password has been updated! You are now able to log in.', 'success')
return redirect(url_for('user.login_en'))
return render_template('reset_token.html', title='Reset Password', form=form)
def reset_token_cn(token):
if current_user.is_authenticated:
# return redirect(url_for('main.home'))
return redirect(url_for('webapp.welcome'))
user = User.verify_reset_token(token)
if user is None:
flash('这是一个无效的连接，或者该连接已过期！', 'warning')
return redirect(url_for('user.reset_request'))
form = ResetPasswordForm()
if form.validate_on_submit():
# hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
# user.password = hashed_password
user.password = form.password.data
db.session.commit()
flash('您的密码已更新！ 请重新登录.', 'success')
return redirect(url_for('user.login_cn'))
return render_template('reset_token.html', title='重置密码', form=form)
def fake_user(count = 50):
for i in range(count):
membership = random.choice(['none', 'week', 'month', 'year'])
if membership == 'none':
vip1 = 'no'
vip2 = 'no'
else:
vip1 = 'yes'
vip2 = random.choice(['yes', 'no'])
user = User(username = fake.name(),
email = fake.email(),
password = '******',
membership = membership,
vip1 = vip1,
vip2 = vip2,
invitation_code_vip1 = generate_verification_code(),
invitation_code_vip2 = generate_verification_code()
)
for j in range(random.randint(1, 5)):
post = Post.query.get(random.randint(1, Post.query.count()))
if post not in user.like:
user.like.append(post)
db.session.commit()