Handle early return from two_factor_helper

2020-06-07 12:35:24 +02:00
parent b35240bdda
commit 2319fc7c4a
1 changed files with 3 additions and 0 deletions
--- a/fbchat/_session.py
+++ b/fbchat/_session.py
@@ -147,6 +147,9 @@ def two_factor_helper(session: requests.Session, r, on_2fa_callback):
        log.info("Saving browser")
        r = session.post(url, data=data, allow_redirects=False)
        log.debug("2FA location: %s", r.headers.get("Location"))
+        url = r.headers.get("Location")
+        if url and url.startswith("https://www.messenger.com/login/auth_token/"):
+            return url
        url, data = find_form_request(r.content.decode("utf-8"))

    log.info("Starting Facebook checkup flow")