Добавил .rst доку, значение по умолчанию

MarshalX · Sep 24, 2024 · 25ae6d6 · 25ae6d6
1 parent 7c4165c
commit 25ae6d6
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 4 deletions.
diff --git a/docs/source/yandex_music.artist.rst b/docs/source/yandex_music.artist.rst
@@ -20,4 +20,5 @@ Submodules
    yandex_music.artist.description
    yandex_music.artist.link
    yandex_music.artist.ratings
+   yandex_music.artist.stats
    yandex_music.artist.vinyl
diff --git a/docs/source/yandex_music.artist.stats.rst b/docs/source/yandex_music.artist.stats.rst
@@ -0,0 +1,7 @@
+yandex\_music.artist.stats
+==========================
+
+.. automodule:: yandex_music.artist.stats
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/tests/test_brief_info.py b/tests/test_brief_info.py
@@ -30,7 +30,9 @@ class TestBriefInfo:
     concerts = None
     has_promotions = False
 
-    def test_expected_values(self, brief_info, artist, track, album, playlist, cover, playlist_id, video, chart, vinyl):
+    def test_expected_values(
+        self, brief_info, artist, track, album, playlist, cover, playlist_id, video, chart, vinyl, stats
+    ):
         assert brief_info.artist == artist
         assert brief_info.albums == [album]
         assert brief_info.playlists == [playlist]
@@ -45,6 +47,7 @@ def test_expected_values(self, brief_info, artist, track, album, playlist, cover
         assert brief_info.vinyls == [vinyl]
         assert brief_info.has_promotions == self.has_promotions
         assert brief_info.playlist_ids == [playlist_id]
+        assert brief_info.stats == stats
         assert brief_info.tracks_in_chart == [chart]
 
     def test_de_json_none(self, client):
@@ -84,7 +87,7 @@ def test_de_json_required(self, client, artist, track, album, playlist, cover, p
         assert brief_info.has_promotions == self.has_promotions
         assert brief_info.playlist_ids == [playlist_id]
 
-    def test_de_json_all(self, client, artist, track, album, playlist, cover, playlist_id, video, chart, vinyl):
+    def test_de_json_all(self, client, artist, track, album, playlist, cover, playlist_id, video, chart, vinyl, stats):
         json_dict = {
             'artist': artist.to_dict(),
             'albums': [album.to_dict()],
@@ -101,6 +104,7 @@ def test_de_json_all(self, client, artist, track, album, playlist, cover, playli
             'playlist_ids': [playlist_id.to_dict()],
             'tracks_in_chart': [chart.to_dict()],
             'playlists': [playlist.to_dict()],
+            'stats': stats.to_dict(),
         }
         brief_info = BriefInfo.de_json(json_dict, client)
 
@@ -118,6 +122,7 @@ def test_de_json_all(self, client, artist, track, album, playlist, cover, playli
         assert brief_info.vinyls == [vinyl]
         assert brief_info.has_promotions == self.has_promotions
         assert brief_info.playlist_ids == [playlist_id]
+        assert brief_info.stats == stats
         assert brief_info.tracks_in_chart == [chart]
 
     def test_equality(self, artist, track, album, playlist, cover, playlist_id, video, vinyl, stats):

diff --git a/yandex_music/__init__.py b/yandex_music/__init__.py
@@ -170,6 +170,7 @@
     'Experiments',
     'Cover',
     'Ratings',
+    'Stats',
     'Counts',
     'Link',
     'Artist',
@@ -275,5 +276,4 @@
     'CustomWave',
     'R128',
     'LyricsInfo',
-    'Stats',
 ]
diff --git a/yandex_music/artist/brief_info.py b/yandex_music/artist/brief_info.py
@@ -59,7 +59,7 @@ class BriefInfo(YandexMusicModel):
     vinyls: List['Vinyl']
     has_promotions: bool
     playlist_ids: List['PlaylistId']
-    stats: Optional['Stats']
+    stats: Optional['Stats'] = None
     tracks_in_chart: List['Chart'] = field(default_factory=list)
     client: Optional['ClientType'] = None