From ffaf6b43a5297dcc6d9def2988c46d7678d4d68c Mon Sep 17 00:00:00 2001 From: MOIS3Y Date: Tue, 20 Sep 2022 21:46:17 +0900 Subject: [PATCH] Add main function and progress messages with access link --- main.py | 19 +++++++++++++++++-- settings.py | 1 + 2 files changed, 18 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 55856f7..74dc79e 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,7 @@ import requests from models_vm import VmAccount from models_dci import AuthUser -from settings import PLATFORM +from settings import CLIENT_HOST, PLATFORM class Access(object): @@ -46,9 +46,24 @@ class Access(object): return result -if __name__ == "__main__": +def main(): + """ + app entrypoint + """ access = Access(PLATFORM) + print("Getting an admin account, please waiting....") account = access.get_admin() + print("Account will be used:") print(account) + print("Getting an access key, please waiting...") key = access.get_key(account) + print("Key will be used:") print(key) + print("Your access link:") + access_link = 'https://{}/auth/key/{}'.format(CLIENT_HOST, key['key']) + print(access_link) + print("ENJOY MY FRIEND!") + + +if __name__ == "__main__": + main() diff --git a/settings.py b/settings.py index e52456e..128fdcb 100644 --- a/settings.py +++ b/settings.py @@ -3,6 +3,7 @@ import json from peewee import MySQLDatabase +CLIENT_HOST = os.environ.get("CLIENT_HOST", "replace_to_client_host") PLATFORM = os.environ.get("PLATFORM", "vm") # or dci DB_ENGINE = os.environ.get("DB_ENGINE", "mysql")