adjustments

This commit is contained in:
Jonh Sady 2025-02-17 18:44:16 -03:00
parent d3615af43c
commit b4cb7a76cf
6 changed files with 102 additions and 92 deletions

View File

@ -2,9 +2,7 @@ import logging
from homeassistant.core import HomeAssistant
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from .const import DOMAIN, API_URL
from homeassistant.helpers.aiohttp_client import async_get_clientsession
import aiohttp
from .const import DOMAIN
from redgtech_api import RedgtechAPI
_LOGGER = logging.getLogger(__name__)
@ -14,8 +12,8 @@ PLATFORMS: list[Platform] = [Platform.SWITCH]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Redgtech from a config entry."""
_LOGGER.debug("Setting up Redgtech entry: %s", entry.entry_id)
hass.data.setdefault(DOMAIN, {})
hass.data[DOMAIN][entry.entry_id] = {
entry.runtime_data = {
"config": entry.data,
"entities": []
}
@ -25,45 +23,40 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
_LOGGER.error("No access token found in config entry")
return False
session = async_get_clientsession(hass)
api = RedgtechAPI(access_token)
try:
async with session.get(f'{API_URL}/home_assistant?access_token={access_token}', timeout=10) as response:
response.raise_for_status()
data = await response.json()
_LOGGER.debug("Received data from API: %s", data)
data = await api.get_data()
_LOGGER.debug("Received data from API: %s", data)
entities = []
for item in data.get("boards", []):
entity_id = item.get('endpointId', '')
entity_name = item.get("name", f"Entity {entity_id}")
entity_value = item.get("value", False)
entity_state = "on" if entity_value else "off"
_LOGGER.debug("Processing entity: id=%s, name=%s, value=%s, state=%s", entity_id, entity_name, entity_value, entity_state)
entities = []
for item in data.get("boards", []):
entity_id = item.get('endpointId', '')
entity_name = item.get("name", f"Entity {entity_id}")
entity_value = item.get("value", False)
entity_state = "on" if entity_value else "off"
_LOGGER.debug("Processing entity: id=%s, name=%s, value=%s, state=%s", entity_id, entity_name, entity_value, entity_state)
entities.append({
"id": entity_id,
"name": entity_name,
"state": entity_state,
"type": 'switch'
})
entities.append({
"id": entity_id,
"name": entity_name,
"state": entity_state,
"type": 'switch'
})
hass.data[DOMAIN][entry.entry_id]["entities"] = entities
entry.runtime_data["entities"] = entities
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
_LOGGER.debug("Successfully set up Redgtech entry: %s", entry.entry_id)
return True
except aiohttp.ClientResponseError as e:
_LOGGER.error("HTTP error while setting up Redgtech entry: %s - Status: %s", e.message, e.status)
return False
except aiohttp.ClientError as e:
_LOGGER.error("Client error while setting up Redgtech entry: %s", e)
return False
except Exception as e:
_LOGGER.exception("Unexpected error setting up Redgtech entry: %s", entry.entry_id)
_LOGGER.error("Error setting up Redgtech entry: %s", e)
return False
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
_LOGGER.debug("Unloading Redgtech entry: %s", entry.entry_id)
# Fechar a sessão HTTP
api = RedgtechAPI(entry.data.get("access_token"))
await api.close()
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

View File

@ -1,8 +1,8 @@
from homeassistant import config_entries
import voluptuous as vol
import aiohttp
import logging
from .const import DOMAIN, API_URL
from .const import DOMAIN
from redgtech_api import RedgtechAPI
_LOGGER = logging.getLogger(__name__)
@ -19,30 +19,21 @@ class RedgtechConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
email = user_input.get("email")
password = user_input.get("password")
api = RedgtechAPI()
try:
async with aiohttp.ClientSession() as session:
async with session.post(
f'{API_URL}/home_assistant/login',
json={'email': email, 'password': password}
) as response:
if response.status == 200:
data = await response.json()
access_token = data.get("data", {}).get("access_token")
if access_token:
_LOGGER.info("Login successful")
access_token = await api.login(email, password)
if access_token:
_LOGGER.info("Login successful")
return self.async_create_entry(
title="Redgtech",
data={"access_token": access_token}
)
else:
_LOGGER.error("Login failed: No access token received")
errors["base"] = "invalid_auth"
else:
_LOGGER.error("Login failed: Invalid credentials")
errors["base"] = "invalid_auth"
except aiohttp.ClientError as e:
_LOGGER.error("Login failed: Cannot connect to server: %s", e)
return self.async_create_entry(
title="Redgtech",
data={"access_token": access_token}
)
else:
_LOGGER.error("Login failed: No access token received")
errors["base"] = "invalid_auth"
except Exception as e:
_LOGGER.error("Login failed: %s", e)
errors["base"] = "cannot_connect"
return self.async_show_form(

View File

@ -8,5 +8,5 @@
"integration_type": "service",
"config_flow": true,
"quality_scale": "silver",
"requirements": ["redgtech-api==0.1.9"]
"requirements": ["redgtech-api==0.1.14"]
}

View File

@ -1,12 +0,0 @@
login:
description: "Log in to the Redgtech service"
fields:
email:
description: "The email address for the Redgtech account"
example: "user@example.com"
password:
description: "The password for the Redgtech account"
example: "your_password"
logout:
description: "Log out from the Redgtech service"

View File

@ -1,38 +1,63 @@
import logging
from datetime import timedelta
from homeassistant.components.switch import SwitchEntity
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from redgtech_api import RedgtechAPI
from .const import DOMAIN, API_URL
from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass, config_entry, async_add_entities):
"""Set up the switch platform."""
access_token = config_entry.data.get("access_token")
if access_token:
api = RedgtechAPI(access_token)
try:
data = await api.get_data()
entities = []
for item in data.get("boards", []):
categories = item.get("displayCategories", "")
if "SWITCH" in categories:
entities.append(RedgtechSwitch(item, api))
async_add_entities(entities)
except Exception as e:
_LOGGER.error("Error fetching data from API: %s", e)
else:
if not access_token:
_LOGGER.error("No access token available")
return
api = RedgtechAPI(access_token)
coordinator = RedgtechDataUpdateCoordinator(hass, api)
await coordinator.async_config_entry_first_refresh()
entities = []
if coordinator.data:
for item in coordinator.data.get("boards", []):
categories = item.get("displayCategories", "")
if "SWITCH" in categories:
entities.append(RedgtechSwitch(coordinator, item, api))
async_add_entities(entities)
class RedgtechDataUpdateCoordinator(DataUpdateCoordinator):
"""Class to manage fetching data from the API."""
def __init__(self, hass, api):
"""Initialize."""
self.api = api
super().__init__(
hass,
_LOGGER,
name=DOMAIN,
update_interval=timedelta(seconds=1),
)
async def _async_update_data(self):
"""Fetch data from API."""
try:
_LOGGER.debug("Fetching data from Redgtech API")
return await self.api.get_data()
except Exception as e:
raise UpdateFailed(f"Error fetching data: {e}")
class RedgtechSwitch(SwitchEntity):
"""Representation of a Redgtech switch."""
def __init__(self, data, api):
def __init__(self, coordinator, data, api):
"""Initialize the switch."""
self.coordinator = coordinator
self.api = api
self._state = data.get("value", False)
self._name = data.get("friendlyName")
self._endpoint_id = data.get("endpointId")
self._description = data.get("description")
self._manufacturer = data.get("manufacturerName")
self._api = api
@property
def name(self):
@ -53,19 +78,32 @@ class RedgtechSwitch(SwitchEntity):
await self._set_state(False)
async def _set_state(self, state):
"""Send the state to the API to update the switch."""
success = await self._api.set_switch_state(self._endpoint_id, state)
"""Send the state to the API and update immediately."""
_LOGGER.debug("Setting state of %s to %s", self._name, state)
success = await self.api.set_switch_state(self._endpoint_id, state)
if success:
self._state = state
self.async_write_ha_state()
_LOGGER.debug("State of %s set to %s", self._name, state)
await self.coordinator.async_request_refresh()
else:
_LOGGER.error("Failed to set state for %s", self._name)
async def async_update(self):
"""Fetch new state data for the switch."""
_LOGGER.debug("Updating switch state: %s", self._name)
await self.coordinator.async_request_refresh()
data = self.coordinator.data
if data:
for item in data.get("boards", []):
if item.get("endpointId") == self._endpoint_id:
self._state = item.get("value", False)
self.async_write_ha_state()
@property
def extra_state_attributes(self):
"""Return the state attributes."""
return {
"endpoint_id": self._endpoint_id,
"description": self._description,
"manufacturer": self._manufacturer,
}

2
requirements_all.txt generated
View File

@ -2600,7 +2600,7 @@ rapt-ble==0.1.2
raspyrfm-client==1.2.8
# homeassistant.components.redgtech
redgtech-api==0.1.2
redgtech-api==0.1.14
# homeassistant.components.refoss
refoss-ha==1.2.5