From 4298b021e91234a5d112cd67a88cedce40526968 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Sun, 19 Jul 2026 15:04:45 +0200 Subject: [PATCH] enhancement: log job timing here, too --- src/eynollah/image_enhancer.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/eynollah/image_enhancer.py b/src/eynollah/image_enhancer.py index fe1e16d..f515549 100644 --- a/src/eynollah/image_enhancer.py +++ b/src/eynollah/image_enhancer.py @@ -3,9 +3,9 @@ Image enhancer. The output can be written as same scale of input or in new predi """ import logging +import time import os from typing import Optional -from pathlib import Path import cv2 @@ -48,6 +48,8 @@ class Enhancer(Eynollah): dir_out: Optional[str] = None, overwrite: bool = False, ) -> None: + t0 = time.time() + self.logger.info(img_filename) image = self.cache_images(image_filename=img_filename, image_pil=img_pil) output_filename = os.path.join(dir_out or "", image['name'] + '.png') @@ -67,6 +69,7 @@ class Enhancer(Eynollah): cv2.imwrite(output_filename, img_res) self.logger.info("output filename: '%s'", output_filename) + self.logger.info("Job done in %.1fs", time.time() - t0) def run(self, overwrite: bool = False, @@ -78,6 +81,7 @@ class Enhancer(Eynollah): Enlarge and enhance the scanned images """ if dir_in: + t0_tot = time.time() ls_imgs = [os.path.join(dir_in, image_filename) for image_filename in filter(is_image_filename, os.listdir(dir_in))] @@ -87,8 +91,8 @@ class Enhancer(Eynollah): raise ValueError("run requires either a single image filename or a directory") for img_filename in ls_imgs: - self.logger.info(img_filename) - self.run_single(img_filename, dir_out=dir_out, overwrite=overwrite) + if dir_in: + self.logger.info("All jobs done in %.1fs", time.time() - t0_tot)