python 產(chǎn)生token及token驗(yàn)證的方法
1、前言
最近在做微信公眾號(hào)開發(fā)在進(jìn)行網(wǎng)頁(yè)授權(quán)時(shí),微信需要用戶自己在授權(quán)url中帶上一個(gè)類似token的state的參數(shù),以防止跨站攻擊。
在經(jīng)過(guò)再三思考之后,自己試著實(shí)現(xiàn)一個(gè)產(chǎn)生token和驗(yàn)證token的方案。接下就把code貼出來(lái)。希望讀者指導(dǎo)一下。
2、產(chǎn)生token
原理:
通過(guò)hmac sha1 算法產(chǎn)生用戶給定的key和token的最大過(guò)期時(shí)間戳的一個(gè)消息摘要,將這個(gè)消息摘要和最大過(guò)期時(shí)間戳通過(guò)":"拼接起來(lái),再進(jìn)行base64編碼,生成最終的token
實(shí)現(xiàn):
import time
import base64
import hmac
def generate_token(key, expire=3600):
r'''
@Args:
key: str (用戶給定的key,需要用戶保存以便之后驗(yàn)證token,每次產(chǎn)生token時(shí)的key 都可以是同一個(gè)key)
expire: int(最大有效時(shí)間,單位為s)
@Return:
state: str
'''
ts_str = str(time.time() + expire)
ts_byte = ts_str.encode("utf-8")
sha1_tshexstr = hmac.new(key.encode("utf-8"),ts_byte,'sha1').hexdigest()
token = ts_str+':'+sha1_tshexstr
b64_token = base64.urlsafe_b64encode(token.encode("utf-8"))
return b64_token.decode("utf-8")
3、驗(yàn)證token
原理:
將token進(jìn)行base64解碼,通過(guò)token得到token最大過(guò)期時(shí)間戳和消息摘要。判斷token是否過(guò)期。
如沒過(guò)期才將 從token中的取得最大過(guò)期時(shí)間戳進(jìn)行hmac sha1 算法運(yùn)算(注意這里的key要與產(chǎn)生token的key要相同),最后將產(chǎn)生的摘要與通過(guò)token取得消息摘要進(jìn)行對(duì)比, 如果兩個(gè)摘要相等,則token有效,否則token無(wú)效 。
實(shí)現(xiàn):
import time
import base64
import hmac
def certify_token(key, token):
r'''
@Args:
key: str
token: str
@Returns:
boolean
'''
token_str = base64.urlsafe_b64decode(state).decode('utf-8')
token_list = token_str.split(':')
if len(token_list) != 2:
return False
ts_str = token_list[0]
if float(ts_str) < time.time():
# token expired
return False
known_sha1_tsstr = token_list[1]
sha1 = hmac.new(key.encode("utf-8"),ts_str.encode('utf-8'),'sha1')
calc_sha1_tsstr = sha1.hexdigest()
if calc_sha1_tsstr != known_sha1_tsstr:
# token certification failed
return False
# token certification success
return True
4、用法
key = "JD98Dskw=23njQndW9D" # 一小時(shí)后過(guò)期 token = generate_token(key, 3600) certify_token(key, token)
5、Note!!!
本代碼只能在python3.x 中運(yùn)行。
以上這篇python 產(chǎn)生token及token驗(yàn)證的方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
python控制nao機(jī)器人身體動(dòng)作實(shí)例詳解
這篇文章主要為大家詳細(xì)介紹了python控制nao機(jī)器人身體動(dòng)作實(shí)例,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2019-04-04
在Python中使用xlrd和xlwt讀寫Excel文件代碼實(shí)例
這篇文章主要介紹了在Python中使用xlrd和xlwt讀寫Excel文件代碼實(shí)例,python操作excel主要用到xlrd和xlwt兩個(gè)庫(kù),即xlrd是讀excel,xlwt是寫excel庫(kù),文中提供了部分實(shí)例代碼,需要的朋友可以參考下2023-08-08
使用GitHub和Python實(shí)現(xiàn)持續(xù)部署的方法
這篇文章主要介紹了使用GitHub和Python實(shí)現(xiàn)持續(xù)部署的方法,小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-05-05
Python統(tǒng)計(jì)單詞出現(xiàn)的次數(shù)
最近經(jīng)理交給我一項(xiàng)任務(wù),統(tǒng)計(jì)一個(gè)文件中每個(gè)單詞出現(xiàn)的次數(shù),列出出現(xiàn)頻率最多的5個(gè)單詞。本文給大家?guī)?lái)了python 統(tǒng)計(jì)單詞次數(shù)的思路解析,需要的朋友參考下吧2018-04-04

