avoid hanging UI when undoing in browse screen

pylib/anki/collection.py

@@ -72,8 +72,10 @@ class Checkpoint:
 class BackendUndo:
     name: str
 
+
 UndoResult = Union[None, BackendUndo, Checkpoint, ReviewUndo]
 
+
 class Collection:
     sched: Union[V1Scheduler, V2Scheduler]
 

qt/aqt/browser.py

@@ -518,7 +518,7 @@ class Browser(QMainWindow):
         # actions
         f = self.form
         # edit
-        qconnect(f.actionUndo.triggered, self.mw.onUndo)
+        qconnect(f.actionUndo.triggered, self.undo)
         qconnect(f.actionInvertSelection.triggered, self.invertSelection)
         qconnect(f.actionSelectNotes.triggered, self.selectNotes)
         if not isMac:
@@ -1498,6 +1498,16 @@ where id in %s"""
     def on_tag_list_update(self) -> None:
         self.sidebar.refresh()
 
+    # Undo
+    ######################################################################
+
+    def undo(self) -> None:
+        # need to make sure we don't hang the UI by redrawing the card list
+        # during the long-running op. mw.undo will take care of the progress
+        # dialog
+        self.setUpdatesEnabled(False)
+        self.mw.undo(lambda _: self.setUpdatesEnabled(True))
+
     def onUndoState(self, on: bool) -> None:
         self.form.actionUndo.setEnabled(on)
         if on:

qt/aqt/main.py

@@ -27,8 +27,14 @@ import aqt.toolbar
 import aqt.webview
 from anki import hooks
 from anki._backend import RustBackend as _RustBackend
-from anki.collection import BackendUndo, Checkpoint, Collection, Config, \
+from anki.collection import (
-    ReviewUndo, UndoResult
+    BackendUndo,
+    Checkpoint,
+    Collection,
+    Config,
+    ReviewUndo,
+    UndoResult,
+)
 from anki.decks import Deck
 from anki.hooks import runHook
 from anki.sound import AVTag, SoundOrVideoTag