Flask站点登录后将登录/注册替换为登出按钮的技术实现咨询

Hey there! Let's get that navigation bar to toggle between login/signup and logout buttons based on user authentication status. Looking at your code, there are a few small tweaks we need to make to get this working smoothly:

1. Fix the Template Authentication Check #

Right now your template is checking for a variable is_authenticated that isn't being passed from your route, and we can simplify this using Flask-Login's built-in current_user object (which is automatically available in all templates when using Flask-Login).

Update the navigation bar logic in index.html to use current_user.is_authenticated instead:

<ul class="nav navbar-nav navbar-right ul1">
 <li class="li1"><a id="a1" style="color: red;" href="{{ url_for('index') }}">Home</a></li>
 <li class="li1"><a id="a1" href="{{ url_for('.index', _anchor='about') }}">About Product</a></li>
 <li class="li1"><a id="a1" href="{{ url_for('.index', _anchor='foot') }}">Contact Us</a></li>
 {% if current_user.is_authenticated %}
 <li class="li1"><a id="a1" href="{{ url_for('logout') }}">Logout</a></li>
 {% else %}
 <li class="li1"><a id="a1" href="{{ url_for('signup') }}">Signup</a></li>
 <li class="li1"><a id="a1" href="{{ url_for('login') }}">Login</a></li>
 {% endif %}
</ul>

2. Remove Redundant Code from Your User Model #

Your User class inherits from UserMixin, which already provides a working is_authenticated property. The custom method you added will conflict with this, so go ahead and delete it from app.py:

class User(UserMixin, db.Model):
 id = db.Column(db.Integer, primary_key=True)
 username = db.Column(db.String(15), unique=True)
 email = db.Column(db.String(50), unique=True)
 password = db.Column(db.String(80))
 # Delete this entire method - UserMixin handles it for you
 # def is_authenticated(self):
 #     """Return True if the user is authenticated."""
 #     return self.authenticated

3. Verify the Route Setup #

Your existing routes for login, logout, and signup are already correctly using Flask-Login's utilities (login_user, logout_user, @login_required), so no changes are needed there. The index route doesn't need to pass any extra variables to the template anymore since we're using current_user directly.

Test It Out #

1. Start your app and visit the homepage - you'll see the Signup and Login buttons.
2. Register a new user, then log in. When you redirect back to the homepage, you'll now see the Logout button instead.
3. Click Logout, and you'll be sent back to the homepage with the Signup/Login buttons visible again.

内容的提问来源于stack exchange，提问作者Sneha Roy