123 lines
3.6 KiB
Python
123 lines
3.6 KiB
Python
from typing import Tuple, Type, Dict, List, Generator, Union
|
|
from dataclasses import dataclass
|
|
|
|
from ..objects import DatabaseObject
|
|
from ..utils.enums.source import SourcePages
|
|
from ..pages import Page, EncyclopaediaMetallum, Musify
|
|
|
|
|
|
@dataclass
|
|
class Option:
|
|
index: int
|
|
music_object: DatabaseObject
|
|
|
|
|
|
class Results:
|
|
def __init__(self, max_items_per_page: int = 10, **kwargs) -> None:
|
|
self._by_index: Dict[int, DatabaseObject] = dict()
|
|
self._page_by_index: Dict[int: Type[Page]] = dict()
|
|
|
|
self.max_items_per_page = max_items_per_page
|
|
|
|
def __iter__(self) -> Generator[DatabaseObject, None, None]:
|
|
for option in self.formatted_generator():
|
|
if isinstance(option, Option):
|
|
yield option.music_object
|
|
|
|
def formatted_generator(self) -> Generator[Union[Type[Page], Option], None, None]:
|
|
self._by_index = dict()
|
|
self._page_by_index = dict()
|
|
|
|
def __len__(self) -> int:
|
|
return max(self._by_index.keys())
|
|
|
|
def __getitem__(self, index: int):
|
|
return self._by_index[index]
|
|
|
|
|
|
class SearchResults(Results):
|
|
def __init__(
|
|
self,
|
|
pages: Tuple[Type[Page], ...] = None,
|
|
**kwargs,
|
|
) -> None:
|
|
super().__init__(**kwargs)
|
|
|
|
self.pages = pages or []
|
|
# this would initialize a list for every page, which I don't think I want
|
|
# self.results = Dict[Type[Page], List[DatabaseObject]] = {page: [] for page in self.pages}
|
|
self.results: Dict[Type[Page], List[DatabaseObject]] = {}
|
|
|
|
def add(self, page: Type[Page], search_result: List[DatabaseObject]):
|
|
"""
|
|
adds a list of found music objects to the according page
|
|
WARNING: if a page already has search results, they are just gonna be overwritten
|
|
"""
|
|
|
|
self.results[page] = search_result
|
|
|
|
def get_page_results(self, page: Type[Page]) -> "PageResults":
|
|
return PageResults(page, self.results.get(page, []))
|
|
|
|
def __len__(self) -> int:
|
|
return sum(min(self.max_items_per_page, len(results)) for results in self.results.values())
|
|
|
|
def formatted_generator(self):
|
|
super().formatted_generator()
|
|
i = 0
|
|
|
|
for page in self.results:
|
|
yield page
|
|
|
|
j = 0
|
|
for option in self.results[page]:
|
|
yield Option(i, option)
|
|
self._by_index[i] = option
|
|
self._page_by_index[i] = page
|
|
i += 1
|
|
j += 1
|
|
|
|
if j >= self.max_items_per_page:
|
|
break
|
|
|
|
|
|
class GoToResults(Results):
|
|
def __init__(self, results: List[DatabaseObject], **kwargs):
|
|
self.results: List[DatabaseObject] = results
|
|
|
|
super().__init__(**kwargs)
|
|
|
|
def __getitem__(self, index: int):
|
|
return self.results[index]
|
|
|
|
def __len__(self) -> int:
|
|
return len(self.results)
|
|
|
|
def formatted_generator(self):
|
|
yield from (Option(i, o) for i, o in enumerate(self.results))
|
|
|
|
|
|
|
|
class PageResults(Results):
|
|
def __init__(self, page: Type[Page], results: List[DatabaseObject], **kwargs) -> None:
|
|
super().__init__(**kwargs)
|
|
|
|
self.page: Type[Page] = page
|
|
self.results: List[DatabaseObject] = results
|
|
|
|
|
|
def formatted_generator(self, max_items_per_page: int = 10):
|
|
super().formatted_generator()
|
|
i = 0
|
|
|
|
yield self.page
|
|
|
|
for option in self.results:
|
|
yield Option(i, option)
|
|
self._by_index[i] = option
|
|
self._page_by_index[i] = self.page
|
|
i += 1
|
|
|
|
def __len__(self) -> int:
|
|
return len(self.results)
|