feat 新增 2fa helper

app/helper/twofa.py

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