mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-11-03 08:05:13 +00:00
[utils] Add improved jwt_encode function (#14071)
Also deprecates `jwt_encode_hs256` Authored by: bashonly
This commit is contained in:
@@ -1,4 +1,8 @@
|
||||
"""Deprecated - New code should avoid these"""
|
||||
import base64
|
||||
import hashlib
|
||||
import hmac
|
||||
import json
|
||||
import warnings
|
||||
|
||||
from ..compat.compat_utils import passthrough_module
|
||||
@@ -28,4 +32,18 @@ def intlist_to_bytes(xs):
|
||||
return struct.pack('%dB' % len(xs), *xs)
|
||||
|
||||
|
||||
def jwt_encode_hs256(payload_data, key, headers={}):
|
||||
header_data = {
|
||||
'alg': 'HS256',
|
||||
'typ': 'JWT',
|
||||
}
|
||||
if headers:
|
||||
header_data.update(headers)
|
||||
header_b64 = base64.b64encode(json.dumps(header_data).encode())
|
||||
payload_b64 = base64.b64encode(json.dumps(payload_data).encode())
|
||||
h = hmac.new(key.encode(), header_b64 + b'.' + payload_b64, hashlib.sha256)
|
||||
signature_b64 = base64.b64encode(h.digest())
|
||||
return header_b64 + b'.' + payload_b64 + b'.' + signature_b64
|
||||
|
||||
|
||||
compiled_regex_type = type(re.compile(''))
|
||||
|
||||
Reference in New Issue
Block a user