Add pretty format for command 'get_token'

Cloud-CV · Dec 18, 2019 · f1b9d46 · f1b9d46
1 parent a658e23
commit f1b9d46
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/evalai/get_token.py b/evalai/get_token.py
@@ -1,11 +1,13 @@
 import click
 import os
 
-from click import echo, style
+from click import echo, style, secho
 
 from evalai.utils.config import AUTH_TOKEN_PATH
 import json
 
+from evalai.variables import Colors
+
 
 @click.group(invoke_without_command=True)
 def get_token():
@@ -27,6 +29,7 @@ def get_token():
             try:
                 data = fr.read()
                 tokendata = json.loads(data)
-                echo("Current token is {}".format(tokendata["token"]))
+                echo("Current token is ", nl=False)
+                secho(format(tokendata["token"]), fg=Colors.token)
             except (OSError, IOError) as e:
                 echo(e)
diff --git a/evalai/variables.py b/evalai/variables.py
@@ -0,0 +1,3 @@
+class Colors:
+    links = 'blue'
+    token = 'green'