Fixed logical flow error

main
Graham Helton 2 years ago
parent 5ec6daad56
commit 593c6c0fc1

@ -23,6 +23,7 @@ tickBad = Fore.YELLOW + "[" + Fore.RED + "X" + Fore.YELLOW+ "]" + reset
awaitingInput = Fore.BLUE + "-----" + Fore.GREEN + "> " + reset
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")
@ -37,18 +38,23 @@ def path():
# Handle control+c
def handler(signum, frame):
os.system("clear")
print(red, "Exiting")
exit(0)
signal.signal(signal.SIGINT, handler)
# Kick off program, check if any .csv files are in directory, if not, prompt user to create one
def init():
os.system("clear")
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
if any(".csv" in c for c in csvCheck):
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.
@ -72,8 +78,6 @@ def init():
print(tick, "You chose: ", csvName)
return csvName
# If selection is not to select a current CSV, create a new one
else:
csvName = input(tick + green + " Creating a new index, what do you want to name your CSV file?\n" + reset)
if not csvName.endswith(".csv"):
@ -83,7 +87,8 @@ def init():
write = csv.writer(f)
write.writerow(headers)
print(csvName)
print(tick, "Creating file named: ", csvName)
os.system("clear")
return csvName
def writeCsv():
@ -92,7 +97,8 @@ def writeCsv():
write.writerows(index)
def readIndex(csvName):
print(tick, "Displaying abbreviated entries for",cyan, csvName,reset)
os.system("clear")
print(tick, "Displaying entries for",cyan, csvName,reset)
df = pd.read_csv(csvName)
#bottom = df.tail()
print(df)

Loading…
Cancel
Save