python | authlib，一个强大的 Python 库！

本文来源公众号“python”，仅用于学术分享，侵权删，干货满满。

原文链接：authlib，一个强大的 Python 库！

大家好，今天为大家分享一个强大的 Python 库 - authlib。

Github地址：https://github.com/lepture/authlib

在现代 Web 开发中，认证与授权是保障用户数据安全和访问控制的核心环节。OAuth 和 OpenID Connect 等协议已经成为行业标准，用于实现安全的认证和授权机制。而 authlib 是一个专注于简化这些协议实现的 Python 库，提供了从 OAuth 客户端到服务端的全面支持，并涵盖 JWT 和加密相关工具。

通过 authlib，开发者可以轻松实现第三方登录、构建 OAuth 服务器以及保护 API 接口等功能，同时支持 Flask、Django 和 FastAPI 等主流框架的无缝集成。这使得它成为 Web 开发中不可或缺的安全工具。

安装

authlib 的安装非常简单，可以直接通过 pip 命令完成：

pip install authlib

如果使用特定框架（如 Flask 或 Django），可以安装相应的扩展包。例如，安装 Flask 支持：

pip install authlib[flask]

或者安装 Django 支持：

pip install authlib[django]

安装完成后，可以运行以下代码验证是否安装成功：

import authlib
print(authlib.__version__)

当输出正确的版本号时，说明安装已经成功。

特性

• 全面的 OAuth 支持：支持 OAuth 1 和 OAuth 2 的客户端和服务端实现，适合多种认证和授权场景。

• 内置 OpenID Connect 支持：轻松扩展 OAuth 2 功能，实现用户认证和单点登录。

• 丰富的加密工具：支持 JWT（JSON Web Token）、JWS（JSON Web Signature）、JWK（JSON Web Key）等标准，满足现代加密需求。

• 多框架兼容：与 Flask、Django、FastAPI 和 Starlette 等主流框架无缝集成。

• 高度可扩展性：提供灵活的扩展机制，可根据业务需求自定义实现。

• 简洁的 API 设计：便于开发者快速上手并实现复杂功能。

• 优秀的性能表现：基于异步处理架构，适合高并发场景下的安全认证需求。

• 支持多种授权方式：内置密码模式、授权码模式、客户端凭据模式等 OAuth 授权模式。

基本功能

通过 Authlib，开发者可以快速实现 OAuth 客户端和 JWT 令牌的生成与验证等功能。例如，以下代码展示了如何使用 Authlib 实现 GitHub 登录。

首先，初始化一个 Flask 应用，并注册 OAuth 客户端：

from flask import Flask, redirect, url_for, session
from authlib.integrations.flask_client import OAuth

app = Flask(__name__)
app.secret_key = 'your_secret_key'

oauth = OAuth(app)
github = oauth.register(
    name='github',
    client_id='YOUR_CLIENT_ID',
    client_secret='YOUR_CLIENT_SECRET',
    access_token_url='https://github.com/login/oauth/access_token',
    authorize_url='https://github.com/login/oauth/authorize',
    api_base_url='https://api.github.com/',
    client_kwargs={'scope': 'user:email'},
)

然后，实现登录和回调逻辑：

@app.route('/')
def home():
    return '欢迎使用 GitHub 登录示例！<a href="/login">登录</a>'

@app.route('/login')
def login():
    return github.authorize_redirect(url_for('authorize', _external=True))

@app.route('/authorize')
def authorize():
    token = github.authorize_access_token()
    user_info = github.get('user').json()
    return f'欢迎，{user_info["login"]}'

运行该代码后，即可实现简单的 GitHub OAuth 登录功能。

另一个常见场景是 JWT 的生成与验证。以下示例展示了如何使用 Authlib 签发和解析 JWT：

from authlib.jose import jwt, JsonWebKey

key = JsonWebKey.generate_key('oct', 256)
header = {'alg': 'HS256'}
payload = {'user_id': 123, 'name': 'Alice'}
token = jwt.encode(header, payload, key)

claims = jwt.decode(token, key)
print(f'Payload: {claims}')

通过这些基本功能，Authlib 简化了常见的认证和授权任务。

高级功能

在高级场景中，Authlib 支持构建完整的 OAuth 服务器以及 OpenID Connect 服务。以下示例展示了如何实现一个简单的 OAuth 服务器。

首先，注册 OAuth 授权逻辑：

from authlib.integrations.flask_oauth2 import AuthorizationServer
from authlib.oauth2.rfc6749 import grants

class PasswordGrant(grants.ResourceOwnerPasswordCredentialsGrant):
    def authenticate_user(self, username, password):
        if username == 'admin' and password == 'password':
            return {'user_id': 1}

server = AuthorizationServer(app)
server.register_grant(PasswordGrant)

@app.route('/oauth/token', methods=['POST'])
def issue_token():
    return server.create_token_response()

同时，Authlib 支持 OpenID Connect，通过扩展 OAuth 的功能来实现用户认证：

from authlib.oidc.core import UserInfo

class MyUserInfo(UserInfo):
    def __init__(self, user):
        self.user = user

    def get(self, key, default=None):
        return getattr(self.user, key, default)

server.register_grant(OpenIDConnectGrant)

这些高级功能使开发者能够轻松实现定制化的授权和认证服务。

实际应用场景

在实际应用中，Authlib 可以帮助开发者解决多种认证与授权需求。通过 OAuth 客户端，可以快速实现第三方登录，例如 Google 或 Facebook 登录，为用户提供便捷的访问方式。对于需要保护 API 的应用，可以使用 OAuth 2 和 JWT 实现安全的访问控制。此外，Authlib 还适合构建企业级单点登录（SSO）系统，通过 OpenID Connect 提供统一的用户认证服务。

以下是一个 API 保护示例，通过 JWT 验证访问权限：

from flask import Flask, jsonify
from authlib.integrations.flask_oauth2 import ResourceProtector
from authlib.oauth2.rfc6749 import BearerTokenValidator

require_auth = ResourceProtector()

class MyTokenValidator(BearerTokenValidator):
    def authenticate_token(self, token):
        return {'user_id': 1} if token == 'valid_token' else None

require_auth.register_token_validator(MyTokenValidator())

app = Flask(__name__)

@app.route('/api/resource')
@require_auth(None)
def protected_resource():
    return jsonify({'message': '这是受保护的资源！'})

通过这些功能，Authlib 在 API 安全、单点登录和第三方登录等场景中发挥了关键作用。

总结

Authlib 是一个功能全面且灵活的 Python 安全认证与授权工具。它支持 OAuth 客户端和服务端实现，内置 OpenID Connect 功能，同时提供丰富的加密工具，帮助开发者快速构建安全的 Web 应用。无论是实现第三方登录，保护 API，还是构建单点登录服务，Authlib 都能提供简洁高效的解决方案。

THE END !

文章结束，感谢阅读。您的点赞，收藏，评论是我继续更新的动力。大家有推荐的公众号可以评论区留言，共同学习，一起进步。