update challange.py

- add a code parameter in a challenge_code_handler
Jmallone · Mar 16, 2022 · cdef5f1 · cdef5f1
1 parent 54712c5
commit cdef5f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/instagrapi/mixins/challenge.py b/instagrapi/mixins/challenge.py
@@ -186,7 +186,7 @@ def challenge_resolve_contact_form(self, challenge_url: str) -> bool:
         ), result
         for retry_code in range(5):
             for attempt in range(1, 11):
-                code = self.challenge_code_handler(self.username, choice)
+                code = self.challenge_code_handler(self.username, choice, self.code)
                 if code:
                     break
                 time.sleep(WAIT_SECONDS * attempt)
@@ -388,7 +388,7 @@ def challenge_resolve_simple(self, challenge_url: str) -> bool:
                     raise ChallengeError(f'ChallengeResolve: Choice "email" or "phone_number" (sms) not available to this account {self.last_json}')
             wait_seconds = 5
             for attempt in range(24):
-                code = self.challenge_code_handler(self.username, ChallengeChoice.EMAIL)
+                code = self.challenge_code_handler(self.username, ChallengeChoice.EMAIL, self.code)
                 if code:
                     break
                 time.sleep(wait_seconds)