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='注册')