Multiple fixes and changes

1. Shouldn't error on non posix (linux)/nt (windows) systems when clearing screen (in theory)
2. Added shortcuts when typing in required variables
3. Added revision cloning
4. Grammar fixes
5. Updated confirmation message
6. Check if running in Kaggle environment (Not getting token secret yet)
7. Check if HF_TOKEN is set to read only then exit

manage branches.py

@@ -3,8 +3,14 @@ import os
 from huggingface_hub import create_branch, delete_branch, login, get_token, whoami
 
 #define clear screen function
+if os.name == 'nt':
+    osclear = 'cls'
+elif os.name == 'posix':
+    osclear = 'clear'
+else:
+    osclear = ''
 def clear_screen():
-    os.system('cls' if os.name == 'nt' else 'clear')
+    os.system(osclear)
 
 #clear screen before starting
 clear_screen()
@@ -14,28 +20,44 @@ clear_screen()
 while True:
     cord = input("What would you like to do? (create) (delete): ").lower()
     
-    if cord not in ['create', 'delete']:
+    if cord not in ['create', 'delete', 'c', 'd']:
         clear_screen()
         print("Please choose one of the following two options.")
         continue
+    if cord == 'c':
+        cord = 'create'
+    elif cord == 'd':
+        cord = 'delete'
     break
 clear_screen()
-#name of effected repository
+#name of affected repository
 repo = input("Repository name (User/Repo): ")
 clear_screen()
 #type of huggingface repository (restricted)
 while True:
     r_type = input("Repo type (model) (dataset) (space): ").lower()
     
-    if r_type not in ['model', 'dataset', 'space']:
+    if r_type not in ['model', 'dataset', 'space', 'm', 'd', 's']:
         clear_screen()
         print("Please choose one of the following three options.")
         continue
+    if r_type == 'm':
+        r_type = 'model'
+    elif r_type == 'd':
+        r_type = 'dataset'
+    elif r_type == 's':
+        r_type = 'space'
     break
 clear_screen()
 #name of created or deleted branch
 branch = input("Branch name (No spaces): ")
 clear_screen()
+#promt user for revision, or clone from main
+if cord == 'create':
+    rev = input("Revision to clone from (Can be a branch name or the OID/SHA of a commit) (Empty clones main): ")
+    if rev == '':
+        rev = 'main'
+clear_screen()
 
 #get token
 if get_token() is not None:
@@ -43,6 +65,12 @@ if get_token() is not None:
     login(get_token())
     tfound = "Where are my doritos?"
 else:
+    if os.environ.get('KAGGLE_KERNEL_RUN_TYPE', None) is not None: #check if user in kaggle
+        print('''
+              When using Kaggle, make sure to use the secret key HF_TOKEN with a 'WRITE' token.
+              This will prevent the need to login every time you run the script.
+              Set your secrets with the secrets add-on on the top of the screen.
+              ''')
     #if the token is not found then prompt user to provide it:
     login(input("API token not detected. Enter your HuggingFace (WRITE) token: "))
     tfound = "false"
@@ -51,33 +79,67 @@ else:
 while True:
     if whoami().get('auth', {}).get('accessToken', {}).get('role', None) != 'write':
         clear_screen()
+        if os.environ.get('HF_TOKEN', None) is not None: #if environ finds HF_TOKEN as write then display following text and exit:
+            print(f'''
+          You have the environment variable HF_TOKEN set.
+          You cannot log in.
+          Either set the environment variable to a (WRITE) token or remove it.
+          ''')
+            exit()
+            
+        if os.environ.get('KAGGLE_KERNEL_RUN_TYPE', None) is not None: #check if user in kaggle
+            print('''
+                  When using Kaggle, make sure to use the secret key HF_TOKEN with a 'WRITE' token.
+                  This will prevent the need to login every time you run the script.
+                  Set your secrets with the secrets add-on on the top of the screen.
+                  ''')
         print("You do not have write access to this repository. Please use a valid token with (WRITE) access.")
         login(input("Enter your HuggingFace (WRITE) token: "))
         continue
     break
 
 clear_screen()
-#prompt the user for confirmation on creation/deletion of the branch
-while True:
-    yorn = input(f"Are you sure you want to {cord} branch '{branch}' in {repo} (Y/n): ").lower()
-    if yorn == '':
-        yorn = 'y'
+if cord == 'delete':
+    #prompt the user for confirmation on deletion of the branch
+    while True:
+        yorn = input(f"Are you sure you want to remove branch '{branch}' in {repo}? (Y/n): ").lower()
+        if yorn == '':
+            yorn = 'y'
+            break
+        else:
+            if yorn not in ['y', 'n']:
+                clear_screen()
+                print("Please choose one of the following two options carefully.")
+                continue
+        break
+else:
+    #prompt the user for confirmation on creation of the branch
+    while True:
+        yorn = input(f"Are you sure you want to clone revision '{rev}' to create branch '{branch}' in {repo}? (Y/n): ").lower()
+        if yorn == '':
+            yorn = 'y'
+        elif yorn == 'yes':
+            yorn = 'y'
+        elif yorn == 'no':
+            yorn = 'n'
+            break
+        else:
+            if yorn not in ['y', 'n']:
+                clear_screen()
+                print("Please choose one of the following two options carefully.")
+                continue
         break
-    else:
-        if yorn not in ['y', 'n']:
-            clear_screen()
-            print("Please choose one of the following two options carefully.")
-            continue
-    break
 clear_screen()
 
 #create or delete the branch
+#if user selected yes then continue, else exit
 if yorn == 'y':
     if cord == 'create':
-        create_branch(repo, repo_type=r_type, branch=branch)
+        create_branch(repo, revision=rev, repo_type=r_type, branch=branch)
     else:
         delete_branch(repo, repo_type=r_type, branch=branch)
 else:
+    print("Cancelled action")
     exit()
 clear_screen()
 
@@ -97,7 +159,7 @@ else:
         print(f"Branch {branch} deleted on {r_type} https://huggingface.co/datasets/{repo}")
     elif r_type == 'space':
         print(f"Branch {branch} deleted on {r_type} https://huggingface.co/spaces/{repo}")
-#if token wasn't found from line 36 then display following text:
+#if token wasn't found from line 60 then display following text:
 if tfound == 'false':
     print(f'''
           You are now logged in as {whoami().get('fullname', None)}.
@@ -105,4 +167,5 @@ if tfound == 'false':
           To logout, use the hf command line interface 'huggingface-cli logout'
           To view your active account, use 'huggingface-cli whoami'
           ''')
+
 input("Press enter to continue.")