Skip to content

Commit

Permalink
Remove ANSI escape sequences
Browse files Browse the repository at this point in the history 
- This removes the ANSI escape sequences from gen_list.py and gen_token.py files since Windows doesn't support it by default
  • Loading branch information
@l3v11
l3v11 committed May 30, 2022
1 parent feb9cb7 commit 3132d9d
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 43 deletions.
76 changes: 36 additions & 40 deletions gen_list.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,26 +4,26 @@
import time

print("\nGenerate a file from the below list")
print("\n\t\033[1;95mA.\033[m drive_list")
print("\t\033[1;95mB.\033[m dest_list")
print("\n\tA. drive_list")
print("\tB. dest_list")

choice = input("\nChoose either \033[1;95mA\033[m or \033[1;95mB\033[m > ")
choice = input("\nChoose either A or B > ")

if choice == 'A' or choice == 'a' or choice == '1':
print("\nGenerate the \033[1;96mdrive_list\033[m file with the below methods")
print("\n\t\033[1;95mA.\033[m All Drives (automatic)")
print("\t\033[1;95mB.\033[m Selected Drives (manual)")
if choice in ['A', 'a', '1']:
print("\nGenerate the drive_list file with the below methods")
print("\n\tA. All Drives (automatic)")
print("\tB. Selected Drives (manual)")

choice = input("\nChoose either \033[1;95mA\033[m or \033[1;95mB\033[m > ")
choice = input("\nChoose either A or B > ")

if choice == 'A' or choice == 'a' or choice == '1':
input("\n\033[1;95mNOTICE:\033[m Make sure Rclone is installed on your system PATH variable and Google Drive remote is configured properly\n\nPress ENTER to continue")
if choice in ['A', 'a', '1']:
input("\nNOTICE: Make sure Rclone is installed on your system PATH variable and Google Drive remote is configured properly\n\nPress ENTER to continue")

print("\n\033[1;93mList of remotes")
print("---------------\033[m")
print("\nList of remotes")
print("---------------")
subprocess.run(['rclone', 'listremotes', '--long'])

remote = input("\n\033[1;94mEnter a drive remote >\033[m ")
remote = input("\nEnter a drive remote > ")

print("\nProcessing all drives")

Expand All @@ -45,89 +45,85 @@
time.sleep(2)

os.remove('drives.txt')
print(f"\nGenerated \033[1;96mdrive_list\033[m file with \033[1;96m{len(lines)}\033[m drives")
print(f"\nGenerated the drive_list file with {len(lines)} drives")
exit()

elif choice == 'B' or choice == 'b' or choice == '2':
print("\n\033[1;93mInstructions" \
"\n------------\033[m" \
elif choice in ['B', 'b', '2']:
print("\nInstructions" \
"\n------------" \
"\nDrive Name > Name of the drive" \
"\nDrive ID > ID of the drive" \
"\nIndex URL > Index link for the drive (Optional)")

num = int(input("\n\033[1;94mTotal number of drives >\033[m "))
count = 1
num = int(input("\nTotal number of drives > "))
msg = ''
while count <= num:
print(f"\n\033[1;93mDRIVE - {count}\n" \
f"----------\033[m")
for count in range(1, num + 1):
print(f"\nDRIVE - {count}\n" \
f"----------")
name = input("Drive Name > ")
if not name:
print("\n\033[1;91mERROR:\033[m Drive Name cannot be empty")
print("\nERROR: Drive Name cannot be empty")
exit(1)
name = name.replace(" ", "_")
id = input("Drive ID > ")
if not id:
print("\n\033[1;91mERROR:\033[m Drive ID cannot be empty")
print("\nERROR: Drive ID cannot be empty")
exit(1)
index = input("Index URL > ")
if index:
if index[-1] == "/":
index = index[:-1]
else:
index = ''
count += 1
msg += f"{name} {id} {index}\n"

with open('drive_list', 'w') as f:
f.truncate(0)
f.write(msg)

print(f"\nGenerated \033[1;96mdrive_list\033[m file with \033[1;96m{num}\033[m drives")
print(f"\nGenerated the drive_list file with {num} drives")
exit()

else:
print("\n\033[1;91mERROR:\033[m Wrong input")
print("\nERROR: Wrong input")
exit(1)

elif choice == 'B' or choice == 'b' or choice == '2':
print("\n\033[1;93mInstructions" \
"\n------------\033[m" \
elif choice in ['B', 'b', '2']:
print("\nInstructions" \
"\n------------" \
"\nDrive Key > A random short name for the drive" \
"\nDrive ID > ID of the drive" \
"\nIndex URL > Index link for the drive (Optional)")

num = int(input("\n\033[1;94mTotal number of drives >\033[m "))
count = 1
num = int(input("\nTotal number of drives > "))
msg = ''
while count <= num:
print(f"\n\033[1;93mDRIVE - {count}\n" \
f"----------\033[m")
for count in range(1, num + 1):
print(f"\nDRIVE - {count}\n" \
f"----------")
key = input("Drive Key > ")
if not key:
print("\n\033[1;91mERROR:\033[m Drive Key cannot be empty")
print("\nERROR: Drive Key cannot be empty")
exit(1)
key = key.replace(" ", "_")
id = input("Drive ID > ")
if not id:
print("\n\033[1;91mERROR:\033[m Drive ID cannot be empty")
print("\nERROR: Drive ID cannot be empty")
exit(1)
index = input("Index URL > ")
if index:
if index[-1] == "/":
index = index[:-1]
else:
index = ''
count += 1
msg += f"{key} {id} {index}\n"

with open('dest_list', 'w') as f:
f.truncate(0)
f.write(msg)

print(f"\nGenerated \033[1;96mdest_list\033[m file with \033[1;96m{num}\033[m drives")
print(f"\nGenerated the dest_list file with {num} drives")
exit()

else:
print("\n\033[1;91mERROR:\033[m Wrong input")
print("\nERROR: Wrong input")
exit(1)
6 changes: 3 additions & 3 deletions gen_token.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
# time.
if os.path.exists('token.json'):
creds = Credentials.from_authorized_user_file('token.json', SCOPES)
print("\n\033[1;96mtoken.json\033[m file exists")
print("\nThe token.json file exists")

# If the file token.pickle exists and token.json doesn't exist, then
# let it be converted to token.json
Expand All @@ -23,7 +23,7 @@
creds = pickle.load(f)
with open('token.json', 'w') as token:
token.write(creds.to_json())
print("\nConverted \033[1;94mtoken.pickle\033[m file to \033[1;96mtoken.json\033[m")
print("\nConverted the token.pickle file to token.json")

# If there are no (valid) credentials available, let the user log in.
if not creds or not creds.valid:
Expand All @@ -36,4 +36,4 @@
# Save the credentials for the next run
with open('token.json', 'w') as token:
token.write(creds.to_json())
print("\nGenerated \033[1;96mtoken.json\033[m file")
print("\nGenerated the token.json file")

0 comments on commit 3132d9d

Please sign in to comment.