Fix API calls

bramstroker · Nov 27, 2022 · 75eedb1 · 75eedb1
1 parent 3f167fc
commit 75eedb1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/utils/measure/config.py b/utils/measure/config.py
@@ -72,6 +72,8 @@
 
 CSV_ADD_DATETIME_COLUMN = config("CSV_ADD_DATETIME_COLUMN", default=False, cast=bool)
 
+SELECTED_DEVICE_TYPE = config("SELECTED_DEVICE_TYPE")
+
 try:
     RESUME = config("RESUME", cast=bool)
 except UndefinedValueError:

diff --git a/utils/measure/measure.py b/utils/measure/measure.py
@@ -92,8 +92,12 @@ def start(self) -> None:
         if DeviceType.LIGHT:
             _LOGGER.info(f"Selected light controller: {config.SELECTED_LIGHT_CONTROLLER}")
 
-        self.device_type = inquirer.list_input("What kind of device do you want to measure the power of?",
-                                     choices=[cls.value for cls in DeviceType])
+        if config.SELECTED_DEVICE_TYPE:
+            self.device_type = DeviceType(config.SELECTED_DEVICE_TYPE)
+        else:
+            self.device_type = inquirer.list_input("What kind of device do you want to measure the power of?",
+                                         choices=[cls.value for cls in DeviceType])
+
         self.runner = RunnerFactory().create_runner(self.device_type)
 
         answers = self.ask_questions()

diff --git a/utils/measure/media_controller/hass.py b/utils/measure/media_controller/hass.py
@@ -18,7 +18,7 @@ def set_volume(self, volume: int) -> None:
         self.client.trigger_service("media_player", "volume_set", entity_id=self._entity_id, volume_level=round(volume/100, 2))
 
     def mute_volume(self) -> None:
-        self.client.trigger_service("media_player", "mute_volume")
+        self.client.trigger_service("media_player", "mute_volume", entity_id=self._entity_id)
 
     def play_audio(self, stream_url: str) -> None:
         self.client.trigger_service(
@@ -30,7 +30,7 @@ def play_audio(self, stream_url: str) -> None:
         )
 
     def turn_off(self) -> None:
-        self.client.trigger_service('media_player', 'turn_off')
+        self.client.trigger_service("media_player", "turn_off", entity_id=self._entity_id)
 
     def get_questions(self) -> list[inquirer.questions.Question]:
         entities = self.client.get_entities()