Add token collation using server-obtained token

evalai/add_token.py

@@ -7,6 +7,7 @@
 from click import echo, style
 
 from evalai.utils.config import AUTH_TOKEN_DIR, AUTH_TOKEN_PATH, LEN_OF_TOKEN
+from evalai.utils.auth import validate_user_auth_token_by_profile
 
 
 @click.group(invoke_without_command=True)
@@ -16,11 +17,12 @@ def set_token(auth_token):
     Configure EvalAI Token.
     """
     """
-    Invoked by `evalai settoken <your_evalai_auth_token>`.
+    Invoked by `evalai set_token <your_evalai_auth_token>`.
     """
     if validators.length(auth_token, min=LEN_OF_TOKEN, max=LEN_OF_TOKEN):
         if not os.path.exists(AUTH_TOKEN_DIR):
             os.makedirs(AUTH_TOKEN_DIR)
+        validate_user_auth_token_by_profile(auth_token)
         with open(AUTH_TOKEN_PATH, "w+") as fw:
             try:
                 auth_token = {"token": "{}".format(auth_token)}  # noqa

evalai/utils/auth.py

@@ -76,6 +76,37 @@ def get_user_auth_token():
         sys.exit(1)
 
 
+def validate_user_auth_token_by_profile(auth_token):
+    """
+    Checks tokens availability via server.
+    """
+    url = "{}{}".format(get_host_url(), URLS.profile.value)
+    try:
+        headers = {"Authorization": "Token {}".format(auth_token)}
+        response = requests.post(url, headers=headers)
+        response.raise_for_status()
+    except requests.exceptions.HTTPError:
+        if response.status_code in EVALAI_ERROR_CODES:
+            echo(
+                style(
+                    "Failed: Could not make a authentication with this token.",
+                    bold=True,
+                    fg="red"
+                )
+            )
+            sys.exit(1)
+    except requests.exceptions.RequestException:
+        echo(
+            style(
+                "\nCould not establish a connection to EvalAI."
+                " Please check the Host URL.\n",
+                bold=True,
+                fg="red"
+            )
+        )
+        sys.exit(1)
+
+
 def get_request_header():
     """
     Returns user auth token formatted in header for sending requests.

evalai/utils/urls.py

@@ -3,6 +3,7 @@
 
 class URLS(Enum):
     login = "/api/auth/login"
+    profile = "/api/auth/user"
     challenge_list = "/api/challenges/challenge/all"
     past_challenge_list = "/api/challenges/challenge/past"
     future_challenge_list = "/api/challenges/challenge/future"