Fix mood handling

pytradfri/__main__.py

@@ -73,12 +73,6 @@
     else:
         print("No groups found!")
         group = None
-    moods = api(gateway.get_moods())
-    if moods:
-        mood = moods[0]
-    else:
-        print("No moods found!")
-        mood = None
     tasks = api(gateway.get_smart_tasks())
     homekit_id = api(gateway.get_gateway_info()).homekit_id
 
@@ -112,7 +106,6 @@ def dump_devices():
         print("> tasks[0].repeat_days_list")
     print("> api(gateway.reboot())")
     print("> groups")
-    print("> moods")
     print("> tasks")
     print("> dump_devices()")
     print("> dump_all()")

pytradfri/gateway.py

@@ -103,19 +103,18 @@ def process_result(result):
                        [ROOT_GATEWAY, ATTR_GATEWAY_INFO],
                        process_result=process_result)
 
-    def get_moods(self):
+    def get_moods(self, group_id):
         """
-        Return moods defined on the gateway.
+        Return moods available in given group.
 
         Returns a Command.
         """
-        mood_parent = self._get_mood_parent()
 
         def process_result(result):
-            return [self.get_mood(mood, mood_parent=mood_parent) for mood in
+            return [self.get_mood(mood, mood_parent=group_id) for mood in
                     result]
 
-        return Command('get', [ROOT_MOODS, mood_parent],
+        return Command('get', [ROOT_MOODS, group_id],
                        process_result=process_result)
 
     def get_mood(self, mood_id, *, mood_parent=None):
@@ -124,26 +123,13 @@ def get_mood(self, mood_id, *, mood_parent=None):
 
         Returns a Command.
         """
-        if mood_parent is None:
-            mood_parent = self._get_mood_parent()
 
         def process_result(result):
             return Mood(result, mood_parent)
 
         return Command('get', [ROOT_MOODS, mood_parent, mood_id],
                        mood_parent, process_result=process_result)
 
-    def _get_mood_parent(self):
-        """
-        Get the parent of all moods.
-
-        Returns a Command.
-        """
-        def process_result(result):
-            return result[0]
-
-        return Command('get', [ROOT_MOODS], process_result=process_result)
-
     def get_smart_tasks(self):
         """
         Return the transitions linked to the gateway.

pytradfri/group.py

@@ -67,14 +67,19 @@ def members(self):
         """Return device objects of members of this group."""
         return [self._gateway.get_device(dev) for dev in self.member_ids]
 
+    def moods(self):
+        """Return mood objects of moods in this group."""
+        return self._gateway.get_moods(self.id)
+
     def mood(self):
         """"Active mood."""
-        return self._gateway.get_mood(self.mood_id)
+        return self._gateway.get_mood(self.mood_id, mood_parent=self.id)
 
     def activate_mood(self, mood_id):
         """Activate a mood."""
         return self.set_values({
-            ATTR_MOOD: mood_id
+            ATTR_MOOD: mood_id,
+            ATTR_DEVICE_STATE: int(self.state)
         })
 
     def set_state(self, state):

tests/moods.py

@@ -0,0 +1,9 @@
+#  Retrieved from Gateway running on 1.9.27
+MOOD = {
+    '9001': 'FOCUS',
+    '9002': 1577189497,
+    '9003': 196625,
+    '9057': 2,
+    '9068': 1,
+    '15013': [{'5850':1,'5851':254,'9003':65547}]
+}

tests/test_group.py

@@ -6,14 +6,19 @@
     ATTR_LIGHT_MIREDS,
     ATTR_LIGHT_COLOR_HUE,
     ATTR_LIGHT_COLOR_SATURATION,
-    ATTR_LIGHT_DIMMER
+    ATTR_LIGHT_DIMMER,
+    ROOT_MOODS
 )
 from pytradfri.group import Group
+from pytradfri.gateway import Gateway
 
+@pytest.fixture
+def gateway():
+    return Gateway()
 
 @pytest.fixture
-def group():
-    return Group(GROUP)
+def group(gateway):
+    return Group(gateway, GROUP)
 
 
 def test_setters():
@@ -48,3 +53,7 @@ def test_setters():
         ATTR_LIGHT_COLOR_SATURATION: 200,
         ATTR_LIGHT_DIMMER: 100,
     }
+
+def test_moods(group):
+    cmd = group.moods()
+    assert cmd.path == [ROOT_MOODS, group.id]

tests/test_mood.py

@@ -0,0 +1,12 @@
+import pytest
+from moods import MOOD
+
+from pytradfri.const import ROOT_MOODS
+from pytradfri.mood import Mood
+
+@pytest.fixture
+def mood():
+    return Mood(MOOD, 131080)
+
+def test_mood_properties(mood):
+    assert mood.path == [ROOT_MOODS, 131080, 196625]