From 377f2cca8afcdcd03efad1cd9390bce339895dca Mon Sep 17 00:00:00 2001 From: Hellow2 Date: Wed, 29 Mar 2023 09:50:06 +0200 Subject: [PATCH] started choose function --- src/music_kraken/pages/download_center/search.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/src/music_kraken/pages/download_center/search.py b/src/music_kraken/pages/download_center/search.py index fd040ef..0307b84 100644 --- a/src/music_kraken/pages/download_center/search.py +++ b/src/music_kraken/pages/download_center/search.py @@ -148,3 +148,18 @@ class Search: for page in self.pages: self._current_option[page] = page.search_by_query(query=query) + + def choose_page(self, page: Type[Page]) -> MultiPageOptions: + pass + + def choose_index(self, index: int) -> MultiPageOptions: + pass + + def choose(self, choosen: Union[Type[Page], int]) -> MultiPageOptions: + if type(choosen) == int: + return self.choose_index(choosen) + + if choosen in page_attributes.ALL_PAGES: + return self.choose_page(choosen) + + raise ValueError("choose is neiter an integer, nor a page in page_attributes.ALL_PAGES.")