예제 #1
0
def login():
    form = LogIn()
    Branch.branch_types()
    Types.stall_types()
    print current_user
    if current_user.is_active():
        if current_user.roleID == 1:
            return redirect(url_for('index2'))
        else:
            return redirect(url_for('index'))
    else:
        if request.method == "POST" and form.validate_on_submit():
            user = Users.query.filter_by(username=form.username.data).first()
            if user:
                if user.roleID == 2:
                    if user is not None and check_password_hash(
                            user.passwrd, form.passwrd.data):
                        msg = user.username + " logs in"
                        lgdate = str(now)
                        print lgdate
                        logmessage = Logs(details=msg, log_date=lgdate)
                        dbase.session.add(logmessage)
                        login_user(user)
                        return redirect(url_for('index'))
                    #return '<h1>Invalid username or password!</h1>'
                    flash('Invalid username or password')
                elif user.roleID == 1:
                    if user is not None and check_password_hash(
                            user.passwrd, form.passwrd.data):
                        msg = user.username + " logs in"
                        lgdate = str(now)
                        print lgdate
                        logmessage = Logs(details=msg, log_date=lgdate)
                        dbase.session.add(logmessage)
                        login_user(user)
                        login_user(user)
                        flash('You were logged in')
                        return redirect(url_for('index2'))
                    #return '<h1>Invalid username or password!</h1>'
                    flash('Invalid username or password')

                else:
                    #return '<h1>Invalid username or password!</h1>'
                    flash('Invalid username or password')
            else:
                #return '<h1>Invalid username or password!</h1>'
                flash('Invalid username or password')
    return render_template('login.html', form=form)