retry wait for mailbox if user doesn't exist

This commit is contained in:
Jay Lee
2021-07-07 11:03:04 -04:00
parent 0418438b6f
commit c7c1bfbeba

View File

@@ -3,6 +3,7 @@ from time import sleep
import gam import gam
from gam import gapi from gam import gapi
from gam.gapi import directory as gapi_directory from gam.gapi import directory as gapi_directory
from gam.gapi import errors as gapi_errors
def get_primary(email): def get_primary(email):
@@ -53,10 +54,16 @@ def wait_for_mailbox(users):
i += 1 i += 1
user = gam.normalizeEmailAddressOrUID(user) user = gam.normalizeEmailAddressOrUID(user)
while True: while True:
result = gapi.call(cd.users(), try:
'get', result = gapi.call(cd.users(),
'fields=isMailboxSetup', 'get',
userKey=user) 'fields=isMailboxSetup',
userKey=user,
throw_reasons=[gapi_errors.ErrorReason.USER_NOT_FOUND])
except gapi_errors.GapiUserNotFoundError:
print(f'{user} mailboxIsSetup: False (user does not exist yet)')
sleep(3)
continue
mailbox_is_setup = result.get('isMailboxSetup') mailbox_is_setup = result.get('isMailboxSetup')
print(f'{user} mailboxIsSetup: {mailbox_is_setup}') print(f'{user} mailboxIsSetup: {mailbox_is_setup}')
if mailbox_is_setup: if mailbox_is_setup: