import io 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/CustomerNotes/" input = sys.argv[1] company = input.split("/")[6] print(company) #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"): pass else: company = fileName[:-3] findTodos(rootdir, company) def findTodos(rootdir, company): todos = [] with open(rootdir + company, "r") as currentfile: file = currentfile.readlines() for fullTasks in file: if "* [ ] " in fullTasks: tasks = fullTasks deleteTasks(rootdir, company, tasks) todos.append(tasks) writeTasks(rootdir, company, todos) def writeTasks(rootdir, company, todos): today = date.today() currentDate = today.strftime("%m/%d/%Y") company = company[:-3] todoDir = "/Users/normrasmussen/Documents/Northpass/" with open(todoDir + "Todos.md", "a") as taskFile: for singleTask in todos: task = singleTask.rsplit("* [ ] ")[1] taskFile.write("* [ ] (" + company + ") " + "[" + currentDate + "] " + 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__": findTodos(rootdir, company)