Stopwatchクラスのリファクタリング

app.py

@@ -5,12 +5,12 @@
 from argparse import ArgumentParser, BooleanOptionalAction
 
 from src import PROGRAM_NAME, get_package_version
-from src.myapp import run_app
 
 __version__ = get_package_version()
 
 
 def main():
+    from src.myapp import run_app
     """
         エントリーポイント
         1, コマンドライン引数の解析を行います

src/__init__.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 """src Module"""
 from datetime import datetime
-import logging.config
+import logging
 import re
 from zoneinfo import ZoneInfo
 
@@ -35,7 +35,7 @@ class LocalTimeFormatter(logging.Formatter):
 
         t = dt.strftime(re.sub(self.pattern, "", datefmt))
         match = re.search(self.pattern, datefmt)
-        if not match:
+        if match is None:
             return t
 
         groups = match.groups()

src/myapp.py

@@ -56,23 +56,24 @@ def submit_clicked(image: np.ndarray, algorithm: Literal["SpectralResidual", "Fi
         np.ndarray: JET画像
         np.ndarray: HOT画像
     """
-    log.info("#submit_Clicked")
-    watch = Stopwatch.start_new()
+    sw = Stopwatch.start_new()
+    log.info(f"#submit_clicked({sw.elapsed:.3f}s)")
     #
     saliency = SaliencyMap(algorithm)
+    log.info(f"#SaliencyMap({sw.elapsed:.3f}s)")
     success, saliency_map = saliency.compute(image)
-    #log.info("#SaliencyMap compute()")
+    log.info(f"#compute({sw.elapsed:.3f}s)")
 
     if not success:
         return image, image  # エラーが発生した場合は入力画像を返します。
 
-    #log.info("#jet")
+    log.info(f"#jet({sw.elapsed:.3f}s)")
     jet = convert_colormap(image, saliency_map, "jet")
     # jet = None
-    #log.info("#hot")
+    log.info(f"#hot({sw.elapsed:.3f}s)")
     hot = convert_colormap(image, saliency_map, "hot")
     saliency = None
-    log.info(f"#submit_Clicked End{watch.stop():.3f}")
+    log.info(f"#submit_clicked({sw.elapsed:.3f}s)")
     return jet, hot
 
 

src/reporter.py

@@ -11,6 +11,7 @@
 import json
 from logging import Logger, getLogger
 import logging.config
+
 from typing import Optional
 
 from . import PROGRAM_NAME

src/saliency.py

@@ -10,7 +10,7 @@ class SaliencyMap:
     """
     顕著性マップを計算するクラス。
     Example:
-        from lib.saliency import SaliencyMap
+        from src.saliency import SaliencyMap
 
         saliency = SaliencyMap("SpectralResidual")
         success, saliencyMap = saliency.compute(image)

src/utils.py

@@ -7,29 +7,34 @@ def get_package_version() -> str:
     """
     バージョン情報
     """
-    return '0.0.6'
+    return '0.0.7'
 
 
 class Stopwatch:
     """
-    Stopwatch 経過時間を計測するためのクラス。
+    経過時間を計測するためのクラス。
     Example:
         from src.utils import Stopwatch
 
         watch = Stopwatch.start_new()
-        # 計測する処理
-        print(f"{watch.stop():.3f}")
+        ### 計測する処理
+        print(f"{watch.elapsed:.3f}")
     """
 
     def __init__(self):
-        self._start_time = 0
-        self._elapsed = 0
+        self._start_time: float = 0
+        self._elapsed: float = 0
+        self._is_running: bool = False
 
     @property
-    def elapsed(self):
+    def elapsed(self) -> float:
         """
-        経過時間
+        経過時間を取得します。
         """
+        if self._is_running:
+            end_time = time.perf_counter()
+            self._elapsed = end_time - self._start_time
+
         return self._elapsed
 
     def start(self) -> None:
@@ -38,6 +43,7 @@ class Stopwatch:
         """
         self._start_time = time.perf_counter()
         self._elapsed = 0
+        self._is_running = True
 
     @classmethod
     def start_new(cls):
@@ -48,10 +54,19 @@ class Stopwatch:
         stopwatch.start()
         return stopwatch
 
-    def stop(self):
+    def stop(self) -> float:
         """
         計測を終了します。
         """
-        end_time = time.perf_counter()
-        self._elapsed = end_time - self._start_time
+        if self._is_running:
+            end_time = time.perf_counter()
+            self._elapsed = end_time - self._start_time
+            self._is_running = False
         return self._elapsed
+
+    @property
+    def is_running(self) -> bool:
+        """
+        実行中かどうかを取得します。
+        """
+        return self._is_running