RomanNum3ral
/
tubearchivist
mirror of https://github.com/tubearchivist/tubearchivist.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix unknown type query building

This commit is contained in:
Simon 2025-09-27 15:20:54 +07:00
parent e716067429
commit 5efd6ff8f2
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/channel/src/remote_query.py
View File

@ -102,7 +102,7 @@ def get_last_channel_videos(
builder = VideoQueryBuilder(config, limit=limit)
queries = []
if query_filter is None:
if query_filter is None or query_filter == VideoTypeEnum.UNKNOWN:
queries = builder.build_queries()
elif isinstance(query_filter, list):
queries = builder.build_queries(vid_types=query_filter)
@ -113,7 +113,7 @@ def get_last_channel_videos(
last_videos: list[dict] = []
if not query:
if not queries:
return last_videos
for vid_type_enum, limit_amount in queries:

11
backend/channel/tests/test_src/test_remote_query.py
View File

@ -68,6 +68,17 @@ def test_build_specific_query(default_config, empty_overwrites):
assert result == (VideoTypeEnum.VIDEOS, 5)
def test_build_unknown_type(default_config, empty_overwrites):
"""unknown vid_type build list"""
builder = VideoQueryBuilder(default_config, empty_overwrites, limit=None)
result = builder.build_queries()
assert result == [
(VideoTypeEnum.VIDEOS, None),
(VideoTypeEnum.STREAMS, None),
(VideoTypeEnum.SHORTS, None),
]
def test_build_multiple_queries(default_config, empty_overwrites):
"""vid_type list"""
builder = VideoQueryBuilder(default_config, empty_overwrites)