diff --git a/main.py b/main.py index 74dc79e..ead97bc 100644 --- a/main.py +++ b/main.py @@ -31,9 +31,13 @@ class Access(object): ) for field in query_admin: if field.roles[0] == '@admin': - admin = dict(id=field.id, email=field.email, state=field.state) + admin = dict( + id=field.id, + email=field.email, + state=field.state, + role=field.roles[0] + ) break - return admin def get_key(self, admin): @@ -51,18 +55,25 @@ def main(): app entrypoint """ access = Access(PLATFORM) - print("Getting an admin account, please waiting....") + print("="*47) + print("Getting an admin account, please waiting....\n") + account = access.get_admin() + print("Account will be used:") - print(account) - print("Getting an access key, please waiting...") + print(account, "\n") + print("Getting an access key, please waiting...\n") + key = access.get_key(account) + print("Key will be used:") - print(key) - print("Your access link:") + print(key, "\n") + access_link = 'https://{}/auth/key/{}'.format(CLIENT_HOST, key['key']) - print(access_link) - print("ENJOY MY FRIEND!") + + print("Your access link: ---> ", access_link, "\n") + print("ENJOY MY FRIEND! \n") + print("="*47) if __name__ == "__main__":