feat: reworked the genre select

music_kraken/download/components.py

@@ -1,12 +1,25 @@
+from __future__ import annotations
+
 import re
 from pathlib import Path
 from typing import Any, Callable, Dict, Generator, List, Optional
 
+from ..utils import BColors
 from ..utils.config import main_settings
 from ..utils.exception import MKComposeException
 from ..utils.string_processing import unify
 
 
+class HumanIO:
+    @staticmethod
+    def ask_to_create(option: Option) -> bool:
+        return True
+
+    @staticmethod
+    def not_found(key: Any) -> None:
+        return None
+
+
 class Option:
     """
     This could represent a data object, a string or a page.
@@ -56,12 +69,12 @@ class Select:
         option_factory: Callable[[Any], Option] = None,
         raw_options: List[Any] = None,
         parse_option_key: Callable[[Any], Any] = lambda x: x,
-        ask_for_creating_option: Callable[[Option], bool] = lambda x: True,
+        human_io: HumanIO = HumanIO,
         sort: bool = False,
         **kwargs
     ):
         self._parse_option_key: Callable[[Any], Any] = parse_option_key
-        self._ask_for_creating_option: Callable[[Option], bool] = ask_for_creating_option
+        self.human_io: HumanIO = human_io
 
         self._key_to_option: Dict[Any, Option] = dict()
         self._options: List[Option] = []
@@ -117,12 +130,19 @@ class Select:
 
     def choose(self, key: Any) -> Optional[Option]:
         if key not in self:
-            if self.can_create_options and self._ask_for_creating_option(key):
-                return self.create_option(key)
+            if self.can_create_options:
+                c = self.create_option(key)
+                if self.human_io.ask_to_create(c):
+                    return c
+            
+            self.human_io.not_found(key)
             return None
 
         return self[key]
 
+    def pprint(self) -> str:
+        return "\n".join(str(option) for option in self)
+
 
 
 class StringSelect(Select):
@@ -134,7 +154,7 @@ class StringSelect(Select):
         super().__init__(**kwargs)
 
     def next_option(self, value: Any) -> Optional[Option]:
-        o = Option(value=value, keys=[self._current_index], text=f"{self._current_index:0>2}: {value}")
+        o = Option(value=value, keys=[self._current_index], text=f"{BColors.BOLD.value}{self._current_index: >2}{BColors.ENDC.value}: {value}")
         self._current_index += 1
         return o