Add MainWindowState hints to more signatures (#2064)

This commit is contained in:
Abdo 2022-09-12 12:31:22 +03:00 committed by GitHub
parent 1a87937973
commit 497b246b66
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 7 deletions

View file

@ -678,7 +678,7 @@ class AnkiQt(QMainWindow):
self.bottomWeb.show()
gui_hooks.state_did_change(state, oldState)
def _deckBrowserState(self, oldState: str) -> None:
def _deckBrowserState(self, oldState: MainWindowState) -> None:
self.deckBrowser.show()
def _selectedDeck(self) -> DeckDict | None:
@ -688,15 +688,15 @@ class AnkiQt(QMainWindow):
return None
return self.col.decks.get(did)
def _overviewState(self, oldState: str) -> None:
def _overviewState(self, oldState: MainWindowState) -> None:
if not self._selectedDeck():
return self.moveToState("deckBrowser")
self.overview.show()
def _reviewState(self, oldState: str) -> None:
def _reviewState(self, oldState: MainWindowState) -> None:
self.reviewer.show()
def _reviewCleanup(self, newState: str) -> None:
def _reviewCleanup(self, newState: MainWindowState) -> None:
if newState != "resetRequired" and newState != "review":
self.reviewer.cleanup()

View file

@ -531,18 +531,27 @@ hooks = [
# these refer to things like deckbrowser, overview and reviewer state,
Hook(
name="state_will_change",
args=["new_state: str", "old_state: str"],
args=[
"new_state: aqt.main.MainWindowState",
"old_state: aqt.main.MainWindowState",
],
legacy_hook="beforeStateChange",
),
Hook(
name="state_did_change",
args=["new_state: str", "old_state: str"],
args=[
"new_state: aqt.main.MainWindowState",
"old_state: aqt.main.MainWindowState",
],
legacy_hook="afterStateChange",
),
# different sig to original
Hook(
name="state_shortcuts_will_change",
args=["state: str", "shortcuts: list[tuple[str, Callable]]"],
args=[
"state: aqt.main.MainWindowState",
"shortcuts: list[tuple[str, Callable]]",
],
),
# UI state/refreshing
###################