From b968744045a666679e0d80d851fc625a2906a0fa Mon Sep 17 00:00:00 2001
From: Jaap Blom <jblom@beeldengeluid.nl>
Date: Thu, 2 Nov 2023 11:43:05 +0100
Subject: [PATCH] put local test run in separate function

---
 worker.py | 38 +++++++++++++++++++++-----------------
 1 file changed, 21 insertions(+), 17 deletions(-)

diff --git a/worker.py b/worker.py
index bbe4b30..d8c745b 100644
--- a/worker.py
+++ b/worker.py
@@ -35,6 +35,26 @@
 logger = logging.getLogger()
 
 
+# triggered by running: python worker.py --run-test-file
+def process_configured_input_file():
+    logger.info("Triggered processing of configured VISXP_PREP.TEST_INPUT_PATH")
+    proc_result = generate_input_for_feature_extraction(cfg.VISXP_PREP.TEST_INPUT_FILE)
+    if proc_result.provenance:
+        logger.info(
+            f"Successfully processed example file in {proc_result.provenance.processing_time_ms}ms"
+        )
+        logger.info("Result ok, now applying the desired IO on the results")
+        validated_output: CallbackResponse = apply_desired_io_on_output(
+            cfg.VISXP_PREP.TEST_INPUT_FILE,
+            proc_result,
+            cfg.INPUT.DELETE_ON_COMPLETION,
+            cfg.OUTPUT.DELETE_ON_COMPLETION,
+            cfg.OUTPUT.TRANSFER_ON_COMPLETION,
+        )
+    else:
+        logger.info(f"Error: {proc_result.state}: {proc_result.message}")
+
+
 class VideoSegmentationWorker(base_worker):
     def __init__(self, config):
         logger.info(config)
@@ -229,23 +249,7 @@ def save_to_dane_index(
     if args.run_test_file != "n":
         logger.info("Running main_data_processor with VISXP_PREP.TEST_INPUT_FILE ")
         if cfg.VISXP_PREP and cfg.VISXP_PREP.TEST_INPUT_FILE:
-            proc_result = generate_input_for_feature_extraction(
-                cfg.VISXP_PREP.TEST_INPUT_FILE
-            )
-            if proc_result.provenance:
-                logger.info(
-                    f"Successfully processed example file in {proc_result.provenance.processing_time_ms}ms"
-                )
-                logger.info("Result ok, now applying the desired IO on the results")
-                validated_output: CallbackResponse = apply_desired_io_on_output(
-                    cfg.VISXP_PREP.TEST_INPUT_FILE,
-                    proc_result,
-                    cfg.INPUT.DELETE_ON_COMPLETION,
-                    cfg.OUTPUT.DELETE_ON_COMPLETION,
-                    cfg.OUTPUT.TRANSFER_ON_COMPLETION,
-                )
-            else:
-                logger.info(f"Error: {proc_result.state}: {proc_result.message}")
+            process_configured_input_file()
         else:
             logger.error("Please configure an input file in VISXP_PREP.TEST_INPUT_FILE")
             sys.exit()