feat: added missed get_cover_image_url for LandingPlaylist

KatantDev · Dec 28, 2023 · 3fc3acc · 3fc3acc
1 parent 0d165a8
commit 3fc3acc
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ymdantic"
-version = "0.1.6"
+version = "0.1.7"
 description = "Фреймворк предоставляющий доступ к API Яндекс Музыки (с Pydantic валидацией)"
 authors = ["KatantDev <[email protected]>"]
 readme = "README.md"

diff --git a/ymdantic/models/landing/cover.py b/ymdantic/models/landing/cover.py
@@ -29,7 +29,7 @@ def get_image_url(self, size: str = "200x200") -> HttpUrl:
 class LandingCover(YMBaseModel):
     """Pydantic модель, представляющая информацию об обложке на лендинге."""
 
-    uri: str  # TODO: get_url
+    uri: str
     # URI обложки.
     color: str
     # Основной цвет обложки.

diff --git a/ymdantic/models/landing/playlist.py b/ymdantic/models/landing/playlist.py
@@ -1,5 +1,7 @@
 from typing import Literal
 
+from pydantic import HttpUrl
+
 from ymdantic.models.base import YMBaseModel
 from ymdantic.models.landing.cover import LandingPlaylistCover
 
@@ -18,6 +20,15 @@ class LandingPlaylist(YMBaseModel):
     cover: LandingPlaylistCover
     # Обложка плейлиста. Содержит информацию о цветах обложки и URI обложки.
 
+    def get_cover_image_url(self, size: str = "200x200") -> HttpUrl:
+        """
+        Возвращает ссылку на изображение обложки.
+
+        :param size: Размер изображения.
+        :return: Ссылка на изображение обложки.
+        """
+        return self.cover.get_image_url(size)
+
 
 class LandingPlaylistItemData(YMBaseModel):
     """Pydantic модель, представляющая информацию о плейлисте на главной."""