网站首页 > 厂商资讯 > 禾蛙 >

如何在Python中创建JWT令牌？

在当今这个信息爆炸的时代，身份验证和安全认证是保证数据安全和用户隐私的关键。JSON Web Tokens（JWT）作为一种轻量级的安全令牌，因其高效、易于使用和扩展性强的特点，在众多领域得到了广泛应用。本文将详细介绍如何在Python中创建JWT令牌，并分享一些实际案例。

一、JWT简介

JWT（JSON Web Tokens）是一种开放标准（RFC 7519），用于在各方之间安全地传输信息。JWT是一种紧凑且 URL 安全的方式，用于在各方之间作为JSON对象传输信息。它被设计为紧凑和自包含，易于传输，因此适用于作为API的认证信息。

JWT的基本结构如下：

eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c

该结构包含三个部分：

Header：头部信息，描述JWT的算法和类型。
Payload：负载信息，包含用户信息。
Signature：签名信息，用于验证JWT的完整性和真实性。

二、Python中创建JWT令牌

在Python中，我们可以使用PyJWT库来创建JWT令牌。以下是创建JWT令牌的基本步骤：

安装PyJWT库

pip install PyJWT

导入PyJWT库并创建JWT令牌

import jwt

import datetime



# 密钥

SECRET_KEY = 'your_secret_key'



# 创建载荷信息

payload = {

    'sub': '1234567890',

    'name': 'John Doe',

    'iat': datetime.datetime.utcnow(),

    'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)

}



# 创建JWT令牌

token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')



print(token)

在上面的代码中，我们首先定义了一个密钥SECRET_KEY，然后创建了一个载荷信息payload，包括用户信息、发行时间和过期时间。最后，我们使用jwt.encode()函数和HS256算法来创建JWT令牌。

三、验证JWT令牌

在Python中，我们也可以使用PyJWT库来验证JWT令牌。以下是验证JWT令牌的基本步骤：

导入PyJWT库

import jwt

验证JWT令牌

def verify_token(token, secret_key):

    try:

        payload = jwt.decode(token, secret_key, algorithms=['HS256'])

        return payload

    except jwt.ExpiredSignatureError:

        return "Token has expired"

    except jwt.InvalidTokenError:

        return "Invalid token"



# 验证JWT令牌

token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c'

secret_key = 'your_secret_key'



print(verify_token(token, secret_key))

在上面的代码中，我们定义了一个verify_token函数，用于验证JWT令牌。如果令牌有效，函数将返回载荷信息；如果令牌过期或无效，函数将返回相应的错误信息。

四、案例分析

以下是一个使用JWT令牌实现API认证的简单案例：

用户登录并获取JWT令牌

import jwt



# 用户信息

user_info = {

    'username': 'user1',

    'password': 'password1'

}



# 密钥

SECRET_KEY = 'your_secret_key'



# 创建JWT令牌

token = jwt.encode(user_info, SECRET_KEY, algorithm='HS256')



print(token)

用户使用JWT令牌访问受保护的API

import jwt



# 密钥

SECRET_KEY = 'your_secret_key'



# 受保护的API

def protected_api(token):

    try:

        payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])

        return "Access granted"

    except jwt.ExpiredSignatureError:

        return "Token has expired"

    except jwt.InvalidTokenError:

        return "Invalid token"



# 用户访问受保护的API

token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c'

print(protected_api(token))

在这个案例中，用户首先登录并获取JWT令牌，然后使用该令牌访问受保护的API。如果令牌有效，API将返回“Access granted”，否则将返回相应的错误信息。

通过以上内容，我们了解了如何在Python中创建JWT令牌，并学习了如何验证JWT令牌。JWT因其高效、易于使用和扩展性强的特点，在身份验证和安全认证领域得到了广泛应用。希望本文能帮助您更好地理解和应用JWT技术。