Add timestamps for every history update

Author: Golf Player

nbclient/client.py

@@ -308,6 +308,10 @@ def __init__(self, nb, km=None, **kw):
             'jupyter.widget': self.on_comm_open_jupyter_widget
         }
 
+    def _update_state(self, new_state):
+        self.state = new_state
+        self.state_history.append([new_state, timestamp()])
+
     def reset_execution_trackers(self):
         """Resets any per-execution trackers.
         """
@@ -321,6 +325,7 @@ def reset_execution_trackers(self):
         # our front-end mimicing Output widgets
         self.comm_objects = {}
         self.state = ExecutionState.NOTHING
+        self.state_history = []
         self.current_cell = None
         self.current_cell_index = -1
 
@@ -345,7 +350,7 @@ def start_kernel_manager(self):
         return self.km
 
     async def _async_cleanup_kernel(self):
-        self.state = ExecutionState.CLEANING_UP
+        self._update_state(ExecutionState.CLEANING_UP)
         now = self.shutdown_kernel == "immediate"
         try:
             # Queue the manager to kill the process, and recover gracefully if it's already dead.
@@ -382,7 +387,7 @@ async def async_start_new_kernel_client(self, **kwargs):
            The id of the started kernel.
         """
 
-        self.state = ExecutionState.STARTING_KERNEL_CLIENT
+        self._update_state(ExecutionState.STARTING_KERNEL_CLIENT)
         resource_path = self.resources.get('metadata', {}).get('path') or None
         if resource_path and 'cwd' not in kwargs:
             kwargs["cwd"] = resource_path
@@ -515,10 +520,10 @@ async def async_execute(self, reset_kc=False, **kwargs):
         if reset_kc and self.km:
             await self._async_cleanup_kernel()
         self.reset_execution_trackers()
-        self.state = ExecutionState.STARTUP
+        self._update_state(ExecutionState.STARTUP)
 
         async with self.async_setup_kernel(**kwargs):
-            self.state = ExecutionState.EXECUTING
+            self._update_state(ExecutionState.EXECUTING)
             self.log.info("Executing notebook with kernel: %s" % self.kernel_name)
             for index, cell in enumerate(self.nb.cells):
                 # Ignore `'execution_count' in content` as it's always 1
@@ -530,7 +535,7 @@ async def async_execute(self, reset_kc=False, **kwargs):
             info_msg = await self.async_wait_for_reply(msg_id)
             self.nb.metadata['language_info'] = info_msg['content']['language_info']
             self.set_widgets_metadata()
-        self.state = ExecutionState.COMPLETE
+        self._update_state(ExecutionState.COMPLETE)
         return self.nb
 
     execute = run_sync(async_execute)