fix otp code

vladkens · Apr 18, 2024 · 9fa6343 · 9fa6343
1 parent 264ca7a
commit 9fa6343
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/twscrape/accounts_pool.py b/twscrape/accounts_pool.py
@@ -80,7 +80,7 @@ async def add_account(
         user_agent: str | None = None,
         proxy: str | None = None,
         cookies: str | None = None,
-        mfa_code: str | None = None
+        mfa_code: str | None = None,
     ):
         qs = "SELECT * FROM accounts WHERE username = :username"
         rs = await fetchone(self._db_file, qs, {"username": username})

diff --git a/twscrape/login.py b/twscrape/login.py
@@ -2,8 +2,8 @@
 from dataclasses import dataclass
 from datetime import timedelta
 from typing import Any
-import pyotp
 
+import pyotp
 from httpx import AsyncClient, Response
 
 from .account import Account
@@ -121,6 +121,9 @@ async def login_enter_password(ctx: TaskCtx) -> Response:
 
 
 async def login_two_factor_auth_challenge(ctx: TaskCtx) -> Response:
+    if ctx.acc.mfa_code is None:
+        raise ValueError("MFA code is required")
+
     totp = pyotp.TOTP(ctx.acc.mfa_code)
     payload = {
         "flow_token": ctx.prev["flow_token"],
@@ -133,7 +136,7 @@ async def login_two_factor_auth_challenge(ctx: TaskCtx) -> Response:
     }
 
     rep = await ctx.client.post(LOGIN_URL, json=payload)
-    raise_for_status(rep, "login_two_factor_auth_challenge")
+    rep.raise_for_status()
     return rep