diff --git a/environment.yml b/environment.yml index 59c3150f39c..d2542baa7cc 100644 --- a/environment.yml +++ b/environment.yml @@ -16,6 +16,7 @@ dependencies: - nb_conda_kernels - pillow - openpyxl + - tqdm # - python-decouple # - selenium # - webdriver-manager diff --git a/src/captcha/CaptchaGenerator.py b/src/captcha/CaptchaGenerator.py index 97200acd113..3dcd06fe209 100644 --- a/src/captcha/CaptchaGenerator.py +++ b/src/captcha/CaptchaGenerator.py @@ -2,6 +2,7 @@ from PIL import Image, ImageDraw, ImageFont import random import string import shutil +from tqdm import tqdm class CaptchaGenerator: @@ -15,9 +16,7 @@ class CaptchaGenerator: def createAndSaveCaptchas(self): self._prepareDataDir() - for i in range(self.numCaptchas): - # FK-TODO: use https://github.com/tqdm/tqdm - print(f'saving captcha {i+1}/{self.numCaptchas}') + for _ in tqdm(range(self.numCaptchas)): self._createAndSaveCaptcha() def _prepareDataDir(self):