From 03061d1a20b37b8d6bb09375e7126f3cd1a32f98 Mon Sep 17 00:00:00 2001
From: Il`ya <marshal@xaker.ru>
Date: Sun, 2 Feb 2020 14:28:55 +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=20Restrictions?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_restrictions.py           |  2 +-
 yandex_music/rotor/restrictions.py   | 34 ++++++++++++++++++++++------
 yandex_music/rotor/rotor_settings.py |  4 ++--
 3 files changed, 30 insertions(+), 10 deletions(-)

diff --git a/tests/test_restrictions.py b/tests/test_restrictions.py
index a1b7910..c9e3869 100644
--- a/tests/test_restrictions.py
+++ b/tests/test_restrictions.py
@@ -32,7 +32,7 @@ class TestRestrictions:
 
     def test_equality(self, enum, discrete_scale):
         a = Restrictions(enum, enum)
-        b = Restrictions(enum, None, discrete_scale, discrete_scale, discrete_scale, enum)
+        b = Restrictions(enum, None)
         c = Restrictions(enum, enum)
 
         assert a != b
diff --git a/yandex_music/rotor/restrictions.py b/yandex_music/rotor/restrictions.py
index 0c114b8..2835c01 100644
--- a/yandex_music/rotor/restrictions.py
+++ b/yandex_music/rotor/restrictions.py
@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Optional, Union
+from typing import TYPE_CHECKING, Optional
 
 from yandex_music import YandexMusicObject, Enum, DiscreteScale
 
@@ -12,24 +12,44 @@ de_json = {
 
 
 class Restrictions(YandexMusicObject):
-    """Класс, представляющий .
+    """Класс, представляющий ограничения для настроек станции.
 
     Attributes:
+        language (:obj:`yandex_music.Enum`): Объект класса :class:`yandex_music.Enum`, представляющий перечисление
+            значений для языков.
+        diversity (:obj:`yandex_music.Enum`): Объект класса :class:`yandex_music.Enum`, представляющий перечисление
+            значений для разнообразия (треков).
+        mood (:obj:`yandex_music.DiscreteScale`): Объект класса :class:`yandex_music.DiscreteScale`, представляющий
+            ограничения для значения настроения.
+        energy (:obj:`yandex_music.DiscreteScale`): Объект класса :class:`yandex_music.DiscreteScale`, представляющий
+            ограничения для значения энергичности.
+        mood_energy (:obj:`yandex_music.Enum`): Объект класса :class:`yandex_music.Enum`, представляющий перечисление
+            значений для настроения.
         client (:obj:`yandex_music.Client`): Объект класса :class:`yandex_music.Client`, представляющий клиент
                 Yandex Music.
 
     Args:
+        language (:obj:`yandex_music.Enum`): Объект класса :class:`yandex_music.Enum`, представляющий перечисление
+            значений для языков.
+        diversity (:obj:`yandex_music.Enum`): Объект класса :class:`yandex_music.Enum`, представляющий перечисление
+            значений для разнообразия (треков).
+        mood (:obj:`yandex_music.DiscreteScale`, optional): Объект класса :class:`yandex_music.DiscreteScale`,
+            представляющий ограничения для значения настроения.
+        energy (:obj:`yandex_music.DiscreteScale`, optional): Объект класса :class:`yandex_music.DiscreteScale`,
+            представляющий ограничения для значения энергичности.
+        mood_energy (:obj:`yandex_music.Enum`, optional): Объект класса :class:`yandex_music.Enum`, представляющий
+            перечисление значений для настроения.
         client (:obj:`yandex_music.Client`, optional): Объект класса :class:`yandex_music.Client`, представляющий клиент
             Yandex Music.
         **kwargs: Произвольные ключевые аргументы полученные от API.
     """
 
     def __init__(self,
-                 language: Optional[Union['Enum', 'DiscreteScale']],
-                 diversity: Optional[Union['Enum', 'DiscreteScale']],
-                 mood: Optional[Union['Enum', 'DiscreteScale']] = None,
-                 energy: Optional[Union['Enum', 'DiscreteScale']] = None,
-                 mood_energy: Optional[Union['Enum', 'DiscreteScale']] = None,
+                 language: Optional['Enum'],
+                 diversity: Optional['Enum'],
+                 mood: Optional['DiscreteScale'] = None,
+                 energy: Optional['DiscreteScale'] = None,
+                 mood_energy: Optional['Enum'] = None,
                  client: Optional['Client'] = None,
                  **kwargs) -> None:
         self.language = language
diff --git a/yandex_music/rotor/rotor_settings.py b/yandex_music/rotor/rotor_settings.py
index ec35800..a0c6ac8 100644
--- a/yandex_music/rotor/rotor_settings.py
+++ b/yandex_music/rotor/rotor_settings.py
@@ -21,7 +21,7 @@ class RotorSettings(YandexMusicObject):
     Attributes:
         language (:obj:`str`): Язык.
         diversity (:obj:`str`): Разнообразие (треки).
-        mood (:obj:`int`): Грустное.
+        mood (:obj:`int`): Настроение (старое).
         energy (:obj:`int`): Энергичное.
         mood_energy (:obj:`str`): Настроение.
         client (:obj:`yandex_music.Client`): Объект класса :class:`yandex_music.Client`, представляющий клиент
@@ -30,7 +30,7 @@ class RotorSettings(YandexMusicObject):
     Args:
         language (:obj:`str`): Язык.
         diversity (:obj:`str`): Разнообразие (треки).
-        mood (:obj:`int`, optional): Грустное.
+        mood (:obj:`int`, optional): Настроение (старое).
         energy (:obj:`int`, optional): Энергичное.
         mood_energy (:obj:`str`, optional): Настроение.
         client (:obj:`yandex_music.Client`, optional): Объект класса :class:`yandex_music.Client`, представляющий клиент