74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
from __future__ import annotations
|
|
|
|
import aqt
|
|
from anki.collection import LegacyCheckpoint, LegacyReviewUndo, OpChangesAfterUndo
|
|
from anki.errors import UndoEmpty
|
|
from anki.types import assert_exhaustive
|
|
from aqt import gui_hooks
|
|
from aqt.qt import QWidget
|
|
from aqt.utils import showInfo, showWarning, tooltip, tr
|
|
|
|
|
|
def undo(*, mw: aqt.AnkiQt, parent: QWidget) -> None:
|
|
"Undo the last operation, and refresh the UI."
|
|
|
|
def on_success(out: OpChangesAfterUndo) -> None:
|
|
mw.update_undo_actions(out.new_status)
|
|
tooltip(tr.undo_action_undone(action=out.operation), parent=parent)
|
|
|
|
def on_failure(exc: Exception) -> None:
|
|
if isinstance(exc, UndoEmpty):
|
|
# backend has no undo, but there may be a checkpoint
|
|
# or v1/v2 review waiting
|
|
_legacy_undo(mw=mw, parent=parent)
|
|
else:
|
|
showWarning(str(exc), parent=parent)
|
|
|
|
mw.perform_op(mw.col.undo, success=on_success, failure=on_failure)
|
|
|
|
|
|
def _legacy_undo(*, mw: aqt.AnkiQt, parent: QWidget) -> None:
|
|
reviewing = mw.state == "review"
|
|
just_refresh_reviewer = False
|
|
|
|
result = mw.col.undo_legacy()
|
|
|
|
if result is None:
|
|
# should not happen
|
|
showInfo("nothing to undo", parent=parent)
|
|
mw.update_undo_actions()
|
|
return
|
|
|
|
elif isinstance(result, LegacyReviewUndo):
|
|
name = tr.scheduling_review()
|
|
|
|
if reviewing:
|
|
# push the undone card to the top of the queue
|
|
cid = result.card.id
|
|
card = mw.col.getCard(cid)
|
|
mw.reviewer.cardQueue.append(card)
|
|
|
|
gui_hooks.review_did_undo(cid)
|
|
|
|
just_refresh_reviewer = True
|
|
|
|
elif isinstance(result, LegacyCheckpoint):
|
|
name = result.name
|
|
|
|
else:
|
|
assert_exhaustive(result)
|
|
assert False
|
|
|
|
if just_refresh_reviewer:
|
|
mw.reviewer.nextCard()
|
|
else:
|
|
# full queue+gui reset required
|
|
mw.reset()
|
|
|
|
tooltip(tr.undo_action_undone(action=name), parent=parent)
|
|
gui_hooks.state_did_revert(name)
|
|
mw.update_undo_actions()
|