From ccf00da0cda57d476d4a45ee6c0bded827a013a6 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Thu, 16 Mar 2023 21:54:43 +0100 Subject: [PATCH] adding a progress bar (tqdm) --- environment.yml | 1 + src/captcha/CaptchaGenerator.py | 5 ++--- 2 files changed, 3 insertions(+), 3 deletions(-) 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):