Retry on login error when creating API key

Signed-off-by: Florian Pritz <bluewind@xinu.at>

1 files changed, 17 insertions, 5 deletions

diff --git a/fb.py b/fb.py
index 48b34f4..85dd375 100755
--- a/fb.py
+++ b/fb.py
@@ -775,12 +775,24 @@ class FBClient:
         localuser = getpass.getuser()
         data = []
 
-        data.append({'username': self.get_input("Username: ")})
-        data.append({'password': self.get_input("Password: ", display=False)})
-        data.append({'comment': "fb-client %s@%s" % (localuser, hostname)})
-        data.append({'access_level': "apikey"})
+        while True:
+            data.append({'username': self.get_input("Username: ")})
+            data.append({'password': self.get_input("Password: ", display=False)})
+            data.append({'comment': "fb-client %s@%s" % (localuser, hostname)})
+            data.append({'access_level': "apikey"})
+
+            try:
+                resp = self.curlw.send_post_noauth('/user/create_apikey', data)
+                # break out of while loop on success
+                break
+            except APIException as e:
+                if e.error_id == 'user/login-failed':
+                    eprint(e)
+                    eprint("\nPlease try again:")
+                    continue
+                else:
+                    raise
 
-        resp = self.curlw.send_post_noauth('/user/create_apikey', data)
 
         self.makedirs(os.path.dirname(self.config['apikey_file']))
         with open(self.config['apikey_file'], 'w') as outfile: