Add total time display on query

diff --git a/tt.py b/tt.py
index 37cbbaa7a33aa1fce7ebc839ce2bd44c6070535e..c0cdbfb6ef3b458209a4c2ff4b5aa6f3011cf749 100755 (executable)
--- a/tt.py
+++ b/tt.py
@@ -213,6 +213,7 @@ def query_data(entries: TimeEntries, args: List[str]):
 
         i += 1
 
+    total_time = datetime.timedelta(seconds=0)
     for cat in categories:
         td = datetime.timedelta(seconds=0)
         punch_in = None
@@ -229,6 +230,8 @@ def query_data(entries: TimeEntries, args: List[str]):
                 punch_in = None
 
         print(f"{cat}:\t{td}")
+        total_time += td
+    print(f"Total:\t{total_time}")
 
 
 # Service specific