添加虎牙短信登录

core/huya/__init__.py

@@ -12,9 +12,13 @@ __all__ = [
     "HuyaLoginError",
     "HuyaLoginResult",
     "HuyaPasswordLogin",
+    "HuyaSmsCodeResult",
+    "HuyaSmsLogin",
     "HuyaVerificationError",
     "HuyaVerificationSolver",
     "login_huya_password",
+    "login_huya_sms",
+    "send_huya_sms_code",
     "solve_huya_verification",
 ]
 
@@ -53,6 +57,27 @@ def __getattr__(name: str):
         }
         globals().update(values)
         return values[name]
+    if name in {
+        "HuyaSmsCodeResult",
+        "HuyaSmsLogin",
+        "login_huya_sms",
+        "send_huya_sms_code",
+    }:
+        from .sms_login import (
+            HuyaSmsCodeResult,
+            HuyaSmsLogin,
+            login_huya_sms,
+            send_huya_sms_code,
+        )
+
+        values = {
+            "HuyaSmsCodeResult": HuyaSmsCodeResult,
+            "HuyaSmsLogin": HuyaSmsLogin,
+            "login_huya_sms": login_huya_sms,
+            "send_huya_sms_code": send_huya_sms_code,
+        }
+        globals().update(values)
+        return values[name]
     if name in {"HuyaVerificationError", "HuyaVerificationSolver", "solve_huya_verification"}:
         from .verification import HuyaVerificationError, HuyaVerificationSolver, solve_huya_verification