优化这段代码,要求在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 17:26:12 浏览: 258
您的代码已经是一个良好的起点,但有一些可以进行优化的地方。下面是对您提供的代码进行优化的建议:
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`模板文件。
阅读全文
相关推荐












index.html是我的前端页面:<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>儿童肠梗阻医疗辅助诊断系统</title>
</head>
<body>
儿童肠梗阻医疗辅助诊断系统
400-XXX-XXXX
账号登录
<form action="/https/wenku.csdn.net/login" method="post">
<input type="text" id="userid" name="userid" placeholder="请输入账号" required>
<input type="password" id="password" name="password" placeholder="请输入密码" required>
<button type="submit">登录</button>
忘记密码?
<input type="checkbox" id="agree" required>
<label for="agree">我已阅读并同意 《服务条款》 和 《隐私政策》</label>
</form>
</body>
</html>
# 登录路由
@app.route('/login', methods=['POST'])
def login():
userid = request.form.get('userid')
password = request.form.get('password')
...
这是我用flask处理登录逻辑的代码,要求获得userid和password之后,通过userid查询数据库对应的password,正确则跳转页面到home.html







