Deleted batch files when one run only

Author: sgeulette

src/imio/updates/tests/tests.py

@@ -76,6 +76,7 @@ def mock_get_batch_config_se(var):
         # batch_config is the result after the first run
         reset(call_res, params)
         params["env"] = "SPACEX=GREAT"
+        batch_config.update({"pf": "", "cf": ""})
         batch_config.update({"ll": 5, "kc": 5, "bn": 10})
         run_function_parts("a", {"batch": 10, "batching": ["a"]}, params)
         self.assertEqual(len(call_res), 1)

src/imio/updates/update_instances.py

@@ -3,6 +3,7 @@
 from datetime import datetime
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
+from imio.pyutils.batching import batch_delete_files
 from imio.pyutils.system import dump_var
 from imio.pyutils.system import error
 from imio.pyutils.system import get_git_tag
@@ -425,8 +426,8 @@ def run_function_parts(func_parts, batches_conf, params):
                         last = 2 + yet_to_treat // batch_config["bn"]  # int part
                         if yet_to_treat % batch_config["bn"]:  # modulo if p > b or p < b
                             last += 1
-                        # if last == 2:
-                        #     batch_delete_files({}, batch_config)
+                        if last == 2:  # only one run, already done
+                            batch_delete_files({}, batch_config)
             for batch in range(first, last):
                 if " BATCH=" in params["env"] and batch == (last - 1):
                     params["env"] += " BATCH_LAST=1"