Add cell label in Quarto render status with nb_cell_yaml_options()

Author: yuninxia

src/resources/jupyter/notebook.py

@@ -149,19 +149,25 @@ def notebook_execute(options, status):
    # complete progress if necessary
    if (not quiet) and created:
       status("Done\n")
-      
-   # compute total code cells (for progress)
+
    current_code_cell = 1
-   total_code_cells = sum(cell.cell_type == 'code' for cell in client.nb.cells)
+   total_code_cells = 0
+   cell_labels = []
+   max_label_len = 0
 
-   # find max label length (for progress)   
-   max_label_len = max(len(nb_cell_yaml_options(client.nb.metadata.kernelspec.language, cell).get('label', '')) for cell in client.nb.cells)
+   for cell in client.nb.cells:
+      # compute total code cells (for progress)
+      if cell.cell_type == 'code':
+         total_code_cells += 1
+      # map cells to their labels
+      label = nb_cell_yaml_options(client.nb.metadata.kernelspec.language, cell).get('label', '')
+      cell_labels.append(label)
+      # find max label length
+      max_label_len = max(max_label_len, len(label))
 
    # execute the cells
    for index, cell in enumerate(client.nb.cells):
-      # read cell options
-      cell_options = nb_cell_yaml_options(client.nb.metadata.kernelspec.language, cell)
-      cell_label = cell_options.get('label', '')
+      cell_label = cell_labels[index]
       padding = "." * (max_label_len - len(cell_label))
 
       # progress