Added stop_time

Store the stop_time in order to track how long it took for the event to complete not how long the ProgressBarCounter has been around.

Author: Ken Odegard

prompt_toolkit/shortcuts/progress_bar/base.py

@@ -306,6 +306,7 @@ def __init__(self, progress_bar: ProgressBar,
  total: Optional[int] = None) -> None:
 
  self.start_time = datetime.datetime.now()
+ self.stop_time = None
  self.progress_bar = progress_bar
  self.data = data
  self.items_completed = 0
@@ -348,6 +349,10 @@ def done(self) -> bool:
  def done(self, value: bool) -> None:
  self._done = value
 
+ # if declared done then store the stop_time
+ # otherwise clear the stop_time
+ self.stop_time = datetime.datetime.now() if value else None
+
  if value and self.remove_when_done:
  self.progress_bar.counters.remove(self)
 
@@ -363,7 +368,10 @@ def time_elapsed(self) -> datetime.timedelta:
  """
  return how much time has been elapsed since the start.
  """
- return datetime.datetime.now() - self.start_time
+ if self.stop_time is None:
+ return datetime.datetime.now() - self.start_time
+ else:
+ return self.stop_time - self.start_time
 
  @property
  def time_left(self) -> Optional[datetime.timedelta]: