Fix token file

caltechlibrary · May 1, 2024 · d93ee58 · d93ee58
1 parent f9e2561
commit d93ee58
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/caltechdata_api/cli.py b/caltechdata_api/cli.py
@@ -61,9 +61,9 @@ def decrypt_token(encrypted_token, key):
 # Function to get or set token
 def get_or_set_token():
     key = load_or_generate_key()
+    token_file = os.path.join(caltechdata_directory, "token.txt")
     try:
-        token = os.path.join(caltechdata_directory, "token.txt")
-        with open(token, "rb") as f:
+        with open(token_file, "rb") as f:
             encrypted_token = f.read()
             token = decrypt_token(encrypted_token, key)
             return token
@@ -73,7 +73,7 @@ def get_or_set_token():
             confirm_token = input("Confirm your CaltechDATA token: ").strip()
             if token == confirm_token:
                 encrypted_token = encrypt_token(token, key)
-                with open("token.txt", "wb") as f:
+                with open(token_file, "wb") as f:
                     f.write(encrypted_token)
                 return token
             else: