fix: deck chooser causes error

migaku-official · Jan 3, 2024 · 4e3d905 · 4e3d905
1 parent cb91654
commit 4e3d905
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/__init__.py b/src/__init__.py
@@ -60,11 +60,13 @@ def deck_change(id):
             "before",
         )
     else:
-        aqt.deckchooser.DeckChooser.choose_deck = anki.hooks.wrap(
-            aqt.deckchooser.DeckChooser.choose_deck,
-            lambda deckchooser: deck_change(deckchooser.selected_deck_id),
-            "after",
-        )
+        save_func = aqt.deckchooser.DeckChooser.choose_deck
+
+        def on_deck_change(deckchooser):
+            save_func(deckchooser)
+            deck_change(deckchooser.selected_deck_id),
+
+        aqt.deckchooser.DeckChooser.choose_deck = on_deck_change
 
     ### MODEL CHANGE
     def notetype_change(id):