X-Git-Url: http://unremediatedgender.space/source?a=blobdiff_plain;f=notes%2Fmemoir_wordcounts.py;h=bfe63c758e73ba115926cc284ce202de48d00022;hb=80d3614935a69941a76651ab814f2835ce6fdd5b;hp=eeb297acb07ed347c826147725175b9104bbbb20;hpb=be3d61e0a36e366e3985e20c58ef51225aebf976;p=Ultimately_Untrue_Thought.git

diff --git a/notes/memoir_wordcounts.py b/notes/memoir_wordcounts.py
index eeb297a..bfe63c7 100755
--- a/notes/memoir_wordcounts.py
+++ b/notes/memoir_wordcounts.py
@@ -38,14 +38,27 @@ def look_back():
     wordcounts = []
     keep_going = True
     while keep_going:
-        subprocess.run(["git", "checkout", "HEAD~10"])
+        subprocess.run(["git", "checkout", "HEAD~1"])
         wordcount = wordcount_at_this_sha()
         date = date_at_this_sha()
         if date < datetime.date(2022, 4, 1):
             keep_going = False
         wordcounts.append((date, wordcount))
+    # don't leave the head detached
+    subprocess.run(["git", "checkout", "master"])
     return sorted(wordcounts)
 
+def normalize_dates(wordcounts):
+    eod_wordcounts = []
+    for i in range(len(wordcounts)-1):
+        date, wordcount = wordcounts[i]
+        next_date, next_wordcount = wordcounts[i+1]
+        gap = next_date - date
+        if gap >= datetime.timedelta(1):
+            eod_wordcounts.append((date, wordcount))
+            for i in range(gap.days-1):
+                eod_wordcounts.append((date + datetime.timedelta(i+1), wordcount))
+    return eod_wordcounts
 
 def write_csv(wordcounts):
     with open("memoir_wordcounts.csv", 'w') as f:
@@ -55,5 +68,5 @@ def write_csv(wordcounts):
 
 
 if __name__ in "__main__":
-    wordcounts = look_back()
+    wordcounts = normalize_dates(look_back())
     write_csv(wordcounts)