From 5ec6daad560c02cdae446400f1efef088ee46673 Mon Sep 17 00:00:00 2001
From: Graham Helton <no@email.com>
Date: Sat, 5 Mar 2022 19:09:27 -0500
Subject: [PATCH] Files now stored in ~/.Indexes

---
 README.md |  2 +-
 index.py  | 87 +++++++++++++++++++++++++++++++++----------------------
 2 files changed, 54 insertions(+), 35 deletions(-)

diff --git a/README.md b/README.md
index b8f2cb8..3e70bc8 100644
--- a/README.md
+++ b/README.md
@@ -22,7 +22,7 @@ Currently the indexer has the following features
 - ~~Only show CSVs when asking which index you want to work on~~
 - ~~Add "Create new" option to csv selection option~~
 ![](createCsv.gif)
-- Create a .sansindex file in home directory that stores indexes
+- ~~Create a .index file in home directory that stores indexes~~
 - Add delete option for index files
 - ~~Add "ask for book" prompt on startup~~
 - ~~Allow for input with no definition (term, pagenumber)~~
diff --git a/index.py b/index.py
index dd9f5cd..cbff213 100755
--- a/index.py
+++ b/index.py
@@ -1,6 +1,7 @@
 #! /bin/python3
 import csv
 import os
+from os.path import expanduser
 import signal 
 import pandas as pd
 from colorama import Fore, Style
@@ -8,7 +9,7 @@ from colorama import Fore, Style
 headers = ['Title', 'Description','Page','book']
 index = []
 row = []
-dic = {}
+dictionary = {}
 # Define formatting
 reset = Style.RESET_ALL
 red = Fore.RED
@@ -25,6 +26,14 @@ tickInfo = Fore.BLUE + "[" + yellow + "!" + Fore.BLUE + "]" + reset
 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")
+def path():
+    home = expanduser("~")
+    path = '/.Indexes/'
+    if not os.path.exists(home+path):
+        print("~/.Indexes Does not exist! Creating .Index directory in your home folder!")
+        os.mkdir(home+path)
+
+    os.chdir(home+path)
 
 # Handle control+c
 def handler(signum, frame):
@@ -34,53 +43,63 @@ 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():
+    path()
     count = 0
     csvCheck = os.listdir()
-    for i in csvCheck:
-        if any(".csv" in c for c in csvCheck):
-            print(tickInfo,"CSV Exists in current directory, which would you like to use? ")
-
-            for x in os.listdir():
-                if x.endswith((".csv")):
-                    count+=1
-                    dic[count]=x
-            print(sep)
-
-            # 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 key, value in dic.items():
-                print(green, key, red,'-->', cyan , value, reset)
-            print(sep)
-            print(green, "0", red,  "-->" + cyan +  "  New index")
-            selection = input(tick + green + " Type the number of the file you would like\n" + reset)
-            # If selecing existing csv, exit function
-            if selection != "0":
-                csvName = dic[int(selection)]
-                print(tick, "You chose: ", csvName)
-                return csvName 
-            # If selection is to create a new CSV, carry on with creating it
-        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"):
-                csvName += ".csv"
-            with open(csvName, 'a') as f:
-                write = csv.writer(f)
-                write.writerow(headers)
-
-            print(csvName)
-            return csvName
+
+    # 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):
+        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")
+
+        # Ask the user which file they would like
+        selection = input(tick + green + " Type the number of the file you would like\n" + reset)
+
+        # 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 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"):
+            csvName += ".csv"
+
+        with open(csvName, 'a') as f:
+            write = csv.writer(f)
+            write.writerow(headers)
+
+        print(csvName)
+        return csvName
 
 def writeCsv():
     with open(csvName, 'a') as f:
         write = csv.writer(f)
         write.writerows(index)
+
 def readIndex(csvName):
     print(tick, "Displaying abbreviated entries for",cyan, csvName,reset)
     df = pd.read_csv(csvName)
     #bottom = df.tail()
     print(df)
 
+# Call init function and assign csvName to a string
 csvName = str(init())
+
 # Get user input
 while True:
     # Set index and row to null