Add timer

1 files changed, 14 insertions, 1 deletions

diff --git a/solutions/py/main.py b/solutions/py/main.py
index 3456dd6..694c23f 100644
--- a/solutions/py/main.py
+++ b/solutions/py/main.py
@@ -1,3 +1,5 @@
+import time
+
 import d01
 import d02
 import d03
@@ -9,7 +11,18 @@ import d08
 
 mods = [d01, d02, d03, d04, d05, d06, d07, d08]
 
-for mod, day in zip(mods, range(8)):
+timings = [[0 for _ in range(2)] for _ in range(len(mods))]
+clock_type = time.CLOCK_MONOTONIC
+
+for mod, day in zip(mods, range(len(mods))):
     print("Day", str(day+1).zfill(2))
+    t0 = time.clock_gettime_ns(clock_type)
     print("Part", 1, mod.pt1(open("../input/" + str(day+1).zfill(2), "r").readlines()))
+    timings[day][0] = time.clock_gettime_ns(clock_type) - t0
+    t0 = time.clock_gettime_ns(clock_type)
     print("Part", 2, mod.pt2(open("../input/" + str(day+1).zfill(2), "r").readlines()))
+    timings[day][1] = time.clock_gettime_ns(clock_type) - t0
+
+for day in range(len(timings)):
+    for part in range(2):
+        print("day", str(day+1).zfill(2), part+1, "  ", timings[day][part] / 1000000, "ms")