public/tubearchivist
1
0
Fork 0
mirror of https://git.vectorsigma.ru/public/tubearchivist.git synced 2026-08-04 19:09:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix unknown vid_type query building

Browse Source
This commit is contained in:
Simon
2025-07-17 20:51:59 +07:00
parent c22fe144b0
commit 40f6ee60d4
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,7 +23,7 @@ class VideoQueryBuilder:
VideoTypeEnum.SHORTS: self.shorts_query, VideoTypeEnum.SHORTS: self.shorts_query,
} }
if video_type: if video_type and video_type != VideoTypeEnum.UNKNOWN:
# build query for specific type/s # build query for specific type/s
if not isinstance(video_type, list): if not isinstance(video_type, list):
video_type = [video_type] video_type = [video_type]

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

@@ -77,6 +77,17 @@ def test_build_multiple_queries(default_config, empty_overwrites):
assert result == [(VideoTypeEnum.VIDEOS, 5), (VideoTypeEnum.SHORTS, 2)] assert result == [(VideoTypeEnum.VIDEOS, 5), (VideoTypeEnum.SHORTS, 2)]
def test_build_unknown_queries(default_config, empty_overwrites):
"""vid_type unknown"""
builder = VideoQueryBuilder(default_config, empty_overwrites)
result = builder.build_queries(VideoTypeEnum.UNKNOWN)
assert result == [
(VideoTypeEnum.VIDEOS, 5),
(VideoTypeEnum.STREAMS, 3),
(VideoTypeEnum.SHORTS, 2),
]
def test_overwrite_applied(default_config, overwrites): def test_overwrite_applied(default_config, overwrites):
"""with overwrite from channel config""" """with overwrite from channel config"""
builder = VideoQueryBuilder(default_config, overwrites) builder = VideoQueryBuilder(default_config, overwrites)