時間を管理する準備

progress_bar.py

@@ -4,13 +4,17 @@
 
 import enum
 
+from datetime import datetime
+
 
 class ProgressBar(object):
 
     def __init__(self, max_num, unit_num=1):
         self.max_num = max_num
         self.unit_num = unit_num
         self.current_num = 0
+        self.started_at = None
+        self.finished_at = None
 
     @classmethod
     def iteration(cls, iterable, call_back, unit_num=1):
@@ -45,9 +49,10 @@ def update(self, num):
 
     def start(self):
         self.update(0)
+        self.started_at = datetime.now()
 
     def finish(self):
-        pass
+        self.finished_at = datetime.now()
 
     def __enter__(self):
         self.start()