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)
