Merge pull request #31 from brettonw/1.09-x

1.09 x

.gitignore

@@ -2,4 +2,10 @@
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
-*$py.class
+*$py.class
+
+# MacOS drops these files all over the place to store finder view preferences. They can be safelu deleted.
+.DS_Store
+
+# Intellij IDEA stores project information in this directory
+.idea

custom_components/apex/__init__.py

@@ -6,7 +6,6 @@
 import async_timeout
 import voluptuous as vol
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import callback
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
@@ -32,7 +31,6 @@
 _LOGGER = logging.getLogger(__name__)
 
 
-
 async def async_setup(hass: HomeAssistant, config: dict):
     """Set up the Apex component."""
     hass.data.setdefault(DOMAIN, {})
@@ -56,7 +54,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
 
     await coordinator.async_refresh()  # Get initial data
 
-
     if not coordinator.last_update_success:
         raise ConfigEntryNotReady
 
@@ -73,27 +70,35 @@ async def async_set_options_service(service_call):
     async def async_set_variable_service(service_call):
         await hass.async_add_executor_job(set_variable, hass, service_call, coordinator)
 
+    async def async_set_dos_rate_service(service_call):
+        await hass.async_add_executor_job(set_dos_rate, hass, service_call, coordinator)
+
     hass.services.async_register(
         DOMAIN,
-        "set_output", 
+        "set_output",
         async_set_options_service
     )
 
     hass.services.async_register(
         DOMAIN,
-        "set_variable", 
+        "set_variable",
         async_set_variable_service
     )
 
-
+    hass.services.async_register(
+        DOMAIN,
+        "set_dos_rate",
+        async_set_dos_rate_service
+    )
 
     return True
 
 
 def set_output(hass, service, coordinator):
     did = service.data.get("did").strip()
     setting = service.data.get("setting").strip()
-    status = coordinator.apex.toggle_output(did, setting)
+    coordinator.apex.toggle_output(did, setting)
+
 
 def set_variable(hass, service, coordinator):
     did = service.data.get("did").strip()
@@ -103,6 +108,15 @@ def set_variable(hass, service, coordinator):
         raise HomeAssistantError(status["error"])
 
 
+def set_dos_rate(hass, service, coordinator):
+    did = service.data.get("did").strip()
+    profile_id = int(service.data.get("profile_id"))
+    rate = float(service.data.get("rate"))
+    status = coordinator.apex.set_dos_rate(did, profile_id, rate)
+    if status["error"] != "":
+        raise HomeAssistantError(status["error"])
+
+
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
     """Unload a config entry."""
     unload_ok = all(
@@ -147,8 +161,8 @@ async def _async_update_data(self):
                 data["config"] = await self._hass.async_add_executor_job(
                     self.apex.config  # Fetch new status
                 )
-                #_LOGGER.debug("Refreshing Now")
-                #_LOGGER.debug(data)
+                # _LOGGER.debug("Refreshing Now")
+                # _LOGGER.debug(data)
 
                 return data
         except Exception as ex:
@@ -164,14 +178,13 @@ class ApexEntity(CoordinatorEntity):
     """Defines a base Apex entity."""
 
     def __init__(
-        self, *, device_id: str, name: str, coordinator: ApexDataUpdateCoordinator
+            self, *, device_id: str, name: str, coordinator: ApexDataUpdateCoordinator
     ):
         """Initialize the entity."""
         super().__init__(coordinator)
         self._device_id = device_id
         self._name = name
 
-
     async def async_added_to_hass(self) -> None:
         """When entity is added to hass."""
         await super().async_added_to_hass()
@@ -202,5 +215,3 @@ def device_info(self):
             "manufacturer": MANUFACTURER,
             "test": "TEST"
         }
-
-