Merge pull request #14 from aitomatic/feat/implement_logout_cmd

Implement logout command

Author: phamhoangtuan

README.md

@@ -25,7 +25,8 @@ Both the above commands would install the package globally and `aito` will be av
 
 ## How to use
 
-- `aito login`: Login to Aitomatic cloud 
+- `aito login`: Login to Aitomatic cloud
+- `aito logout`: Logout from Aitomatic cloud 
 - `aito deploy app <app_name>`: Deploy app to Aitomatic cluster
 - `aito execute app <app_name>`: Execute app in Aitomatic cluster
 

src/aito.py

@@ -2,7 +2,9 @@
 import json
 from src.execute.main import execute
 from src.deploy.main import deploy
-from src.login.main import login, CREDENTIAL_FILE
+from src.login.main import login
+from src.logout.main import logout
+from src.constants import CREDENTIAL_FILE
 
 
 def load_config():
@@ -23,6 +25,7 @@ def cli():
 
 
 cli.add_command(login)
+cli.add_command(logout)
 cli.add_command(execute)
 cli.add_command(deploy)
 

src/constants.py

@@ -0,0 +1,4 @@
+from pathlib import Path
+
+
+CREDENTIAL_FILE = Path.home().joinpath('.aitomatic/credentials')

src/login/main.py

@@ -4,17 +4,16 @@
 from http.server import HTTPServer
 import time
 import json
-from pathlib import Path
 from functools import update_wrapper, partial
 
 from .server import LoginServer
 from src.utils import get_random_string, create_code_challenger, create_code_verifier
+from src.constants import CREDENTIAL_FILE
 
 ORG = 'aitomaticinc.us.auth0.com'
 CLIENT_ID = 'zk9AB0KtNqJY0gVeF1p0ZmUb2tlcXpYq'
 AUDIENCE = 'https://apps.aitomatic.com/dev'
 SCOPE = 'openid profile email offline_access'
-CREDENTIAL_FILE = Path.home().joinpath('.aitomatic/credentials')
 PORT = 56921
 
 

src/logout/main.py

@@ -0,0 +1,16 @@
+import click
+
+from src.constants import CREDENTIAL_FILE
+
+
+@click.command()
+@click.pass_obj
+def logout(obj):
+    '''Logout from Aitomatic cloud'''
+    if obj.get('access_token') is not None:
+        obj['access_token'] = None
+        obj['refresh_token'] = None
+        obj['id'] = None
+    if CREDENTIAL_FILE.exists():
+        CREDENTIAL_FILE.unlink()
+    click.echo('Logout successfully. Please login to use other commands.')