You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
868 B
35 lines
868 B
import time
|
|
import logging
|
|
|
|
try:
|
|
from tqdm import tqdm
|
|
except ImportError:
|
|
tqdm = None
|
|
|
|
|
|
REPORT_ON = 0.3
|
|
|
|
class Progress:
|
|
def __call__(self, iterable, description, total=None):
|
|
if tqdm is None:
|
|
try:
|
|
total = len(iterable)
|
|
except TypeError:
|
|
total = -1
|
|
|
|
start_time = time.time()
|
|
last_report = start_time - REPORT_ON
|
|
for n, el in enumerate(iterable):
|
|
now = time.time()
|
|
if now - last_report > REPORT_ON:
|
|
logging.info("\r{}: {}/{}".format(description, n, total), end="")
|
|
last_report = now
|
|
yield el
|
|
logging.info(" -> {}".format(time.time() - start_time))
|
|
else:
|
|
yield from tqdm(iterable, desc=description, total=total)
|
|
|
|
|
|
progress = Progress()
|
|
|