todo script finalized

Scripts/TodoMD/todo.py

@@ -1,14 +1,24 @@
 import io
-import markdown
+import sys
 from re import search 
+import re
 import os 
+import fileinput
 from datetime import date
 
 #rootdir = "/Users/normrasmussen/Documents/Northpass/CustomerNotes/"
 # This is for testing
-rootdir = "/Users/normrasmussen/Documents/Northpass/Scripts/Confluence_Notes/SampleNotes/"
+rootdir = "/Users/normrasmussen/Documents/Northpass/CustomerNotes/"
+input = sys.argv[1]
+company = input.split("/")[6]
+print(company)
 
-def findCompany(rootdir):
+#def parsefromVim(rootdir, input):
+#    path = input.split("/")
+#    company = path[7]
+#    findcompany(rootdir, company)
+
+def findCompany(rootdir, company):
     files = os.listdir(rootdir)
     for fileName in files:
         if fileName.startswith(".") or fileName.startswith("Todos"):
@@ -19,29 +29,32 @@ def findCompany(rootdir):
 
 def findTodos(rootdir, company):
     todos = []
-    with open(rootdir + company + ".md", "r") as currentfile:
+    with open(rootdir + company, "r") as currentfile:
         file = currentfile.readlines()
         for fullTasks in file:
             if "* [ ] " in fullTasks:
-                tasks = fullTasks[6:]
+                tasks = fullTasks
+                deleteTasks(rootdir, company, tasks)
                 todos.append(tasks)
-                writeTasks(rootdir, company, todos)
+    writeTasks(rootdir, company, todos)
 
 def writeTasks(rootdir, company, todos):
     today = date.today()
     currentDate = today.strftime("%m/%d/%Y")
-    with open(rootdir + "Todos.md", "w") as taskFile:
+    company = company[:-3]
+    todoDir = "/Users/normrasmussen/Documents/Northpass/" 
+    with open(todoDir + "Todos.md", "a") as taskFile:
         for singleTask in todos:
-            taskFile.write("* [ ] (" + company + ") " + "[" + currentDate + "]" + singleTask)
-#    deleteTasks(rootdir, company)
+            task = singleTask.rsplit("* [ ] ")[1]
+            taskFile.write("* [ ] (" + company + ") " + "[" + currentDate + "]  " + task)
 
-def deleteTasks(rootdir, company, todos):
-    with open(rootdir + company, "r") as input:
-        with open(rootdir + company, "w") as currentfile:
-            for task in input:
-                if task.strip("\n") != todos:
-                    currentfile.write(task)
+def deleteTasks(rootdir, company, tasks):
+    file = rootdir + company
+    for line in fileinput.input(file, inplace=1):
+        if tasks in line:
+            line = line.replace(tasks, '')
+        sys.stdout.write(line)
 
 if __name__ == "__main__":
-    findCompany(rootdir)
+    findTodos(rootdir, company)