From 7818deec2b6fa3728a2e8dd96f654fa5bf826dc2 Mon Sep 17 00:00:00 2001
From: Mark Jordan <mjordan@sfu.ca>
Date: Tue, 3 Dec 2024 18:33:52 -0800
Subject: [PATCH] Fixup from #844.

---
 workbench_utils.py | 54 ++++++++++++++++++++++++++++++----------------
 1 file changed, 36 insertions(+), 18 deletions(-)

diff --git a/workbench_utils.py b/workbench_utils.py
index cf68297..939f74c 100644
--- a/workbench_utils.py
+++ b/workbench_utils.py
@@ -9011,16 +9011,25 @@ def get_rollback_csv_filepath(config):
             config["rollback_csv_filename_template"]
         )
         try:
-            rollback_csv_filename_basename = str(
-                rollback_csv_filename_template.substitute(
-                    {
-                        "config_filename": config_filename,
-                        "input_csv_filename": input_csv_filename,
-                        "csv_start_row": str(config["csv_start_row"]),
-                        "csv_stop_row": str(config["csv_stop_row"]),
-                    }
+            if config["task"] == "create":
+                rollback_csv_filename_basename = str(
+                    rollback_csv_filename_template.substitute(
+                        {
+                            "config_filename": config_filename,
+                            "input_csv_filename": input_csv_filename,
+                            "csv_start_row": str(config["csv_start_row"]),
+                            "csv_stop_row": str(config["csv_stop_row"]),
+                        }
+                    )
+                )
+            if config["task"] == "create_from_files":
+                rollback_csv_filename_basename = str(
+                    rollback_csv_filename_template.substitute(
+                        {
+                            "config_filename": config_filename,
+                        }
+                    )
                 )
-            )
         except Exception as e:
             # We need to account for the very common case where the user has included "valid identifier characters"
             # (as defined in https://peps.python.org/pep-0292/) as part of their template. The most common case will
@@ -9081,16 +9090,25 @@ def get_rollback_config_filepath(config):
             config["rollback_config_filename_template"]
         )
         try:
-            rollback_config_filename_basename = str(
-                rollback_config_filename_template.substitute(
-                    {
-                        "config_filename": config_filename,
-                        "input_csv_filename": input_csv_filename,
-                        "csv_start_row": str(config["csv_start_row"]),
-                        "csv_stop_row": str(config["csv_stop_row"]),
-                    }
+            if config["task"] == "create":
+                rollback_config_filename_basename = str(
+                    rollback_config_filename_template.substitute(
+                        {
+                            "config_filename": config_filename,
+                            "input_csv_filename": input_csv_filename,
+                            "csv_start_row": str(config["csv_start_row"]),
+                            "csv_stop_row": str(config["csv_stop_row"]),
+                        }
+                    )
+                )
+            if config["task"] == "create_from_files":
+                rollback_config_filename_basename = str(
+                    rollback_config_filename_template.substitute(
+                        {
+                            "config_filename": config_filename,
+                        }
+                    )
                 )
-            )
         except Exception as e:
             # We need to account for the very common case where the user has included "valid identifier characters"
             # (as defined in https://peps.python.org/pep-0292/) as part of their template. The most common case will