优化这段代码,要求在login.html登录验证成功后,跳转到index.html: @app.route('/') def index(): data = SourceData() if 'username' not in session: return redirect(url_for('login')) else: return render_template('login.html', form=data, title=data.title) @app.route('/login', methods=['GET', 'POST']) def login(): if request.method == 'POST': username = request.form.get('username') password = request.form.get('password') # 查询数据库检查用户名和密码是否匹配 query = "SELECT * FROM users WHERE username=%s AND password=%s" values = (username, password) cursor.execute(query, values) user = cursor.fetchone() if user: session['username'] = user[1] return redirect(url_for('index')) else: return render_template('login.html', error='用户名或密码错误') else: return render_template('login.html')
时间: 2024-04-14 15:26:12 浏览: 221
您的代码已经是一个良好的起点,但有一些可以进行优化的地方。下面是对您提供的代码进行优化的建议:
1. 使用密码哈希:在数据库中存储密码的哈希值而不是明文密码,可以提高安全性。
2. 使用表单验证:在服务器端进行用户名和密码的验证,而不是直接执行SQL查询。
3. 使用`flash`消息:使用`flash`消息来向用户显示错误信息,而不是将错误信息作为参数传递给模板。
4. 重命名路由:将路由函数的名称更改为更具描述性和一致性的名称。
下面是优化后的代码示例:
```python
from flask import Flask, render_template, request, redirect, url_for, session, flash
from werkzeug.security import check_password_hash
app = Flask(__name__)
app.secret_key = 'your_secret_key'
@app.route('/')
def home():
if 'username' not in session:
return redirect(url_for('login'))
else:
return redirect(url_for('index'))
@app.route('/index')
def index():
return render_template('index.html')
@app.route('/login', methods=['GET', 'POST'])
def login():
if request.method == 'POST':
username = request.form.get('username')
password = request.form.get('password')
# 查询数据库检查用户名和密码是否匹配
query = "SELECT * FROM users WHERE username=%s"
values = (username,)
cursor.execute(query, values)
user = cursor.fetchone()
if user and check_password_hash(user['password'], password):
session['username'] = user['username']
flash('登录成功', 'success')
return redirect(url_for('index'))
else:
flash('用户名或密码错误', 'error')
return redirect(url_for('login'))
else:
return render_template('login.html')
if __name__ == '__main__':
app.run()
```
注意,上述代码假设您已经在数据库中存储了用户的密码哈希值,并且已经配置了数据库连接。请根据您的实际情况进行相应的修改。此外,还需要根据您的实际需求来调整`index.html`和`login.html`模板文件。
阅读全文