@@ -341,7 +341,8 @@ class _ProgressBar(metaclass=_Singleton):
341341 STOP_SIGNAL = None
342342 UPDATE_SIGNAL = 1
343343
344- def __init__ (self , message ):
344+ def __init__ (self , message , enabled = True ):
345+ self .enabled = enabled
345346 self ._message = message
346347 self ._process = None
347348 self ._percentage = 0
@@ -360,14 +361,18 @@ def remaining_percentage(self):
360361
361362 def new (self , message ):
362363 """Fill the current bar, and create a new bar with message."""
363- self ._message = message
364- self .fill ()
365- self ._percentage = 0
366- self ._message_queue .put ((message , 100 ))
364+ if self .enabled :
365+ self ._message = message
366+ self .fill ()
367+ self ._percentage = 0
368+ self ._message_queue .put ((message , 100 ))
367369 return self
368370
369371 def update (self , amount = 1 ):
370372 """Update the progress bar with amount."""
373+ if not self .enabled :
374+ return
375+
371376 self ._update += amount
372377 if self ._update < self ._resolution :
373378 return
@@ -393,6 +398,9 @@ def _stop(self):
393398 self ._process .join ()
394399
395400 def __enter__ (self ):
401+ if not self .enabled :
402+ return self
403+
396404 def progress_runner (message , total , message_queue ):
397405 format = '{l_bar}{bar}|[{elapsed} {remaining}s]'
398406 bar = tqdm .tqdm (total = total , bar_format = format )
@@ -422,7 +430,8 @@ def progress_runner(message, total, message_queue):
422430 return self
423431
424432 def __exit__ (self , exc_type , exc_val , exc_tb ):
425- self ._stop ()
433+ if self .enabled :
434+ self ._stop ()
426435
427436
428437#: Global progress bar used by compare50
0 commit comments