checks if mfa_code is already defined (#40)

joshfraser · Dec 31, 2018 · a789918 · a789918
1 parent dccc35e
commit a789918
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/csv-export.py b/csv-export.py
@@ -51,7 +51,10 @@
 
     logged_in = robinhood.login(username=username, password=password)
     if logged_in != True and logged_in.get('non_field_errors') == None and logged_in['mfa_required'] == True:
-        mfa_code = os.getenv("RH_MFA")
+
+        if mfa_code is None:
+            mfa_code = os.getenv("RH_MFA")
+
         if mfa_code == "":
             print("Robinhood MFA:", end=' ')
             try: