Ejemplo n.º 1
0
def index():
    """
    TODO: Handle if user is connected
    start_div is the div displayed on page load. Useful for forms with errors
    """

    # Uncomment to force https on that page (maybe just for user login ?)
    # if not app.config['DEBUG']:  # and current_user.is_anonymous()
    #     if request.scheme == "http":
    #         return redirect(url_for("index", _scheme="https"))

    if current_user.is_authenticated():
        return render_template("blog/blog_explore.html", **explore_context())

    start_div = "home-div"
    login_form = LoginForm(request.form, prefix="login")
    register_form = RegisterForm(request.form, prefix="register")

    if login_form.has_been_submitted(request):
        if login_form.validate_on_submit():
            user = User.query.filter_by(
                username=login_form.username.data).first()
            login_user(user, remember=login_form.rememberme.data)
            user.last_login = datetime.now()
            user.save()
            flash("Your are now logged in.", category="info")
            return redirect(url_for('index'))
        else:
            start_div = "login-div"

    if register_form.has_been_submitted(request):
        if register_form.validate_on_submit():
            new_user = User(username=register_form.username.data,
                            password=register_form.password.data,
                            active=True,
                            superuser=False)
            new_user.save()
            return redirect(url_for('index'))
        else:
            start_div = "registration-div"

    return render_template("index.html",
                           login_form=login_form,
                           register_form=register_form,
                           start_div=start_div)
Ejemplo n.º 2
0
def index():
    """
    TODO: Handle if user is connected
    start_div is the div displayed on page load. Useful for forms with errors
    """

    # Uncomment to force https on that page (maybe just for user login ?)
    # if not app.config['DEBUG']:  # and current_user.is_anonymous()
    #     if request.scheme == "http":
    #         return redirect(url_for("index", _scheme="https"))

    if current_user.is_authenticated():
        return render_template("blog/blog_explore.html", **explore_context())

    start_div = "home-div"
    login_form = LoginForm(request.form, prefix="login")
    register_form = RegisterForm(request.form, prefix="register")

    if login_form.has_been_submitted(request):
        if login_form.validate_on_submit():
            user = User.query.filter_by(username=login_form.username.data).first()
            login_user(user, remember=login_form.rememberme.data)
            user.last_login = datetime.now()
            user.save()
            flash("Your are now logged in.", category="info")
            return redirect(url_for('index'))
        else:
            start_div = "login-div"

    if register_form.has_been_submitted(request):
        if register_form.validate_on_submit():
            new_user = User(username=register_form.username.data, password=register_form.password.data, active=True,
                            superuser=False)
            new_user.save()
            return redirect(url_for('index'))
        else:
            start_div = "registration-div"

    return render_template("index.html", login_form=login_form, register_form=register_form, start_div=start_div)