Python Flask实现基于Basic Auth的注册登录系统
发布时间：2026-10-13| 3人浏览过
本文详细介绍如何使用Python Flask框架结合Basic Auth机制构建简易的注册登录系统。通过Flask-SQLAlchemy进行数据库建模，利用passlib加密密码，并生成安全Token实现无状态身份验证。文章涵盖初始化配置、用户模型定义、注册与登录接口实现及Token验证逻辑，提供完整代码示例与效果演示。
本文主要讲解如何利用Python的Flask框架实现用户登录与注册功能，身份验证方式采用Basic Auth。
实现该功能主要依赖以下库：
import os
#Flask的基础库
from flask import Flask, abort, request, jsonify, g, url_for
#Flaks的数据库操作的库
from flask.ext.sqlalchemy import SQLAlchemy
#Flask登录注册的库
from flask.ext.httpauth import HTTPBasicAuth
#加密解密密码的库
from passlib.apps import custom_app_context as pwd_context
#URL安全序列化工具
from itsdangerous import (TimedJSONWebSignatureSerializer
as Serializer, BadSignature, SignatureExpired)
首先需要完成环境初始化。
app = Flask(__name__) # 设置密钥 app.config['SECRET_KEY'] = 'the quick brown fox jumps over the lazy dog' # 数据库的配置 app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite' app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True #数据库初始化 db = SQLAlchemy(app) # 验证的初始化 auth = HTTPBasicAuth()
接着进行数据模型构建。
SQLAlchemy作为对象关系映射（ORM）工具，能极大简化数据库操作。除了定义基础属性外，我们还封装了一些必要的方法以增强功能。
class User(db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(32), index=True)
password_hash = db.Column(db.String(64))
# 加密密码
def hash_password(self, password):
self.password_hash = pwd_context.encrypt(password)
# 验证密码
def verify_password(self, password):
return pwd_context.verify(password, self.password_hash)
# 生成token，并设置过期时间
def generate_auth_token(self, expiration=600):
s = Serializer(app.config['SECRET_KEY'], expires_in=expiration)
return s.dumps({'id': self.id})
# 静态的验证token的方法
@staticmethod
def verify_auth_token(token):
s = Serializer(app.config['SECRET_KEY'])
try:
data = s.loads(token)
except SignatureExpired:
return None # token过期
except BadSignature:
return None # token无效
user = User.query.get(data['id'])
return user
以下是用户注册功能的实现逻辑。
@app.route('/api/users', methods=['POST'])
def new_user():
username = request.json.get('username')
password = request.json.get('password')
if username is None or password is None:
abort(400) # 用户名或者密码为空
if User.query.filter_by(username=username).first() is not None:
abort(400) # 用户已存在
user = User(username=username)
# 加密密码
user.hash_password(password)
# 保存进数据库
db.session.add(user)
db.session.commit()
# 成功注册后返回用户名，Location后面接着的是跳转的地址
return (jsonify({'username': user.username}), 201,
{'Location': url_for('get_user', id=user.id, _external=True)})
接下来是登录功能的处理流程。
# 登录后获取token
@app.route('/api/token')
@auth.login_required
def get_auth_token():
# 设置token过期时间
token = g.user.generate_auth_token(600)
return jsonify({'token': token.decode('ascii'), 'duration': 600})
成功获取Token后，后续每次请求只需携带该Token即可。
我们可以编写一个专用方法来验证Token的有效性。
# 可以通过token或者账号密码登录
@app.route('/api/resource')
@auth.login_required
def get_resource():
# 如果token有效的话就返回username
return jsonify({'data': 'Hello, %s!' % g.user.username})
细心的读者可能注意到上述两个方法前均带有@auth.login_required装饰器，这正是实现权限控制的关键所在。
# 有@auth.login_required标志的都要调用这个方法,传token或者传账号和密码
@auth.verify_password
def verify_password(username_or_token, password):
# 首先验证token
user = User.verify_auth_token(username_or_token)
if not user:
# 然后再验证用户名和密码
user = User.query.filter_by(username=username_or_token).first()
if not user or not user.verify_password(password):
return False
g.user = user
return True
最后，我们需要编写程序的入口方法以启动服务。
if __name__ == '__main__':
# 如果这个数据库不存在就创建
if not os.path.exists('db.sqlite'):
db.create_all()
app.run(debug=True)
至此，整个系统开发完成。
以下是系统运行效果的展示。
用户注册界面演示。
用户登录界面演示。
Token验证过程演示。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
