Thomas Riggs

#!/usr/bin/env python
#Filename: wc.py
#Makes a word count for the previous day.
#First version. Written by Thomas Riggs and released into the public domain. www.twostopsdown.com

import os, sys # to work with paths and files.
from datetime import date, timedelta # to work with dates.
import pickle # to save objects.

# Configuration
default_threshold = timedelta(days=1) # The default threshold for what is considered recent, as a timedelta.
default_path = os.path.expanduser('~/Writing') # Default path to search.

# Functions
def findRecentFiles(path = default_path, threshold = default_threshold):
    "This function recurses through a folder, and returns a list of files modified within the threshold value."
    cutoff = date.today() - threshold
    if path == None or threshold == None:
        print("findRecentFiles requires a path and threshold.")
        return void

    recentfiles = []

    for file in os.listdir(path):
        if file[0] == ".":
            continue

        filepath = path + "/" + file
        if date.fromtimestamp(os.path.getmtime(filepath)) > cutoff and os.path.isfile(filepath) == True and file[-3:] == 'txt':
            recentfiles.append(filepath)
        elif os.path.isdir(filepath) == True:
            for item in findRecentFiles(filepath):
                recentfiles.append(item)

    return recentfiles

def countWords(files = []):
    "This function returns a dictionary of files and their current word count."
    if files == None:
        print("countWords requires a list of file paths.")
        return void

    wordcounts = {}

    for file in files:
        words = []
        data = open(file)
        lines = data.readlines()
        for line in lines:
              for word in line.split(None):
                  words.append(word)

        wordcounts[file] = len(words)

    return wordcounts

def writeWordCounts(wordcounts = {}, path = default_path):
    "Writes the given wordcounts object using Pickle."
    wordcountfile = path + "/" + ".wordcount"
    data = open(wordcountfile, 'wb')
    pickle.dump(wordcounts, data)
    data.close()

def loadWordCounts(path = default_path):
    "Uses Pickle to open a path's wordcount object."
    wordcountfile = path + "/" + ".wordcount"
    data = open(wordcountfile, 'rb')
    wordcounts = pickle.load(data)
    return wordcounts
    data.close()

def updateWordCounts(old = {}, new = {}):
    "Combines an old set of word counts with a current one. Replaces existing file's data and keeps unupdated ones."
    updated = old
    updated.update(new)
    writeWordCounts(updated)

def relativePath(path, root = default_path):
    "Gives the given file path, minus a starting root, by counting the directory tree branches of the root. Assumes path is within root."
    rootlength = len(root.split("/"))
    relativepathparts = path.split("/")
    relativepath = "/".join(relativepathparts[rootlength:])
    return relativepath

def currentWordCounts(path):
    "Chains together findRecentFiles and countWords into a single function."
    return countWords(findRecentFiles(path))

def recentChanges(raw = False, path = default_path):
    "This function takes a path and finds recent word counts for files, using currentWordCounts."
    currentwordcount = currentWordCounts(path)
    oldwordcount = loadWordCounts()

    filecount = 0
    totalwordcount = 0

    for (file, count) in currentwordcount.items():
        try:
            newwords = count - oldwordcount[file]
            if newwords == 0:
                continue
            status = str(newwords) + " new words since yesterday."
            totalwordcount = totalwordcount + newwords
        except KeyError:
            status = str(count) + " words in total, but no previous record."
            totalwordcount = totalwordcount + count

        if raw == False:
            print(relativePath(file), "has", status)

        filecount = filecount + 1

    if raw == True:
        print(totalwordcount)
        return

    if filecount == 0 or totalwordcount == 0:
        print("No updates today!")
    else:
        print("In total,", str(totalwordcount), "words have been written today!")

# On with the show!

try:
    arg = sys.argv[1]
except IndexError:
    recentChanges()
    sys.exit()

if arg == "update":
    updateWordCounts(loadWordCounts(), currentWordCounts())
elif arg == "raw":
    recentChanges(True)