Fix handling of activated sixth column when updating README

Author: ggeerraarrdd

src/main/helpers/helpers.py

@@ -9,7 +9,6 @@
 from jinja2 import Template
 
 # Local
-from src.main.config import NB
 from src.main.config import TEMPLATES_DIR
 
 
@@ -64,7 +63,7 @@ def get_files_created(data: Dict[str, Any]) -> int:
     return 1
 
 
-def get_target_line_dict(line: str) -> Dict[str, str]:
+def get_target_line_dict(nb_loc, line: str) -> Dict[str, str]:
     """
     Parse a table line into a dictionary based on notebook configuration.
 
@@ -87,11 +86,11 @@ def get_target_line_dict(line: str) -> Dict[str, str]:
         "nb": "",
     }
 
-    if NB == 0:
+    if nb_loc == 0:
 
         keys = list(data.keys())[:-1]  # Exclude 'nb'
 
-    elif NB == 1:
+    elif nb_loc == 1:
 
         keys = list(data.keys())
 
@@ -115,7 +114,7 @@ def get_target_line_dict(line: str) -> Dict[str, str]:
     return results
 
 
-def get_target_line_updated(data: Dict[str, str], widths: Dict[str, int]) -> str:
+def get_target_line_updated(nb_loc, data: Dict[str, str], widths: Dict[str, int]) -> str:
     """
     Format a table line with proper padding based on column widths.
 
@@ -132,7 +131,7 @@ def get_target_line_updated(data: Dict[str, str], widths: Dict[str, int]) -> str
 
     target_line = "|"
 
-    if NB == 0:
+    if nb_loc == 0:
 
         for key, value in data.items():
 
@@ -148,7 +147,7 @@ def get_target_line_updated(data: Dict[str, str], widths: Dict[str, int]) -> str
 
                 target_line += f" {value_str}{padding} |"
 
-    elif NB == 1:
+    elif nb_loc == 1:
 
         for key, value in data.items():
 

src/main/main.py

@@ -390,7 +390,8 @@ def implement_runs(handler: PackageHandler) -> int:
     if handler.get_value("entry_data", "nb"):
         handler.update_value("entry_data", "nb", "")
 
-    new_entry = get_target_line_updated(handler.get_dictionary("entry_data"),
+    new_entry = get_target_line_updated(handler.get_value("config_base", "nb_loc"),
+                                        handler.get_dictionary("entry_data"),
                                         handler.get_dictionary("config_cols_widths"))
 
 
@@ -423,10 +424,12 @@ def implement_runs(handler: PackageHandler) -> int:
             for i in range(start_line + 1, end_line):
 
                 # Convert target line (str) to data (dict)
-                target_line_data = get_target_line_dict(lines[i])
+                target_line_data = get_target_line_dict(handler.get_value("config_base", "nb_loc"),
+                                                        lines[i])
 
                 # Update target line
-                line_updated = get_target_line_updated(target_line_data,
+                line_updated = get_target_line_updated(handler.get_value("config_base", "nb_loc"),
+                                                       target_line_data,
                                                        handler.get_dictionary("config_cols_widths"))
 
                 # Replace the original line with the updated one