codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix twofa

Browse Source
This commit is contained in:
jxxghp 2024-02-14 21:11:35 +08:00
parent 4248b8fa4e
commit 24c2d3f8ca
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/helper/twofa.py
View File

@ -4,6 +4,7 @@ import hmac
import struct import struct
import sys import sys
import time import time
from app.log import logger from app.log import logger
@ -16,7 +17,8 @@ class TwoFactorAuth:
self.code = code_or_secret self.code = code_or_secret
self.secret = None self.secret = None
def __calc(self, secret_key: str) -> str: @staticmethod
def __calc(secret_key: str) -> str:
try: try:
input_time = int(time.time()) // 30 input_time = int(time.time()) // 30
key = base64.b32decode(secret_key) key = base64.b32decode(secret_key)
@ -28,12 +30,12 @@ class TwoFactorAuth:
else ord(str(google_code[19])) & 15 else ord(str(google_code[19])) & 15
) )
google_code = str( google_code = str(
(struct.unpack(">I", google_code[o : o + 4])[0] & 0x7FFFFFFF) % 1000000 (struct.unpack(">I", google_code[o: o + 4])[0] & 0x7FFFFFFF) % 1000000
) )
return f"0{google_code}" if len(google_code) == 5 else google_code return f"0{google_code}" if len(google_code) == 5 else google_code
except Exception as e: except Exception as e:
logger.error(f"计算动态验证码失败：{str(e)}") logger.error(f"计算动态验证码失败：{str(e)}")
return None return ""
def get_code(self) -> str: def get_code(self) -> str:
return self.code or self.__calc(self.secret) return self.code or self.__calc(self.secret)