BrickTracker/bricktracker/rebrickable_set_list.py

from typing import Self

from .rebrickable_set import RebrickableSet
from .record_list import BrickRecordList


# All the rebrickable sets from the database
class RebrickableSetList(BrickRecordList[RebrickableSet]):

    # Queries
    select_query: str = 'rebrickable/set/list'
    refresh_query: str = 'rebrickable/set/need_refresh'

    # Implementation of abstract list method
    def list(self, /, *, override_query: str | None = None, **context) -> None:
        # Load the sets from the database
        for record in self.select(override_query=override_query, **context):
            rebrickable_set = RebrickableSet(record=record)
            self.records.append(rebrickable_set)

    # All the sets
    def all(self, /) -> Self:
        self.list()
        return self

    # Sets needing refresh
    def need_refresh(self, /) -> Self:
        self.list(override_query=self.refresh_query)
        return self