20 lines
604 B
Python
20 lines
604 B
Python
"""PKCE (Proof Key for Code Exchange) 工具函数"""
|
||
from __future__ import annotations
|
||
|
||
import base64
|
||
import hashlib
|
||
import secrets
|
||
|
||
|
||
def generate_pkce() -> tuple[str, str]:
|
||
"""生成 PKCE code_verifier 与对应的 code_challenge（S256）。
|
||
|
||
Returns:
|
||
(code_verifier, code_challenge) 元组
|
||
"""
|
||
code_verifier = base64.urlsafe_b64encode(secrets.token_bytes(64)).rstrip(b"=").decode("ascii")
|
||
code_challenge = base64.urlsafe_b64encode(
|
||
hashlib.sha256(code_verifier.encode("ascii")).digest()
|
||
).rstrip(b"=").decode("ascii")
|
||
return code_verifier, code_challenge
|