From 7c9feab1b247220a0fe4d55ff0fba6d79b89e5af Mon Sep 17 00:00:00 2001
From: Craig Oates <craig@craigoates.net>
Date: Sat, 8 May 2021 13:11:52 +0100
Subject: [PATCH] put code into functions and 'main' in flicker.py.

This is just so I can start to 'turn things on and off' (via
comments).
---
 src/flicker.py | 53 +++++++++++++++++++++++++++++---------------------
 1 file changed, 31 insertions(+), 22 deletions(-)

diff --git a/src/flicker.py b/src/flicker.py
index 28188fc..2bde0da 100644
--- a/src/flicker.py
+++ b/src/flicker.py
@@ -1,27 +1,36 @@
 import csv
 import datetime
 
-time_tallies = dict()
-tally_totals = dict()
-
-with open("data/test-data.csv") as csv_file:
-    csv_reader = csv.reader(csv_file, delimiter=",")
-    for r in csv_reader:
-        if (r[1] != "Time Stamp"):
-            time_str = r[1]
-            if (time_str in time_tallies):
-                time_tallies[time_str] = (time_tallies[time_str]) + 1
+def tally_readings_per_second():
+    time_tallies = dict()
+    tally_totals = dict()
+    with open("data/test-data.csv") as csv_file:
+        csv_reader = csv.reader(csv_file, delimiter=",")
+        for r in csv_reader:
+            if (r[1] != "Time Stamp"):
+                time_str = r[1]
+                if (time_str in time_tallies):
+                    time_tallies[time_str] = (time_tallies[time_str]) + 1
+                else:
+                    time_tallies[time_str] = 1
+        for r2 in time_tallies.values():
+            if (r2 in tally_totals):
+                tally_totals[r2] = (tally_totals[r2]) + 1
             else:
-                time_tallies[time_str] = 1
-    for r2 in time_tallies.values():
-        if (r2 in tally_totals):
-            tally_totals[r2] = (tally_totals[r2]) + 1
-        else:
-            tally_totals[r2] = 1
-    print(tally_totals.items())
-with open("data/results/reading-per-sec.csv", mode="w") as result:
-    wtr = csv.writer(result)
-    for k, v in tally_totals.items():
-        print(f"{k}: {v}")
-        wtr.writerow([k,v])
+                tally_totals[r2] = 1
+                # print(tally_totals.items())
+    return tally_totals
+
+def save_tally_totals(tally_totals):
+    with open("data/results/readings-per-sec.csv", mode="w") as result:
+        wtr = csv.writer(result)
+        for k, v in tally_totals.items():
+            print(f"{k}: {v}")
+            wtr.writerow([k,v])
 
+def main():
+    time_tallies = tally_readings_per_second()
+    save_tally_totals(time_tallies)
+    
+if __name__ == "__main__":
+    main()