Updated UI and fixed bugs

main
Graham Helton 1 year ago
parent a566821a81
commit 0969ca1fa7

@ -9,8 +9,7 @@ from colorama import Fore, Style
headers = ['Title', 'Page','book']
index = []
row = []
dictionary = {}
# Define formatting
dictionary = {} # Define formatting
reset = Style.RESET_ALL
red = Fore.RED
cyan = Fore.CYAN
@ -26,7 +25,7 @@ tickInfo = Fore.BLUE + "[" + yellow + "!" + Fore.BLUE + "]" + reset
os.system("clear")
print(tick, "Welcome to Sans Terminal Indexer")
print(tick, "Enter" , green, "\"new book\"" , reset, "to change to the next book")
book = input("Which book are you currently working on?\n")
book = input("Which book number are you currently working on?\n")
def path():
home = expanduser("~")
path = '/.Indexes/'
@ -49,36 +48,32 @@ def init():
path()
count = 0
csvCheck = os.listdir()
print(csvCheck)
# If a .csv file exists in the current directory (~.Indexes) then prompt user for which one they would like to use
print(tickInfo,"CSV Exists in current directory, which would you like to use? ")
for i in csvCheck:
print("printing elements", i)
if (i.endswith(".csv")):
print(tickInfo,"CSV Exists in current directory, which would you like to use? ")
# For each file that ends in .csv, add to dictionary, then for each value, assign a key from 1-n.
# This will allow the user to select a number that corresponds to the csv file.
for x in os.listdir():
if x.endswith((".csv")):
count+=1
dictionary[count]=x
print(sep)
# For each key and value, in the dictionary, print key, value for user to select from
for key, value in dictionary.items():
print(green, key, red,'-->', cyan , value, reset)
print(sep)
print(green, "0", red, "-->" + cyan + " New index")
if i.endswith((".csv")):
count+=1
dictionary[count]=i
# For each key and value, in the dictionary, print key and value for user to select from
for key, value in dictionary.items():
print(green, key, red,'-->', cyan , value, reset)
print(sep)
print(green, "0", red, "-->" + cyan + " New index")
# Ask the user which file they would like
selection = input(tick + green + " Type the number of the file you would like\n" + reset)
os.system("clear")
# Ask the user which file they would like
selection = input(tick + green + " Type the number of the file you would like\n" + reset)
os.system("clear")
# If selecing existing csv, set csvName to choice and exit function
if selection != "0":
csvName = dictionary[int(selection)]
print(tick, "You chose: ", csvName)
return csvName
# If selecing existing csv, set csvName to choice and exit function
if selection != "0":
csvName = dictionary[int(selection)]
print(tick, "You chose: ", csvName)
return csvName
csvName = input(tick + green + " Creating a new index, what do you want to name your CSV file?\n" + reset)
@ -100,7 +95,7 @@ def writeCsv():
write.writerows(index)
def readIndex(csvName):
print(tick, "Displaying entries for",cyan, csvName,reset)
print(tick, "Displaying entries for",cyan,csvName,reset)
df = pd.read_csv(csvName)
#bottom = df.tail()
print(df)
@ -114,7 +109,15 @@ while True:
index = []
row = []
# Get user input
newEntry = input(tick + green+" Enter your index entry. Format: Term,Page number\n" + reset)
print(green + "SANS Terminal Indexer")
print(sep)
print(green + "index ->" + purple + " View your current index")
print(green+ "new book ->" + purple + " Change the book number you're working on")
print(green + "exit ->" + purple + " exit the program")
print(sep)
readIndex(csvName)
print(sep)
newEntry = input(tick + green+" Enter your index entry. Format: Term,Page number\n" + purple +"STI~> " + yellow)
# Handle input
if newEntry == "exit":
writeCsv()

Loading…
Cancel
Save