def populte():
print('drop database')
db.drop_all()
print('drop done')
print('create database')
db.create_all()
print('create database done')
print('creating......')
for user in users:
hash_password = bcrypt.generate_password_hash(
user.get('password')).decode('utf-8')
create_user = User(username=user.get('username'),
email=user.get('email'),
password=hash_password)
account = Account()
create_user.account = account
db.session.add(create_user)
db.session.commit()
print('create user success')
print('create users done')
admin = User.query.filter_by(username='******').first()
for post in posts:
create_post = Post(title=post, content=post * 3, author=admin)
db.session.add(create_post)
db.session.commit()
print('create post success')
print('done')
def register():
if current_user.is_authenticated:
flash(f'无效操作：{current_user.username} 请退出当前账号再进行操作', 'danger')
return redirect(url_for('main.home'))
form = RegistrationForm()
if form.validate_on_submit():
hash_password = bcrypt.generate_password_hash(
form.password.data).decode('utf-8')
user = User(username=form.username.data,
email=form.email.data,
password=hash_password)
account = Account()
user.account = account
db.session.add(user)
db.session.commit()
flash(f'注册成功 - { form.username.data } 请用邮箱登陆', 'primary')
return redirect(url_for('users.login'))
return render_template('register.html', form=form, title='注册')
