def login(request): if request.method == 'GET': return render(request, 'login.html') else: user_name = request.POST['name'] pwd = request.POST['pass'] user = Users.objects.filter(user=user_name) if len(user): user = user[0] if check_password(password=pwd, encoded=user.password): request.session['login_user'] = user_name return redirect('/table') return render(request, 'login.html', {'msg': -1})
时间: 2023-12-06 12:40:35 浏览: 26
这段代码是一个简单的登录逻辑,首先通过GET请求渲染出登录页面,然后通过POST请求获取用户提交的用户名和密码,再通过Django的ORM框架查询数据库中是否存在该用户。如果存在,则通过check_password函数验证密码是否正确,如果正确则将用户信息存入session中并重定向到table页面。如果不存在该用户或者密码不正确,再次渲染登录页面并传递错误信息给前端页面。需要注意的是,这段代码中的Users是一个Django的model类,用于操作数据库中的用户表。
相关问题
def register(request): if request.method == 'GET': return render(request, 'login.html') else: user_name = request.POST['regname'] pwd1 = request.POST['regpass'] user = Users() user.user = user_name user.password = pwd1 user.save() request.session['login_user'] = user_name return redirect('/')
这段代码是一个简单的用户注册功能的后端实现,当请求为GET时,返回登录页面,当请求为POST时,获取前端发送的用户名和密码,在数据库中创建一个新的用户,然后将用户名存入session中并重定向到首页。
需要注意的是,这段代码没有进行任何的用户输入验证和密码加密,因此存在安全隐患,建议添加合适的验证和加密机制来保证数据安全。
优化这段代码,要求在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')
您的代码已经是一个良好的起点,但有一些可以进行优化的地方。下面是对您提供的代码进行优化的建议:
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`模板文件。