fix mypy issues

This commit is contained in:
Damien Elmes 2020-05-03 09:05:29 +10:00
parent 5d6581faf2
commit a0160d7f1e

View file

@ -1149,7 +1149,7 @@ QTableView {{ gridline-color: {grid} }}
def _decksTree(self, root) -> None: def _decksTree(self, root) -> None:
tree = self.col.decks.deck_tree() tree = self.col.decks.deck_tree()
def fillGroups(root, nodes: List[DeckTreeNode], head=""): def fillGroups(root, nodes: Sequence[DeckTreeNode], head=""):
for node in nodes: for node in nodes:
if node.deck_id == 1 and not node.children: if node.deck_id == 1 and not node.children:
if not self.mw.col.decks.should_default_be_displayed( if not self.mw.col.decks.should_default_be_displayed(
@ -1157,12 +1157,22 @@ QTableView {{ gridline-color: {grid} }}
): ):
continue continue
def set_filter():
full_name = head + node.name # pylint: disable=cell-var-from-loop
return lambda: self.setFilter("deck", full_name)
def toggle_expand():
did = node.deck_id # pylint: disable=cell-var-from-loop
return lambda _: self.mw.col.decks.collapseBrowser(did)
item = SidebarItem( item = SidebarItem(
node.name, node.name,
":/icons/deck.svg", ":/icons/deck.svg",
lambda baseName=node.name: self.setFilter("deck", head + baseName), set_filter(),
lambda expanded, did=node.deck_id: self.mw.col.decks.collapseBrowser(did), toggle_expand(),
not self.mw.col.decks.get(node.deck_id).get("browserCollapsed", False), not self.mw.col.decks.get(node.deck_id).get(
"browserCollapsed", False
),
) )
root.addChild(item) root.addChild(item)
newhead = head + node.name + "::" newhead = head + node.name + "::"