用python 实现一下用户登录,包含token生成
时间: 2024-02-13 10:05:49 浏览: 18
好的,下面是一个简单的用户登录示例,包含token生成:
```python
import hashlib
import uuid
users = {
'admin': 'admin123',
'user1': '123456'
}
def generate_token(username):
# 生成token
salt = uuid.uuid4().hex
token = hashlib.sha256((username + salt).encode()).hexdigest()
return token
def login(username, password):
# 登录验证
if username in users and users[username] == password:
token = generate_token(username)
return {'token': token}
else:
return None
# 测试
user_info = login('admin', 'admin123')
if user_info:
print('登录成功,token为:', user_info['token'])
else:
print('登录失败')
```
以上代码中,我们先定义了一个users字典,保存了用户名和密码。然后定义了一个generate_token函数,用于生成token。生成token的方法比较简单,我们使用sha256对用户名和一个随机生成的salt进行加密,得到token。最后,我们定义了一个login函数,用于登录验证。如果验证成功,就生成token并返回,否则返回None。最后,我们调用login函数进行测试,如果登录成功,就输出token,否则输出登录失败。