Code: Select all
users = [
{'username': 'admin1', 'code': 'pass1'},
{'username': 'admin2', 'code': 'pass2'},
{'username': 'admin3', 'code': 'pass3'}
]
class User(UserMixin):
def is_active(self):
return True
def is_authenticated(self):
return True
@login_manager.user_loader
def load_user(code):
user = User()
user.id = code
return user
@app.route("/")
@app.route("/login_new", methods=['POST', 'GET'])
def login():
if current_user.is_authenticated:
return redirect(url_for('booking'))
form = LoginForm()
if form.validate_on_submit():
user = [a for a in users if a["username"] == form.username.data and a["code"] == form.code.data]
if user:
user = User()
user.id = form.code.data
login_user(user)
next_page = request.args.get('next')
return redirect(next_page) if next_page else redirect(url_for('login'))
else:
flash('Login Unsuccessful. Please check your name and the code', 'danger')
return redirect(url_for('login'))
return render_template('login_new.html', title='Login_new', form=form)