christofsteel/syng
christofsteel/syng
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity Actions

Configure indexing of source

Browse source
This commit is contained in:
Christoph Stahl 2025-02-02 23:44:30 +01:00
parent 7488b260e1
commit dfbecc1517
2 changed files with 21 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
syng/sources/filebased.py
View file

@ -15,7 +15,7 @@ except ImportError:
PYMEDIAINFO_AVAILABLE = False PYMEDIAINFO_AVAILABLE = False
from .source import Source from .source import Source
from ..config import ListStrOption, ConfigOption from ..config import BoolOption, ListStrOption, ConfigOption
class FileBasedSource(Source): class FileBasedSource(Source):
@ -31,7 +31,8 @@ class FileBasedSource(Source):
ListStrOption(), ListStrOption(),
"List of filename extensions\n(mp3+cdg, mp4, ...)", "List of filename extensions\n(mp3+cdg, mp4, ...)",
["mp3+cdg"], ["mp3+cdg"],
) ),
"build_index": ConfigOption(BoolOption(), "Build index on startup", True),
} }
def __init__(self, config: dict[str, Any]): def __init__(self, config: dict[str, Any]):

15
syng/sources/source.py
View file

@ -128,6 +128,7 @@ class Source(ABC):
self._index: list[str] = config["index"] if "index" in config else [] self._index: list[str] = config["index"] if "index" in config else []
self.extra_mpv_options: dict[str, str] = {} self.extra_mpv_options: dict[str, str] = {}
self._skip_next = False self._skip_next = False
self.build_index = config.get("build_index", False)
async def get_entry( async def get_entry(
self, self,
@ -361,13 +362,16 @@ class Source(ABC):
:rtype: Optional[dict[str, Any] | list[dict[str, Any]] :rtype: Optional[dict[str, Any] | list[dict[str, Any]]
""" """
if not self.build_index:
return None
logger.warning(f"{self.source_name}: updating index") logger.warning(f"{self.source_name}: updating index")
new_index = await self.update_file_list() new_index = await self.update_file_list()
logger.warning(f"{self.source_name}: done") logger.warning(f"{self.source_name}: done")
if new_index is not None: if new_index is not None:
self._index = new_index self._index = new_index
chunked = zip_longest(*[iter(new_index)] * 1000, fillvalue="") # chunked = zip_longest(*[iter(new_index)] * 1000, fillvalue="")
return [{"index": list(filter(lambda x: x != "", chunk))} for chunk in chunked] # return [{"index": list(filter(lambda x: x != "", chunk))} for chunk in chunked]
return await self.get_config()
return None return None
async def get_config(self) -> dict[str, Any] | list[dict[str, Any]]: async def get_config(self) -> dict[str, Any] | list[dict[str, Any]]:
@ -388,13 +392,18 @@ class Source(ABC):
:return: The part of the config, that should be sended to the server. :return: The part of the config, that should be sended to the server.
:rtype: dict[str, Any] | list[dict[str, Any]] :rtype: dict[str, Any] | list[dict[str, Any]]
""" """
packages = []
if self.build_index:
if not self._index: if not self._index:
self._index = [] self._index = []
logger.warning(f"{self.source_name}: generating index") logger.warning(f"{self.source_name}: generating index")
self._index = await self.get_file_list() self._index = await self.get_file_list()
logger.warning(f"{self.source_name}: done") logger.warning(f"{self.source_name}: done")
chunked = zip_longest(*[iter(self._index)] * 1000, fillvalue="") chunked = zip_longest(*[iter(self._index)] * 1000, fillvalue="")
return [{"index": list(filter(lambda x: x != "", chunk))} for chunk in chunked] packages = [{"index": list(filter(lambda x: x != "", chunk))} for chunk in chunked]
if len(packages) == 1:
return first_package
return packages
def add_to_config(self, config: dict[str, Any], running_number: int) -> None: def add_to_config(self, config: dict[str, Any], running_number: int) -> None:
""" """