From c183c37124951d30d3f7ed68c27a96c8c0590ffc Mon Sep 17 00:00:00 2001 From: oskvr37 Date: Sat, 8 Feb 2025 16:28:52 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20add=20video=20to=20`TidalResource`?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_utils.py | 2 ++ tiddl/utils.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index a255c21..15b268a 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -9,6 +9,8 @@ class TestTidalResource(unittest.TestCase): positive_cases = [ ("https://tidal.com/browse/track/12345678", "track", "12345678"), ("track/12345678", "track", "12345678"), + ("https://tidal.com/browse/video/12345678", "video", "12345678"), + ("video/12345678", "video", "12345678"), ("https://tidal.com/browse/album/12345678", "album", "12345678"), ("album/12345678", "album", "12345678"), ("https://tidal.com/browse/playlist/12345678", "playlist", "12345678"), diff --git a/tiddl/utils.py b/tiddl/utils.py index a72bbb0..d6bd540 100644 --- a/tiddl/utils.py +++ b/tiddl/utils.py @@ -12,7 +12,7 @@ from typing import Literal, Union, get_args from tiddl.models.constants import TrackQuality, QUALITY_TO_ARG from tiddl.models.resource import Track, Video -ResourceTypeLiteral = Literal["track", "album", "playlist", "artist"] +ResourceTypeLiteral = Literal["track", "video", "album", "playlist", "artist"] class TidalResource(BaseModel):