PyKubeGrader 0.2.17__tar.gz → 0.2.19__tar.gz

{pykubegrader-0.2.17/src/PyKubeGrader.egg-info → pykubegrader-0.2.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyKubeGrader
-Version: 0.2.17
+Version: 0.2.19
 Summary: Add a short description here!
 Home-page: https://github.com/pyscaffold/pyscaffold/
 Author: jagar2

{pykubegrader-0.2.17 → pykubegrader-0.2.19/src/PyKubeGrader.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyKubeGrader
-Version: 0.2.17
+Version: 0.2.19
 Summary: Add a short description here!
 Home-page: https://github.com/pyscaffold/pyscaffold/
 Author: jagar2

{pykubegrader-0.2.17 → pykubegrader-0.2.19}/src/pykubegrader/tokens/validate_token.py

@@ -1,5 +1,5 @@
 import requests
-
+import os
 
 class TokenValidationError(Exception):
     """
@@ -16,7 +16,7 @@ class TokenValidationError(Exception):
         super().__init__(message)
 
 
-def validate_token(token):
+def validate_token(token = None):
     """
     Validate a token by making a GET request to the validation endpoint.
 
@@ -29,7 +29,19 @@ def validate_token(token):
     Returns:
         None: If the token is valid, the function will pass silently.
     """
+    # sets the token in the environment variable
+    if token is not None:
+        os.environ["TOKEN"] = token
+
+    if token is None:
+        token = os.getenv("TOKEN", None)
+
+    if token is None:
+        raise TokenValidationError("No token provided")
+
+    # Define the validation endpoint
     endpoint = f"https://engr-131-api.eastus.cloudapp.azure.com/validate-token/{token}"
+
     try:
         response = requests.get(endpoint, timeout=10)