Merge pull request #2 from smpotts/functions

smpotts · web-flow · commit 15eb58e4693a · 2024-06-03T21:07:43.000-05:00
added time functions
diff --git a/src/time.py b/src/time.py
@@ -22,4 +22,22 @@ def is_after(t1, t2):
         t2_time = datetime.strptime(t2.print_time(), time_format).time()
         return t1_time > t2_time
 
+    @staticmethod
+    def time_to_int(time):
+        minutes = time.hours * 60 + time.minutes
+        seconds = minutes * 60 + time.seconds
+        return seconds
+
+    @staticmethod
+    def int_to_time(seconds):
+        time = Time()
+        minutes, time.seconds = divmod(seconds, 60)
+        time.hours, time.minutes = divmod(minutes, 60)
+        return time
+
+    @staticmethod
+    def add_time(t1, t2):
+        seconds = Time.time_to_int(t1) + Time.time_to_int(t2)
+        return Time.int_to_time(seconds)
+
 
diff --git a/tests/test_time.py b/tests/test_time.py
@@ -23,6 +23,11 @@ def test_is_after(self):
         self.assertTrue(Time.is_after(t3, t2))
         self.assertFalse(Time.is_after(t2, t3))
 
+    def test_time_to_int_to_time(self):
+        t = Time(8, 54, 16)
+        self.assertEqual(Time.time_to_int(t), 32056)
+        self.assertEqual(Time.time_to_int(Time.int_to_time(32056)), 32056)
+
 
 if __name__ == '__main__':
     unittest.main()
