change mode in tui

Author: paranoik1

image.png

@@ -1,10 +1,10 @@
 import argparse
 import asyncio
 
-from .core import Mode, search
+from .core import SearchMode, search
 
 
-async def main(query: str, mode: Mode) -> None:
+async def main(query: str, mode: SearchMode) -> None:
     async for sound in search(query, mode=mode):
         print(
             f"Name: {sound.title}\n"
@@ -34,15 +34,15 @@ def execute() -> None:
         help="Режим поиска: fast (только первые страницы) или full (все страницы)",
     )
     args = parser.parse_args()
+    mode = SearchMode(args.mode)
 
     if args.tui:
         from .textual_app import MsocApp
 
-        app = MsocApp()
+        app = MsocApp(search_mode=mode)
         app.run()
         return
 
-    mode = Mode(args.mode)
     query = args.query if args.query else input("Запрос: ")
     asyncio.run(main(query, mode))
 

src/msoc/__main__.py

@@ -10,7 +10,7 @@
 from .sound import Sound
 
 __all__ = [
-    "Mode",
+    "SearchMode",
     "search",
     "get_engines",
     "register_engine",
@@ -20,7 +20,7 @@
 ]
 
 
-class Mode(Enum):
+class SearchMode(Enum):
     Fast = "fast"
     Full = "full"
 
@@ -86,7 +86,7 @@ def unload_search_engine(name: str) -> None:
     logger.info("Удалён движок: %s", name)
 
 
-async def search(query: str, mode: Mode = Mode.Fast) -> AsyncGenerator[Sound, None]:
+async def search(query: str, mode: SearchMode = SearchMode.Fast) -> AsyncGenerator[Sound, None]:
     """
     Запускает параллельный поиск музыки по всем зарегистрированным движкам.
 
@@ -95,17 +95,19 @@ async def search(query: str, mode: Mode = Mode.Fast) -> AsyncGenerator[Sound, No
 
     Args:
         query: Поисковый запрос (название трека, исполнитель или и то и другое).
-        mode: Режим поиска — ``Fast`` (только первые страницы) или ``Full``
-            (все страницы через ``search_full``, если он реализован).
+
+        mode: Режим поиска — Fast (только первые страницы) или Full
+            (все страницы через search_full, если он реализован).
 
     Yields:
         Sound — унифицированная информация о найденном треке.
-        Поле ``_engine`` автоматически заполняется именем движка-источника.
+
+        Поле _engine автоматически заполняется именем движка-источника.
 
     Note:
         Если ни один движок не зарегистрирован, генератор немедленно завершится.
-        Ошибки отдельных движков логируются на уровне CRITICAL и не прерывают
-        поиск остальными движками.
+        
+        Ошибки отдельных движков логируются и не прерывают поиск остальными движками.
     """
     engines = get_engines()
     queue: asyncio.Queue[Sound | None] = asyncio.Queue()
@@ -156,7 +158,7 @@ async def _engine_search(
             )
             continue
 
-        if mode is Mode.Full:
+        if mode is SearchMode.Full:
             search_full = getattr(engine_module, "search_full", None)
             if search_full is not None:
                 callback = search_full

src/msoc/core.py

@@ -1,4 +1,5 @@
 import os
+import re
 import queue
 import string
 import subprocess
@@ -10,15 +11,17 @@
 from functools import cached_property, cache
 from enum import Enum
 import sounddevice as sd  # type: ignore[import-untyped]
-from textual.app import App, ComposeResult
+from textual.app import App, ComposeResult, Binding
 from textual.widgets import Header, Footer, Input, Button, Label
+from textual.reactive import reactive
 from textual.containers import VerticalScroll, VerticalGroup, HorizontalGroup
 
-from msoc import Sound, search
+from msoc import Sound, search, SearchMode
 
 
 logger = logging.getLogger()
 
+
 Channels = int
 SampleRate = float
 Duration = float
@@ -130,9 +133,6 @@ def _stream_play_ffmpeg(self, sound_filepath: str, channels: Channels, samplerat
         ffmpeg_cmd = [
             "ffmpeg",
             "-y",  # Перезаписывать sound_filepath без вопросов
-            # "-reconnect", "1",
-            # "-reconnect_streamed", "1",
-            # "-reconnect_delay_max", "2",
             "-i", sound_filepath,
             # для sounddeivce (numpy тип)
             "-f", "f32le", "-acodec", "pcm_f32le", "-ac", str(channels), "-ar", str(samplerate), "pipe:1",
@@ -469,26 +469,44 @@ class MsocApp(App):
     }
     """
 
-    def __init__(self):
+    suffix_title_pattern = re.compile(r' \([a-zA-Z]* mode\)$')
+
+    BINDINGS = [
+        Binding('m', 'toggle_search_mode', "Change Search Mode")
+    ]
+
+    def __init__(self, search_mode: SearchMode = SearchMode.Fast):
         super().__init__()
         self.player = AudioPlayer(self)
+        self.search_mode = search_mode
+        self.update_title()
+
+    def update_title(self, title: str | None = None):
+        if not title:
+            title = self.suffix_title_pattern.sub('', self.title)
+
+        self.title = title + f' ({self.search_mode.value.title()} mode)'
+
+    def action_toggle_search_mode(self):
+        self.search_mode = SearchMode.Fast if self.search_mode == SearchMode.Full else SearchMode.Full
+        self.update_title()
 
     def on_input_submitted(self, event: Input.Submitted):
         if event.control.id != "search":
             return
-        self.title = "Идет поиск..."
+        self.update_title("Идет поиск...")
         event.control.disabled = True
         self.run_worker(self.search_task(event.value), name="search")
 
     async def search_task(self, query: str):
         list_sounds_container = self.query_one("#list-sounds")
         list_sounds_container.remove_children()
 
-        async for sound in search(query):
+        async for sound in search(query, mode=self.search_mode):
             sound_widget = SoundWidget(sound, self.player)
             await list_sounds_container.mount(sound_widget)
 
-        self.title = "Поиск завершен"
+        self.update_title("Поиск завершен")
         self.query_one("#search").disabled = False
 
     def compose(self) -> ComposeResult:

src/msoc/textual_app.py

@@ -3,14 +3,16 @@
 
 logger = logging.getLogger("msoc")
 
+http_pattern = re.compile(r'^https?://')
+
 
 def validate_and_return_absolute_url(sound_url: str, base_url: str) -> str:
     """
     Преобразует относительный URL в абсолютный на основе base_url.
 
     Args:
         sound_url: Относительный или абсолютный URL трека.
-        
+
         base_url: Базовый URL сайта для преобразования относительных ссылок.
 
     Returns:
@@ -20,11 +22,10 @@ def validate_and_return_absolute_url(sound_url: str, base_url: str) -> str:
         ValueError: Если base_url не является корректным абсолютным URL.
     """
 
-    pattern = re.compile(r'^https?://')
-    if pattern.search(sound_url):
+    if http_pattern.search(sound_url):
         return sound_url
 
-    if not pattern.search(base_url):
+    if not http_pattern.search(base_url):
         raise ValueError('Базовый url должен быть абсолютным')
 
     url_fragments = base_url.split('/')