def login(): # login in func if current_user.is_authenticated: return redirect(url_for('public.home')) form = LoginForm() if form.validate_on_submit(): username_or_email = form.username.data password = form.password.data # remember = form.remember.data user = [ User.query.filter(User.username == username_or_email).first(), User.query.filter(User.email == username_or_email).first() ] if user[0]: if user[0].check_password(password): # login_user(user[0], remember) login_user(user[0]) flash('Welcome back.', 'info') return redirect_back() else: flash('账号或者密码错误,请重新输入!', 'warning') elif user[1]: if user[1].check_password(password): login_user(user[1]) flash('Welcome back.', 'info') return redirect_back() else: flash('账号或者密码错误,请重新输入!', 'warning') else: flash('No account.', 'warning') return render_template('public/login.html', form=form)
def add_all_active(): if current_user.is_admin: users = User.query.order_by(User.created_at.desc()).all() for user in users: user.active = True db.session.commit() return redirect_back() else: return abort(404)
def cancleactive(): if current_user.is_admin: username = request.args.get('username') user = User.query.filter(User.username == username).first() user.active = False db.session.commit() return redirect_back() else: return abort(404)