dd16890c11
* Add Deleted error and disable all bad browser rows * Avoid error when opening the browse screen to a card with a missing note (dae) * In cards mode, a missing note is NotFound, not Deleted (dae) So we distinguish between referential integrity error, and explicit deletion. * Remove redundant try block
103 lines
2.9 KiB
Python
103 lines
2.9 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 time
|
|
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING, Generator, Sequence, Union
|
|
|
|
import aqt
|
|
import aqt.browser
|
|
from anki.cards import CardId
|
|
from anki.collection import BrowserColumns as Columns
|
|
from anki.collection import BrowserRow
|
|
from anki.notes import NoteId
|
|
from aqt import colors
|
|
from aqt.utils import tr
|
|
|
|
Column = Columns.Column
|
|
ItemId = Union[CardId, NoteId]
|
|
ItemList = Union[Sequence[CardId], Sequence[NoteId]]
|
|
|
|
|
|
@dataclass
|
|
class SearchContext:
|
|
search: str
|
|
browser: aqt.browser.Browser
|
|
order: bool | str | Column = True
|
|
reverse: bool = False
|
|
# if set, provided ids will be used instead of the regular search
|
|
ids: Sequence[ItemId] | None = None
|
|
|
|
|
|
@dataclass
|
|
class Cell:
|
|
text: str
|
|
is_rtl: bool
|
|
|
|
|
|
class CellRow:
|
|
is_disabled: bool = False
|
|
|
|
def __init__(
|
|
self,
|
|
cells: Generator[tuple[str, bool], None, None],
|
|
color: BrowserRow.Color.V,
|
|
font_name: str,
|
|
font_size: int,
|
|
) -> None:
|
|
self.refreshed_at: float = time.time()
|
|
self.cells: tuple[Cell, ...] = tuple(Cell(*cell) for cell in cells)
|
|
self.color: tuple[str, str] | None = backend_color_to_aqt_color(color)
|
|
self.font_name: str = font_name or "arial"
|
|
self.font_size: int = font_size if font_size > 0 else 12
|
|
|
|
def is_stale(self, threshold: float) -> bool:
|
|
return self.refreshed_at < threshold
|
|
|
|
@staticmethod
|
|
def generic(length: int, cell_text: str) -> CellRow:
|
|
return CellRow(
|
|
((cell_text, False) for cell in range(length)),
|
|
BrowserRow.COLOR_DEFAULT,
|
|
"arial",
|
|
12,
|
|
)
|
|
|
|
@staticmethod
|
|
def placeholder(length: int) -> CellRow:
|
|
return CellRow.generic(length, "...")
|
|
|
|
@staticmethod
|
|
def disabled(length: int, cell_text: str) -> CellRow:
|
|
row = CellRow.generic(length, cell_text)
|
|
row.is_disabled = True
|
|
return row
|
|
|
|
|
|
def backend_color_to_aqt_color(color: BrowserRow.Color.V) -> tuple[str, str] | None:
|
|
if color == BrowserRow.COLOR_MARKED:
|
|
return colors.MARKED_BG
|
|
if color == BrowserRow.COLOR_SUSPENDED:
|
|
return colors.SUSPENDED_BG
|
|
if color == BrowserRow.COLOR_FLAG_RED:
|
|
return colors.FLAG1_BG
|
|
if color == BrowserRow.COLOR_FLAG_ORANGE:
|
|
return colors.FLAG2_BG
|
|
if color == BrowserRow.COLOR_FLAG_GREEN:
|
|
return colors.FLAG3_BG
|
|
if color == BrowserRow.COLOR_FLAG_BLUE:
|
|
return colors.FLAG4_BG
|
|
if color == BrowserRow.COLOR_FLAG_PINK:
|
|
return colors.FLAG5_BG
|
|
if color == BrowserRow.COLOR_FLAG_TURQUOISE:
|
|
return colors.FLAG6_BG
|
|
if color == BrowserRow.COLOR_FLAG_PURPLE:
|
|
return colors.FLAG7_BG
|
|
return None
|
|
|
|
|
|
from .model import DataModel
|
|
from .state import CardState, ItemState, NoteState
|
|
from .table import StatusDelegate, Table
|