예제 #1
0
파일: views.py 프로젝트: mpdevilleres/bprc
def login():
    # if current_user is not None and current_user.is_authenticated():
    #     return redirect(url_for('main.index' ))

    if request.method=='POST':
        form = LoginForm(request.form)
        if form.validate_on_submit():
            user = User.query.filter_by(username = form.username.data).first()
            if user is None:
                form.username.errors.append('Username not found')
                return render_template('user/login.html',  form = form)

            if not bcrypt.check_password_hash(user.password, form.password.data):
                form.password.errors.append('Password did not match')
                return render_template('user/login.html',  form = form)

            else:
                login_user(user, remember = form.remember_me.data)
                if current_user.group_id == 3:
                    return redirect(url_for('section_kpi_mgt.dashboard_kpi'))
                return redirect(url_for('main.index' ))

    return render_template('user/login.html', form = LoginForm())
 def test_validate_success_login_form(self):
     # Ensure correct data validates.
     form = LoginForm(username='******', password='******')
     self.assertTrue(form.validate())