Handle error case for no MFA methods (#203)

Handle error case for no MFA methods
jeffreydwalter · Mar 28, 2023 · 4fc678e · 4fc678e
1 parent 70f66a0
commit 4fc678e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/arlo.py b/arlo.py
@@ -219,9 +219,14 @@ def LoginMFA(self, username, password, google_credential_file):
             headers=headers,
             raw=True
         )
-        email_factor_id = next(i for i in factors_body['data']['items'] if i['factorType'] == 'EMAIL' and i['factorRole'] == "PRIMARY")['factorId']
-        if email_factor_id is None:
-            email_factor_id = next(i for i in factors_body['data']['items'] if i['factorType'] == 'EMAIL' and i['factorRole'] == "SECONDARY")['factorId']
+        email_factor = next((i for i in factors_body['data']['items'] if i['factorType'] == 'EMAIL' and i['factorRole'] == "PRIMARY"), None)
+        if email_factor is None:
+            email_factor = next((i for i in factors_body['data']['items'] if i['factorType'] == 'EMAIL' and i['factorRole'] == "SECONDARY"), None)
+
+        if email_factor is None:
+            raise Exception('No EMAIL MFA method registered with the account.')
+
+        email_factor_id = email_factor['factorId']
 
         # Start factor auth
         start_auth_body = self.request.post(