Merge pull request #414 from irgolic/store-updated-prompt

Set updated prompt as previous prompt object in JupyterWidget
jupyter · Mar 25, 2020 · ad21b6c · ad21b6c
2 parents 2e3a17b + 482a9d6
commit ad21b6c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/qtconsole/jupyter_widget.py b/qtconsole/jupyter_widget.py
@@ -406,6 +406,10 @@ def _update_prompt(self, new_prompt_number):
             # Update the prompt cursor
             self._prompt_cursor.setPosition(cursor.position() - 1)
 
+            # Store the updated prompt.
+            block = self._control.document().lastBlock()
+            length = len(self._prompt)
+            self._previous_prompt_obj = self._PromptBlock(block, length, new_prompt_number)
 
     def _show_interpreter_prompt_for_reply(self, msg):
         """ Reimplemented for IPython-style prompts.