From 8a9b230fac479f5173344fbc4d6827f148335c3f Mon Sep 17 00:00:00 2001
From: Il`ya <marshal@xaker.ru>
Date: Sun, 2 Feb 2020 13:55:00 +0300
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=BA=D1=83=D0=BC=D0=B5=D0=BD?=
 =?UTF-8?q?=D1=82=D0=B0=D1=86=D0=B8=D1=8F=20=D0=BA=D0=BB=D0=B0=D1=81=D1=81?=
 =?UTF-8?q?=D0=B0=20RotorSettings?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_rotor_settings.py         |  2 +-
 yandex_music/rotor/rotor_settings.py | 23 +++++++++++++++++++++--
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/tests/test_rotor_settings.py b/tests/test_rotor_settings.py
index a17804c..d384d8a 100644
--- a/tests/test_rotor_settings.py
+++ b/tests/test_rotor_settings.py
@@ -6,7 +6,7 @@ class TestRotorSettings:
     diversity = 'default'
     mood = 2
     energy = 2
-    mood_energy = None
+    mood_energy = 'fun'
 
     def test_expected_values(self, rotor_settings):
         assert rotor_settings.language == self.language
diff --git a/yandex_music/rotor/rotor_settings.py b/yandex_music/rotor/rotor_settings.py
index f499cb8..ec35800 100644
--- a/yandex_music/rotor/rotor_settings.py
+++ b/yandex_music/rotor/rotor_settings.py
@@ -7,13 +7,32 @@ if TYPE_CHECKING:
 
 
 class RotorSettings(YandexMusicObject):
-    """Класс, представляющий .
+    """Класс, представляющий настройки станции.
+
+    Note:
+        Поля `energy`, `mood` используются в старых настройках (`settings1`).
+
+        Значения `mood_energy`: `fun`, `active`, `calm`, `sad`, `all`.
+
+        Значения `diversity`: `favorite`, `popular`, `discover`, `default`.
+
+        Значения `language`: `not-russian`, `russian`, `any`.
 
     Attributes:
+        language (:obj:`str`): Язык.
+        diversity (:obj:`str`): Разнообразие (треки).
+        mood (:obj:`int`): Грустное.
+        energy (:obj:`int`): Энергичное.
+        mood_energy (:obj:`str`): Настроение.
         client (:obj:`yandex_music.Client`): Объект класса :class:`yandex_music.Client`, представляющий клиент
                 Yandex Music.
 
     Args:
+        language (:obj:`str`): Язык.
+        diversity (:obj:`str`): Разнообразие (треки).
+        mood (:obj:`int`, optional): Грустное.
+        energy (:obj:`int`, optional): Энергичное.
+        mood_energy (:obj:`str`, optional): Настроение.
         client (:obj:`yandex_music.Client`, optional): Объект класса :class:`yandex_music.Client`, представляющий клиент
             Yandex Music.
         **kwargs: Произвольные ключевые аргументы полученные от API.
@@ -24,7 +43,7 @@ class RotorSettings(YandexMusicObject):
                  diversity: str,
                  mood: Optional[int] = None,
                  energy: Optional[int] = None,
-                 mood_energy=None,
+                 mood_energy: Optional[str] = None,
                  client: Optional['Client'] = None,
                  **kwargs) -> None:
         self.language = language