refactor(chore): done timeouts as external service function

Author: Serhii Naumenko

file_operations/clean_annotations.py

@@ -1,11 +1,13 @@
 import argparse
+import time
 from pathlib import Path
 from typing import Union
 
 from const_utils.arguments import Arguments
 from const_utils.default_values import AppSettings
 from const_utils.parser_help import HelpStrings
 from file_operations.file_operation import FileOperation
+from services.timeout import wait
 from tools.mixins.file_remover import FileRemoverMixin
 
 
@@ -80,6 +82,8 @@ def do_task(self) -> None:
 
         self.logger.info(f"Removed {orphans_removed} orphan annotations")
 
+        wait(logger=self.logger, timeout=self.sleep)
+
 
     @property
     def a_source(self) -> Path:

file_operations/deduplicate.py

@@ -1,9 +1,12 @@
 import argparse
+import time
+
 from const_utils.arguments import Arguments
 from const_utils.copmarer import Constants
 from const_utils.default_values import AppSettings
 from const_utils.parser_help import HelpStrings
 from file_operations.file_operation import FileOperation
+from services.timeout import wait
 from tools.mixins.file_remover import FileRemoverMixin
 from tools.comparer.img_comparer.img_comparer import ImageComparer
 
@@ -99,7 +102,9 @@ def do_task(self):
         self.logger.info(f"Found {duplicates_count} duplicates in {len(self.files_for_task)} files")
 
         if duplicates_count > 0 and self.confirm_removing():
-                self.remove_all(duplicates)
+            self.remove_all(duplicates)
+
+        wait(logger=self.logger, timeout=self.sleep)
 
     def confirm_removing(self) -> bool:
         """

services/timeout.py

@@ -0,0 +1,13 @@
+import logging
+import time
+
+def wait(logger: logging.Logger, timeout: float):
+    """
+    Waits until timeout occurs. Logs exception while waiting until timeout occurs.
+
+    Args:
+        logger (logging.Logger): Logger to wait until timeout occurs.
+        timeout (float): Timeout in seconds.
+    """
+    logger.info(f" Wait for {timeout} seconds...\n")
+    time.sleep(timeout)

tst_commands.py

@@ -56,8 +56,8 @@
     Commands.stats: [
         "data_forge.py",
         "stats",
-        "./media/Pascal VOC 2012.v1-raw.voc/",
-        # "./media/annotated/",
+        # "./media/Pascal VOC 2012.v1-raw.voc/",
+        "./media/annotated/",
         "-p", ".xml",
         "--destination-type", "voc",
     ]