From eb703f9bd6a445e4063dbb6cfa04178b7f9bee4f Mon Sep 17 00:00:00 2001
From: Teemu Rytilahti <tpr@iki.fi>
Date: Tue, 14 Dec 2021 01:43:33 +0100
Subject: [PATCH] Use @only_on_soundbars for surround_enabled setter

---
 soco/core.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/soco/core.py b/soco/core.py
index 216bdc143..707cf4ecd 100755
--- a/soco/core.py
+++ b/soco/core.py
@@ -211,7 +211,6 @@ class SoCo(_SocoSingletonBase):
         is_bridge
         is_coordinator
         is_soundbar
-        surround_enabled
         is_satellite
         has_satellites
         sub_enabled
@@ -224,6 +223,7 @@ class SoCo(_SocoSingletonBase):
         balance
         night_mode
         dialog_mode
+        surround_enabled
         surround_music_playback
         surround_volume_tv
         surround_volume_music
@@ -1034,16 +1034,13 @@ def surround_enabled(self):
         return bool(int(response["CurrentValue"]))
 
     @surround_enabled.setter
+    @only_on_soundbars
     def surround_enabled(self, enable):
         """Enable/disable the connected surround speakers.
 
         :param enable: Enable or disable surround speakers
         :type enable: bool
         """
-        if not self.is_soundbar:
-            message = "This device does not support surrounds"
-            raise NotSupportedException(message)
-
         self.renderingControl.SetEQ(
             [
                 ("InstanceID", 0),