diff --git a/progress_bar.py b/progress_bar.py index d025c80..ac1f70b 100644 --- a/progress_bar.py +++ b/progress_bar.py @@ -4,6 +4,8 @@ import enum +from datetime import datetime + class ProgressBar(object): @@ -11,6 +13,8 @@ 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()