diff --git a/CONTRIBUTORS b/CONTRIBUTORS index 77a9eb842..1f1ad16fc 100644 --- a/CONTRIBUTORS +++ b/CONTRIBUTORS @@ -160,6 +160,7 @@ Pedro Lameiras Kai Knoblich Lucas Scharenbroch Antonio Cavallo +Han Yeong-woo ******************** diff --git a/pylib/anki/scheduler/base.py b/pylib/anki/scheduler/base.py index d84194cee..4316170cd 100644 --- a/pylib/anki/scheduler/base.py +++ b/pylib/anki/scheduler/base.py @@ -22,7 +22,7 @@ ScheduleCardsAsNewDefaults = scheduler_pb2.ScheduleCardsAsNewDefaultsResponse FilteredDeckForUpdate = decks_pb2.FilteredDeckForUpdate RepositionDefaults = scheduler_pb2.RepositionDefaultsResponse -from typing import Sequence +from typing import Sequence, overload from anki import config_pb2 from anki.cards import CardId @@ -65,6 +65,14 @@ class SchedulerBase(DeprecatedNamesMixin): # Deck list ########################################################################## + @overload + def deck_due_tree(self, top_deck_id: None = None) -> DeckTreeNode: + ... + + @overload + def deck_due_tree(self, top_deck_id: DeckId) -> DeckTreeNode | None: + ... + def deck_due_tree(self, top_deck_id: DeckId | None = None) -> DeckTreeNode | None: """Returns a tree of decks with counts. If top_deck_id provided, only the according subtree is returned."""