mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-31 12:24:39 +08:00
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
"""
|
|
生成与验证基于时间的一次性密码，并构造绑定 URI。
|
|
"""
|
|
|
|
from typing import Tuple
|
|
|
|
import pyotp
|
|
|
|
|
|
class OtpUtils:
|
|
"""
|
|
提供基于 TOTP 的二次验证辅助能力。
|
|
"""
|
|
|
|
@staticmethod
|
|
def generate_secret_key(username: str) -> Tuple[str, str]:
|
|
"""
|
|
生成 TOTP 密钥及其配置 URI。
|
|
"""
|
|
try:
|
|
secret = pyotp.random_base32()
|
|
uri = pyotp.totp.TOTP(secret).provisioning_uri(name='MoviePilot',
|
|
issuer_name='MoviePilot(' + username + ')')
|
|
return secret, uri
|
|
except Exception as err:
|
|
print(str(err))
|
|
return "", ""
|
|
|
|
@staticmethod
|
|
def is_legal(otp_uri: str, password: str) -> bool:
|
|
"""
|
|
校验二次验证是否正确。
|
|
"""
|
|
try:
|
|
valid: bool = pyotp.TOTP(pyotp.parse_uri(otp_uri).secret).verify(password)
|
|
return valid
|
|
except Exception as err:
|
|
print(str(err))
|
|
return False
|
|
|
|
@staticmethod
|
|
def check(secret: str, password: str) -> bool:
|
|
"""
|
|
校验二次验证是否正确。
|
|
"""
|
|
try:
|
|
totp = pyotp.TOTP(secret)
|
|
valid: bool = totp.verify(password)
|
|
return valid
|
|
except Exception as err:
|
|
print(str(err))
|
|
return False
|
|
|
|
@staticmethod
|
|
def get_secret(otp_uri: str) -> str:
|
|
"""
|
|
获取uri中的secret。
|
|
"""
|
|
try:
|
|
secret: str = pyotp.parse_uri(otp_uri).secret
|
|
return secret
|
|
except Exception as err:
|
|
print(str(err))
|
|
return ""
|