快捷導(dǎo)航

Python JWT認證與pyjwt包詳細介紹

更新時間：2023年05月23日 14:29:41 作者：alwaysrun

JWT的聲明一般被用來在身份提供者和服務(wù)提供者間傳遞被認證的用戶身份信息，以便于從資源服務(wù)器獲取資源，也增加一些額外的其它業(yè)務(wù)邏輯所必須的聲明信息，該token也可直接被用于認證，也可被加密,這篇文章主要介紹了Python JWT認證與pyjwt包簡介,需要的朋友可以參考下

JWT認證簡介

JWT（JSON Web Tokens），是為了在網(wǎng)絡(luò)應(yīng)用環(huán)境間傳遞聲明而執(zhí)行的一種開放的行業(yè)標(biāo)準(zhǔn)（RFC7519）。該token被設(shè)計為緊湊且安全的，特別適用于分布式站點的單點登錄（SSO）場景。JWT的聲明一般被用來在身份提供者和服務(wù)提供者間傳遞被認證的用戶身份信息，以便于從資源服務(wù)器獲取資源，也可以增加一些額外的其它業(yè)務(wù)邏輯所必須的聲明信息，該token也可直接被用于認證，也可被加密。

構(gòu)成

JWT就是由三段信息構(gòu)成，通過點號.連接的一段字符串，如：

eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyX2lkIjoidGVzdCIsImV4cCI6MTY4NDQwOTE4MH0.K-nvXX3kS0I0RKny0J33tuns7SFttYhQwAUtqOS40aw

第一部分為頭部（header，base64編碼）：

聲明類型：jwt；
聲明加密算法：如’HS256’。

{
  'typ': 'JWT',
  'alg': 'HS256'   
}

第二部分為載荷（payload，base64編碼），存放認證信息、有效期等：

{
    "user_id":"test",
    "exp":1684410152
}

第三部是簽名（使用第一部分聲明的算法進行HASH處理），Hash內(nèi)容：

header (base64后的)
payload (base64后的)
secret：加密鹽

HMACSHA256(
 base64UrlEncode(header) + "." + base64UrlEncode(payload),
 secret
)

載荷聲明

標(biāo)準(zhǔn)中注冊的payload聲明 (建議但不強制使用) ：

iss: jwt簽發(fā)者；
sub: jwt所面向的用戶；
aud: 接收jwt的一方；
exp: jwt的過期時間，這個過期時間必須要大于簽發(fā)時間；過期解碼會拋出ExpiredSignatureError異常；
nbf: 定義在什么時間之前，該jwt都是不可用的；在此時間之前解碼會拋出ImmatureSignatureError異常；
iat: jwt的簽發(fā)時間；在此時間之前解碼會拋出InvalidIssuedAtError異常；
jti: jwt的唯一身份標(biāo)識，主要用來作為一次性token,從而回避重放攻擊。

注意：不要在jwt的payload部分存放敏感信息，該部分是客戶端可加碼的。

pyjwt

使用前先安裝pip install pyjwt。

編解碼

在pyjwt中通過encode進行編碼，通過decode進行解碼。

import jwt
JWT_TOKEN_EXPIRE_SECONDS = 3600 * 2  # token有效時間 2小時
JWT_TOKEN_SECRET_SALT = 'salt.2023.05.18'
JWT_TOKEN_ALGORITHM = 'HS256'  # HASH算法
def generate_jwt_token(user: str) -> str:
    """根據(jù)用戶id生成token"""
    data = {'user_id': user, 'exp': int(time.time()) + JWT_TOKEN_EXPIRE_SECONDS}
    print("generate data:", data)
    jwtToken = jwt.encode(data, JWT_TOKEN_SECRET_SALT, algorithm=JWT_TOKEN_ALGORITHM)
    return jwtToken
def verify_jwt_token(user: str, jwtToken: str) -> bool:
    """驗證用戶token"""
    data = {'user_id': user}
    try:
        payload = jwt.decode(jwtToken, JWT_TOKEN_SECRET_SALT, algorithms=[JWT_TOKEN_ALGORITHM])
        print("verify:", payload)
        exp = int(payload.pop('exp'))
        if time.time() > exp:
            print('已失效')
            return False
        return data == payload
    except jwt.exceptions.ExpiredSignatureError as ex:
        print('token簽名過期:', ex)
    except jwt.PyJWTError as ex:
        print('token解析失敗:', ex)
    return False

flask中驗證

flask是一個輕量級的web框架，可方便與jwt結(jié)合進行驗證。

先定義一個修飾器（token通過header來攜帶）：

from flask import Flask, request
from functools import wraps
def request_token_validate(f):
    @wraps(f)
    def toValidate(*args, **kwargs):
        token = request.headers.get('Authorization')
        if not token:
            logger.error("token invalid")
            return {'data': 'Token invalid'}, status.HTTP_401_UNAUTHORIZED
        return f(*args, **kwargs)
    return toValidate

在需要的route上添加驗證修飾器：

@app.route("/users", methods=["GET"])
@request_token_validate
def query_models():
    userList = []
    # get user list
    response_data = {"data": userList}
    return response_data, status.HTTP_200_OK

到此這篇關(guān)于Python JWT認證與pyjwt包簡介的文章就介紹到這了,更多相關(guān)JWT認證與pyjwt包內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: