Protecting part of a page with Flask-Security

Question

I have a web page generated by Flask that should show content depending on the status of a user.

At the moment, the relevant part of the jinja2 template looks like this:

{% if service.spotting_data.loco_id %}
    Spotter Info *:<br />
    {% if current_user.is_authenticated() %}
        Loco: {{ service.spotting_data.loco_id }} <br />
        Consist: {{ service.spotting_data.consist_info }}
    {% else %}
        We have information on the locomotive and consist for this service, however you need to <a href="{{ url_for('security.login') }}"> log in </a> to see it
    {% endif %}
{% endif %}

The problem arises when someone clicks on the "log in" link - at the moment they just get redirected back to the index page and not the page that includes this template.

Is there any way to feed the <a href> with an additional flag on the {{url_for()}} to ensure that after log-in the user is redirected back to this page?

If you solved this yourself, you can add it as an answer and then accept it. No need to add the answer in the question itself. You cannot close a question but you can certainly add an answer. — codegeek, Apr 17 '14 at 14:02
OK, thanks. I tried that but it said I wasn't allowed to answer my won question until 8hrs after posting. — Matthew Macdonald-Wallace, Apr 17 '14 at 14:29

score 0 · Answer 1 · answered Apr 19 '14 at 10:56

You need to pass the argument "next" to url_for.

when you render_template, send the request object to the template:

render_template("template.html",request=request)

then in your template you can do

and you will get redirected back to the current page on login...

Protecting part of a page with Flask-Security

1 Answers1