Update cli.py

Author: RohanBhattaraiNP

caltechdata_api/cli.py

@@ -59,8 +59,14 @@ def decrypt_token(encrypted_token, key):
     return f.decrypt(encrypted_token).decode()
 
 
-# Function to get or set token with support for test system
+# Function to get or set token with support for test systems
 def get_or_set_token(production=True):
+    # First check for environment variable
+    env_token = os.environ.get("CALTECHDATA_TOKEN")
+    if env_token:
+        print("Using token from environment variable")
+        return env_token
+
     key = load_or_generate_key()
 
     # Use different token files for production and test environments
@@ -180,6 +186,56 @@ def get_funding_details():
         "funderIdentifierType": "ROR",
     }
 
+# Add profile handling functions
+def save_profile():
+    profile_file = os.path.join(caltechdata_directory, "profile.json")
+    
+    # Get ORCID
+    while True:
+        orcid = get_user_input("Enter your ORCID identifier: ")
+        family_name, given_name = get_names(orcid)
+        if family_name is not None and given_name is not None:
+            break
+        retry = input("Do you want to try again? (y/n): ")
+        if retry.lower() != "y":
+            return None
+
+    # Get funding details
+    funding_references = []
+    num_funding_entries = get_funding_entries()
+    for _ in range(num_funding_entries):
+        funding_references.append(get_funding_details())
+
+    profile_data = {
+        "orcid": orcid,
+        "family_name": family_name,
+        "given_name": given_name,
+        "funding_references": funding_references
+    }
+
+    with open(profile_file, "w") as f:
+        json.dump(profile_data, f, indent=2)
+    
+    print("Profile saved successfully!")
+    return profile_data
+
+def load_profile():
+    profile_file = os.path.join(caltechdata_directory, "profile.json")
+    try:
+        with open(profile_file, "r") as f:
+            return json.load(f)
+    except FileNotFoundError:
+        return None
+
+def get_or_create_profile():
+    profile = load_profile()
+    if profile:
+        use_saved = input("Use saved profile? (y/n): ").lower()
+        if use_saved == 'y':
+            return profile
+    
+    print("Creating new profile...")
+    return save_profile()
 
 def parse_arguments():
     welcome_message()
@@ -217,23 +273,17 @@ def parse_arguments():
             break
         else:
             print("Invalid input. Please enter a number between 1 and 8.")
+    # Load or create profile
+    profile = get_or_create_profile()
+    if profile:
+        args["orcid"] = profile["orcid"]
+        args["family_name"] = profile["family_name"]
+        args["given_name"] = profile["given_name"]
+        args["fundingReferences"] = profile["funding_references"]
+    else:
+        print("Failed to load or create profile. Exiting.")
+        return None
 
-    while True:
-        orcid = get_user_input("Enter your ORCID identifier: ")
-        family_name, given_name = get_names(orcid)
-        if family_name is not None and given_name is not None:
-            args["orcid"] = orcid
-            break  # Break out of the loop if names are successfully retrieved
-        retry = input("Do you want to try again? (y/n): ")
-        if retry.lower() != "y":
-            print("Exiting program.")
-            return
-    # Optional arguments
-    num_funding_entries = get_funding_entries()
-    funding_references = []
-    for _ in range(num_funding_entries):
-        funding_references.append(get_funding_details())
-    args["fundingReferences"] = funding_references
     return args
 
 
@@ -411,19 +461,24 @@ def parse_args():
 
 def main():
     args = parse_args()
+    production = not args.test
 
-    production = not args.test  # Set production to False if -test flag is provided
-
-    choice = get_user_input(
-        "Do you want to create or edit a CaltechDATA record? (create/edit): "
-    ).lower()
-    if choice == "create":
-        create_record(production)
-    elif choice == "edit":
-        edit_record(production)
-    else:
-        print("Invalid choice. Please enter 'create' or 'edit'.")
-
+    while True:
+        choice = get_user_input(
+            "What would you like to do? (create/edit/profile/exit): "
+        ).lower()
+        
+        if choice == "create":
+            create_record(production)
+        elif choice == "edit":
+            edit_record(production)
+        elif choice == "profile":
+            save_profile()
+        elif choice == "exit":
+            break
+        else:
+            print("Invalid choice. Please enter 'create', 'edit', 'profile', or 'exit'.")
+            
 
 def create_record(production):
     token = get_or_set_token(production)