mirror of
				https://github.com/home-assistant/core.git
				synced 2025-10-31 06:29:31 +00:00 
			
		
		
		
	Compare commits
	
		
			2 Commits
		
	
	
		
			block_pyse
			...
			zjs-config
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | fe35fac8ee | ||
|   | 4bccc57b46 | 
							
								
								
									
										720
									
								
								.github/workflows/ci.yaml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										720
									
								
								.github/workflows/ci.yaml
									
									
									
									
										vendored
									
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										4
									
								
								.github/workflows/codeql.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								.github/workflows/codeql.yml
									
									
									
									
										vendored
									
									
								
							| @@ -24,11 +24,11 @@ jobs: | ||||
|         uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0 | ||||
|  | ||||
|       - name: Initialize CodeQL | ||||
|         uses: github/codeql-action/init@f443b600d91635bebf5b0d9ebc620189c0d6fba5 # v4.30.8 | ||||
|         uses: github/codeql-action/init@3599b3baa15b485a2e49ef411a7a4bb2452e7f93 # v3.30.5 | ||||
|         with: | ||||
|           languages: python | ||||
|  | ||||
|       - name: Perform CodeQL Analysis | ||||
|         uses: github/codeql-action/analyze@f443b600d91635bebf5b0d9ebc620189c0d6fba5 # v4.30.8 | ||||
|         uses: github/codeql-action/analyze@3599b3baa15b485a2e49ef411a7a4bb2452e7f93 # v3.30.5 | ||||
|         with: | ||||
|           category: "/language:python" | ||||
|   | ||||
							
								
								
									
										6
									
								
								.github/workflows/stale.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										6
									
								
								.github/workflows/stale.yml
									
									
									
									
										vendored
									
									
								
							| @@ -17,7 +17,7 @@ jobs: | ||||
|       # - No PRs marked as no-stale | ||||
|       # - No issues (-1) | ||||
|       - name: 60 days stale PRs policy | ||||
|         uses: actions/stale@5f858e3efba33a5ca4407a664cc011ad407f2008 # v10.1.0 | ||||
|         uses: actions/stale@3a9db7e6a41a89f618792c92c0e97cc736e1b13f # v10.0.0 | ||||
|         with: | ||||
|           repo-token: ${{ secrets.GITHUB_TOKEN }} | ||||
|           days-before-stale: 60 | ||||
| @@ -57,7 +57,7 @@ jobs: | ||||
|       # - No issues marked as no-stale or help-wanted | ||||
|       # - No PRs (-1) | ||||
|       - name: 90 days stale issues | ||||
|         uses: actions/stale@5f858e3efba33a5ca4407a664cc011ad407f2008 # v10.1.0 | ||||
|         uses: actions/stale@3a9db7e6a41a89f618792c92c0e97cc736e1b13f # v10.0.0 | ||||
|         with: | ||||
|           repo-token: ${{ steps.token.outputs.token }} | ||||
|           days-before-stale: 90 | ||||
| @@ -87,7 +87,7 @@ jobs: | ||||
|       # - No Issues marked as no-stale or help-wanted | ||||
|       # - No PRs (-1) | ||||
|       - name: Needs more information stale issues policy | ||||
|         uses: actions/stale@5f858e3efba33a5ca4407a664cc011ad407f2008 # v10.1.0 | ||||
|         uses: actions/stale@3a9db7e6a41a89f618792c92c0e97cc736e1b13f # v10.0.0 | ||||
|         with: | ||||
|           repo-token: ${{ steps.token.outputs.token }} | ||||
|           only-labels: "needs-more-information" | ||||
|   | ||||
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -79,6 +79,7 @@ junit.xml | ||||
| .project | ||||
| .pydevproject | ||||
|  | ||||
| .python-version | ||||
| .tool-versions | ||||
|  | ||||
| # emacs auto backups | ||||
|   | ||||
| @@ -1 +0,0 @@ | ||||
| 3.13 | ||||
| @@ -221,7 +221,6 @@ homeassistant.components.generic_thermostat.* | ||||
| homeassistant.components.geo_location.* | ||||
| homeassistant.components.geocaching.* | ||||
| homeassistant.components.gios.* | ||||
| homeassistant.components.github.* | ||||
| homeassistant.components.glances.* | ||||
| homeassistant.components.go2rtc.* | ||||
| homeassistant.components.goalzero.* | ||||
| @@ -556,7 +555,6 @@ homeassistant.components.vacuum.* | ||||
| homeassistant.components.vallox.* | ||||
| homeassistant.components.valve.* | ||||
| homeassistant.components.velbus.* | ||||
| homeassistant.components.vivotek.* | ||||
| homeassistant.components.vlc_telnet.* | ||||
| homeassistant.components.vodafone_station.* | ||||
| homeassistant.components.volvo.* | ||||
|   | ||||
							
								
								
									
										20
									
								
								CODEOWNERS
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										20
									
								
								CODEOWNERS
									
									
									
										generated
									
									
									
								
							| @@ -46,8 +46,6 @@ build.json @home-assistant/supervisor | ||||
| /tests/components/accuweather/ @bieniu | ||||
| /homeassistant/components/acmeda/ @atmurray | ||||
| /tests/components/acmeda/ @atmurray | ||||
| /homeassistant/components/actron_air/ @kclif9 @JagadishDhanamjayam | ||||
| /tests/components/actron_air/ @kclif9 @JagadishDhanamjayam | ||||
| /homeassistant/components/adax/ @danielhiversen @lazytarget | ||||
| /tests/components/adax/ @danielhiversen @lazytarget | ||||
| /homeassistant/components/adguard/ @frenck | ||||
| @@ -764,8 +762,8 @@ build.json @home-assistant/supervisor | ||||
| /homeassistant/components/intent/ @home-assistant/core @synesthesiam @arturpragacz | ||||
| /tests/components/intent/ @home-assistant/core @synesthesiam @arturpragacz | ||||
| /homeassistant/components/intesishome/ @jnimmo | ||||
| /homeassistant/components/iometer/ @jukrebs | ||||
| /tests/components/iometer/ @jukrebs | ||||
| /homeassistant/components/iometer/ @MaestroOnICe | ||||
| /tests/components/iometer/ @MaestroOnICe | ||||
| /homeassistant/components/ios/ @robbiet480 | ||||
| /tests/components/ios/ @robbiet480 | ||||
| /homeassistant/components/iotawatt/ @gtdiehl @jyavenard | ||||
| @@ -1067,8 +1065,6 @@ build.json @home-assistant/supervisor | ||||
| /homeassistant/components/nilu/ @hfurubotten | ||||
| /homeassistant/components/nina/ @DeerMaximum | ||||
| /tests/components/nina/ @DeerMaximum | ||||
| /homeassistant/components/nintendo_parental_controls/ @pantherale0 | ||||
| /tests/components/nintendo_parental_controls/ @pantherale0 | ||||
| /homeassistant/components/nissan_leaf/ @filcole | ||||
| /homeassistant/components/noaa_tides/ @jdelaney72 | ||||
| /homeassistant/components/nobo_hub/ @echoromeo @oyvindwe | ||||
| @@ -1137,8 +1133,6 @@ build.json @home-assistant/supervisor | ||||
| /tests/components/opengarage/ @danielhiversen | ||||
| /homeassistant/components/openhome/ @bazwilliams | ||||
| /tests/components/openhome/ @bazwilliams | ||||
| /homeassistant/components/openrgb/ @felipecrs | ||||
| /tests/components/openrgb/ @felipecrs | ||||
| /homeassistant/components/opensky/ @joostlek | ||||
| /tests/components/opensky/ @joostlek | ||||
| /homeassistant/components/opentherm_gw/ @mvn23 | ||||
| @@ -1202,6 +1196,8 @@ build.json @home-assistant/supervisor | ||||
| /tests/components/plex/ @jjlawren | ||||
| /homeassistant/components/plugwise/ @CoMPaTech @bouwew | ||||
| /tests/components/plugwise/ @CoMPaTech @bouwew | ||||
| /homeassistant/components/plum_lightpad/ @ColinHarrington @prystupa | ||||
| /tests/components/plum_lightpad/ @ColinHarrington @prystupa | ||||
| /homeassistant/components/point/ @fredrike | ||||
| /tests/components/point/ @fredrike | ||||
| /homeassistant/components/pooldose/ @lmaertin | ||||
| @@ -1417,8 +1413,8 @@ build.json @home-assistant/supervisor | ||||
| /tests/components/sfr_box/ @epenet | ||||
| /homeassistant/components/sftp_storage/ @maretodoric | ||||
| /tests/components/sftp_storage/ @maretodoric | ||||
| /homeassistant/components/sharkiq/ @JeffResc @funkybunch @TheOneOgre | ||||
| /tests/components/sharkiq/ @JeffResc @funkybunch @TheOneOgre | ||||
| /homeassistant/components/sharkiq/ @JeffResc @funkybunch | ||||
| /tests/components/sharkiq/ @JeffResc @funkybunch | ||||
| /homeassistant/components/shell_command/ @home-assistant/core | ||||
| /tests/components/shell_command/ @home-assistant/core | ||||
| /homeassistant/components/shelly/ @bieniu @thecode @chemelli74 @bdraco | ||||
| @@ -1483,8 +1479,8 @@ build.json @home-assistant/supervisor | ||||
| /tests/components/snoo/ @Lash-L | ||||
| /homeassistant/components/snooz/ @AustinBrunkhorst | ||||
| /tests/components/snooz/ @AustinBrunkhorst | ||||
| /homeassistant/components/solaredge/ @frenck @bdraco @tronikos | ||||
| /tests/components/solaredge/ @frenck @bdraco @tronikos | ||||
| /homeassistant/components/solaredge/ @frenck @bdraco | ||||
| /tests/components/solaredge/ @frenck @bdraco | ||||
| /homeassistant/components/solaredge_local/ @drobtravels @scheric | ||||
| /homeassistant/components/solarlog/ @Ernst79 @dontinelli | ||||
| /tests/components/solarlog/ @Ernst79 @dontinelli | ||||
|   | ||||
| @@ -34,11 +34,9 @@ WORKDIR /usr/src | ||||
|  | ||||
| COPY --from=ghcr.io/astral-sh/uv:latest /uv /usr/local/bin/uv | ||||
|  | ||||
| RUN uv python install 3.13.2 | ||||
|  | ||||
| USER vscode | ||||
|  | ||||
| COPY .python-version ./ | ||||
| RUN uv python install | ||||
|  | ||||
| ENV VIRTUAL_ENV="/home/vscode/.local/ha-venv" | ||||
| RUN uv venv $VIRTUAL_ENV | ||||
| ENV PATH="$VIRTUAL_ENV/bin:$PATH" | ||||
|   | ||||
| @@ -635,6 +635,14 @@ async def async_enable_logging( | ||||
|         err_log_path = os.path.abspath(log_file) | ||||
|  | ||||
|     if err_log_path: | ||||
|         err_path_exists = os.path.isfile(err_log_path) | ||||
|         err_dir = os.path.dirname(err_log_path) | ||||
|  | ||||
|         # Check if we can write to the error log if it exists or that | ||||
|         # we can create files in the containing directory if not. | ||||
|         if (err_path_exists and os.access(err_log_path, os.W_OK)) or ( | ||||
|             not err_path_exists and os.access(err_dir, os.W_OK) | ||||
|         ): | ||||
|             err_handler = await hass.async_add_executor_job( | ||||
|                 _create_log_file, err_log_path, log_rotate_days | ||||
|             ) | ||||
| @@ -644,6 +652,8 @@ async def async_enable_logging( | ||||
|  | ||||
|             # Save the log file location for access by other components. | ||||
|             hass.data[DATA_LOGGING] = err_log_path | ||||
|         else: | ||||
|             _LOGGER.error("Unable to set up error log %s (access denied)", err_log_path) | ||||
|  | ||||
|     async_activate_log_queue_handler(hass) | ||||
|  | ||||
|   | ||||
							
								
								
									
										5
									
								
								homeassistant/brands/ibm.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								homeassistant/brands/ibm.json
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,5 @@ | ||||
| { | ||||
|   "domain": "ibm", | ||||
|   "name": "IBM", | ||||
|   "integrations": ["watson_iot", "watson_tts"] | ||||
| } | ||||
| @@ -12,13 +12,11 @@ from homeassistant.components.bluetooth import async_get_scanner | ||||
| from homeassistant.config_entries import ConfigEntry | ||||
| from homeassistant.const import CONF_ADDRESS | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.debounce import Debouncer | ||||
| from homeassistant.helpers.update_coordinator import DataUpdateCoordinator | ||||
|  | ||||
| from .const import CONF_IS_NEW_STYLE_SCALE | ||||
|  | ||||
| SCAN_INTERVAL = timedelta(seconds=15) | ||||
| UPDATE_DEBOUNCE_TIME = 0.2 | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
| @@ -40,19 +38,11 @@ class AcaiaCoordinator(DataUpdateCoordinator[None]): | ||||
|             config_entry=entry, | ||||
|         ) | ||||
|  | ||||
|         debouncer = Debouncer( | ||||
|             hass=hass, | ||||
|             logger=_LOGGER, | ||||
|             cooldown=UPDATE_DEBOUNCE_TIME, | ||||
|             immediate=True, | ||||
|             function=self.async_update_listeners, | ||||
|         ) | ||||
|  | ||||
|         self._scale = AcaiaScale( | ||||
|             address_or_ble_device=entry.data[CONF_ADDRESS], | ||||
|             name=entry.title, | ||||
|             is_new_style_scale=entry.data[CONF_IS_NEW_STYLE_SCALE], | ||||
|             notify_callback=debouncer.async_schedule_call, | ||||
|             notify_callback=self.async_update_listeners, | ||||
|             scanner=async_get_scanner(hass), | ||||
|         ) | ||||
|  | ||||
|   | ||||
| @@ -71,4 +71,4 @@ POLLEN_CATEGORY_MAP = { | ||||
| } | ||||
| UPDATE_INTERVAL_OBSERVATION = timedelta(minutes=10) | ||||
| UPDATE_INTERVAL_DAILY_FORECAST = timedelta(hours=6) | ||||
| UPDATE_INTERVAL_HOURLY_FORECAST = timedelta(minutes=30) | ||||
| UPDATE_INTERVAL_HOURLY_FORECAST = timedelta(hours=30) | ||||
|   | ||||
| @@ -1,9 +1,6 @@ | ||||
| { | ||||
|   "entity": { | ||||
|     "sensor": { | ||||
|       "air_quality": { | ||||
|         "default": "mdi:air-filter" | ||||
|       }, | ||||
|       "cloud_ceiling": { | ||||
|         "default": "mdi:weather-fog" | ||||
|       }, | ||||
| @@ -37,6 +34,9 @@ | ||||
|       "thunderstorm_probability_night": { | ||||
|         "default": "mdi:weather-lightning" | ||||
|       }, | ||||
|       "translation_key": { | ||||
|         "default": "mdi:air-filter" | ||||
|       }, | ||||
|       "tree_pollen": { | ||||
|         "default": "mdi:tree-outline" | ||||
|       }, | ||||
|   | ||||
| @@ -1,57 +0,0 @@ | ||||
| """The Actron Air integration.""" | ||||
|  | ||||
| from actron_neo_api import ( | ||||
|     ActronAirNeoACSystem, | ||||
|     ActronNeoAPI, | ||||
|     ActronNeoAPIError, | ||||
|     ActronNeoAuthError, | ||||
| ) | ||||
|  | ||||
| from homeassistant.const import CONF_API_TOKEN, Platform | ||||
| from homeassistant.core import HomeAssistant | ||||
|  | ||||
| from .const import _LOGGER | ||||
| from .coordinator import ( | ||||
|     ActronAirConfigEntry, | ||||
|     ActronAirRuntimeData, | ||||
|     ActronAirSystemCoordinator, | ||||
| ) | ||||
|  | ||||
| PLATFORM = [Platform.CLIMATE] | ||||
|  | ||||
|  | ||||
| async def async_setup_entry(hass: HomeAssistant, entry: ActronAirConfigEntry) -> bool: | ||||
|     """Set up Actron Air integration from a config entry.""" | ||||
|  | ||||
|     api = ActronNeoAPI(refresh_token=entry.data[CONF_API_TOKEN]) | ||||
|     systems: list[ActronAirNeoACSystem] = [] | ||||
|  | ||||
|     try: | ||||
|         systems = await api.get_ac_systems() | ||||
|         await api.update_status() | ||||
|     except ActronNeoAuthError: | ||||
|         _LOGGER.error("Authentication error while setting up Actron Air integration") | ||||
|         raise | ||||
|     except ActronNeoAPIError as err: | ||||
|         _LOGGER.error("API error while setting up Actron Air integration: %s", err) | ||||
|         raise | ||||
|  | ||||
|     system_coordinators: dict[str, ActronAirSystemCoordinator] = {} | ||||
|     for system in systems: | ||||
|         coordinator = ActronAirSystemCoordinator(hass, entry, api, system) | ||||
|         _LOGGER.debug("Setting up coordinator for system: %s", system["serial"]) | ||||
|         await coordinator.async_config_entry_first_refresh() | ||||
|         system_coordinators[system["serial"]] = coordinator | ||||
|  | ||||
|     entry.runtime_data = ActronAirRuntimeData( | ||||
|         api=api, | ||||
|         system_coordinators=system_coordinators, | ||||
|     ) | ||||
|  | ||||
|     await hass.config_entries.async_forward_entry_setups(entry, PLATFORM) | ||||
|     return True | ||||
|  | ||||
|  | ||||
| async def async_unload_entry(hass: HomeAssistant, entry: ActronAirConfigEntry) -> bool: | ||||
|     """Unload a config entry.""" | ||||
|     return await hass.config_entries.async_unload_platforms(entry, PLATFORM) | ||||
| @@ -1,259 +0,0 @@ | ||||
| """Climate platform for Actron Air integration.""" | ||||
|  | ||||
| from typing import Any | ||||
|  | ||||
| from actron_neo_api import ActronAirNeoStatus, ActronAirNeoZone | ||||
|  | ||||
| from homeassistant.components.climate import ( | ||||
|     FAN_AUTO, | ||||
|     FAN_HIGH, | ||||
|     FAN_LOW, | ||||
|     FAN_MEDIUM, | ||||
|     ClimateEntity, | ||||
|     ClimateEntityFeature, | ||||
|     HVACMode, | ||||
| ) | ||||
| from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.device_registry import DeviceInfo | ||||
| from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
| from homeassistant.helpers.update_coordinator import CoordinatorEntity | ||||
|  | ||||
| from .const import DOMAIN | ||||
| from .coordinator import ActronAirConfigEntry, ActronAirSystemCoordinator | ||||
|  | ||||
| PARALLEL_UPDATES = 0 | ||||
|  | ||||
| FAN_MODE_MAPPING_ACTRONAIR_TO_HA = { | ||||
|     "AUTO": FAN_AUTO, | ||||
|     "LOW": FAN_LOW, | ||||
|     "MED": FAN_MEDIUM, | ||||
|     "HIGH": FAN_HIGH, | ||||
| } | ||||
| FAN_MODE_MAPPING_HA_TO_ACTRONAIR = { | ||||
|     v: k for k, v in FAN_MODE_MAPPING_ACTRONAIR_TO_HA.items() | ||||
| } | ||||
| HVAC_MODE_MAPPING_ACTRONAIR_TO_HA = { | ||||
|     "COOL": HVACMode.COOL, | ||||
|     "HEAT": HVACMode.HEAT, | ||||
|     "FAN": HVACMode.FAN_ONLY, | ||||
|     "AUTO": HVACMode.AUTO, | ||||
|     "OFF": HVACMode.OFF, | ||||
| } | ||||
| HVAC_MODE_MAPPING_HA_TO_ACTRONAIR = { | ||||
|     v: k for k, v in HVAC_MODE_MAPPING_ACTRONAIR_TO_HA.items() | ||||
| } | ||||
|  | ||||
|  | ||||
| async def async_setup_entry( | ||||
|     hass: HomeAssistant, | ||||
|     entry: ActronAirConfigEntry, | ||||
|     async_add_entities: AddConfigEntryEntitiesCallback, | ||||
| ) -> None: | ||||
|     """Set up Actron Air climate entities.""" | ||||
|     system_coordinators = entry.runtime_data.system_coordinators | ||||
|     entities: list[ClimateEntity] = [] | ||||
|  | ||||
|     for coordinator in system_coordinators.values(): | ||||
|         status = coordinator.data | ||||
|         name = status.ac_system.system_name | ||||
|         entities.append(ActronSystemClimate(coordinator, name)) | ||||
|  | ||||
|         entities.extend( | ||||
|             ActronZoneClimate(coordinator, zone) | ||||
|             for zone in status.remote_zone_info | ||||
|             if zone.exists | ||||
|         ) | ||||
|  | ||||
|     async_add_entities(entities) | ||||
|  | ||||
|  | ||||
| class BaseClimateEntity(CoordinatorEntity[ActronAirSystemCoordinator], ClimateEntity): | ||||
|     """Base class for Actron Air climate entities.""" | ||||
|  | ||||
|     _attr_has_entity_name = True | ||||
|     _attr_temperature_unit = UnitOfTemperature.CELSIUS | ||||
|     _attr_supported_features = ( | ||||
|         ClimateEntityFeature.TARGET_TEMPERATURE | ||||
|         | ClimateEntityFeature.FAN_MODE | ||||
|         | ClimateEntityFeature.TURN_ON | ||||
|         | ClimateEntityFeature.TURN_OFF | ||||
|     ) | ||||
|     _attr_name = None | ||||
|     _attr_fan_modes = list(FAN_MODE_MAPPING_ACTRONAIR_TO_HA.values()) | ||||
|     _attr_hvac_modes = list(HVAC_MODE_MAPPING_ACTRONAIR_TO_HA.values()) | ||||
|  | ||||
|     def __init__( | ||||
|         self, | ||||
|         coordinator: ActronAirSystemCoordinator, | ||||
|         name: str, | ||||
|     ) -> None: | ||||
|         """Initialize an Actron Air unit.""" | ||||
|         super().__init__(coordinator) | ||||
|         self._serial_number = coordinator.serial_number | ||||
|  | ||||
|  | ||||
| class ActronSystemClimate(BaseClimateEntity): | ||||
|     """Representation of the Actron Air system.""" | ||||
|  | ||||
|     _attr_supported_features = ( | ||||
|         ClimateEntityFeature.TARGET_TEMPERATURE | ||||
|         | ClimateEntityFeature.FAN_MODE | ||||
|         | ClimateEntityFeature.TURN_ON | ||||
|         | ClimateEntityFeature.TURN_OFF | ||||
|     ) | ||||
|  | ||||
|     def __init__( | ||||
|         self, | ||||
|         coordinator: ActronAirSystemCoordinator, | ||||
|         name: str, | ||||
|     ) -> None: | ||||
|         """Initialize an Actron Air unit.""" | ||||
|         super().__init__(coordinator, name) | ||||
|         serial_number = coordinator.serial_number | ||||
|         self._attr_unique_id = serial_number | ||||
|         self._attr_device_info = DeviceInfo( | ||||
|             identifiers={(DOMAIN, serial_number)}, | ||||
|             name=self._status.ac_system.system_name, | ||||
|             manufacturer="Actron Air", | ||||
|             model_id=self._status.ac_system.master_wc_model, | ||||
|             sw_version=self._status.ac_system.master_wc_firmware_version, | ||||
|             serial_number=serial_number, | ||||
|         ) | ||||
|  | ||||
|     @property | ||||
|     def min_temp(self) -> float: | ||||
|         """Return the minimum temperature that can be set.""" | ||||
|         return self._status.min_temp | ||||
|  | ||||
|     @property | ||||
|     def max_temp(self) -> float: | ||||
|         """Return the maximum temperature that can be set.""" | ||||
|         return self._status.max_temp | ||||
|  | ||||
|     @property | ||||
|     def _status(self) -> ActronAirNeoStatus: | ||||
|         """Get the current status from the coordinator.""" | ||||
|         return self.coordinator.data | ||||
|  | ||||
|     @property | ||||
|     def hvac_mode(self) -> HVACMode | None: | ||||
|         """Return the current HVAC mode.""" | ||||
|         if not self._status.user_aircon_settings.is_on: | ||||
|             return HVACMode.OFF | ||||
|  | ||||
|         mode = self._status.user_aircon_settings.mode | ||||
|         return HVAC_MODE_MAPPING_ACTRONAIR_TO_HA.get(mode) | ||||
|  | ||||
|     @property | ||||
|     def fan_mode(self) -> str | None: | ||||
|         """Return the current fan mode.""" | ||||
|         fan_mode = self._status.user_aircon_settings.fan_mode | ||||
|         return FAN_MODE_MAPPING_ACTRONAIR_TO_HA.get(fan_mode) | ||||
|  | ||||
|     @property | ||||
|     def current_humidity(self) -> float: | ||||
|         """Return the current humidity.""" | ||||
|         return self._status.master_info.live_humidity_pc | ||||
|  | ||||
|     @property | ||||
|     def current_temperature(self) -> float: | ||||
|         """Return the current temperature.""" | ||||
|         return self._status.master_info.live_temp_c | ||||
|  | ||||
|     @property | ||||
|     def target_temperature(self) -> float: | ||||
|         """Return the target temperature.""" | ||||
|         return self._status.user_aircon_settings.temperature_setpoint_cool_c | ||||
|  | ||||
|     async def async_set_fan_mode(self, fan_mode: str) -> None: | ||||
|         """Set a new fan mode.""" | ||||
|         api_fan_mode = FAN_MODE_MAPPING_HA_TO_ACTRONAIR.get(fan_mode.lower()) | ||||
|         await self._status.user_aircon_settings.set_fan_mode(api_fan_mode) | ||||
|  | ||||
|     async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None: | ||||
|         """Set the HVAC mode.""" | ||||
|         ac_mode = HVAC_MODE_MAPPING_HA_TO_ACTRONAIR.get(hvac_mode) | ||||
|         await self._status.ac_system.set_system_mode(ac_mode) | ||||
|  | ||||
|     async def async_set_temperature(self, **kwargs: Any) -> None: | ||||
|         """Set the temperature.""" | ||||
|         temp = kwargs.get(ATTR_TEMPERATURE) | ||||
|         await self._status.user_aircon_settings.set_temperature(temperature=temp) | ||||
|  | ||||
|  | ||||
| class ActronZoneClimate(BaseClimateEntity): | ||||
|     """Representation of a zone within the Actron Air system.""" | ||||
|  | ||||
|     _attr_supported_features = ( | ||||
|         ClimateEntityFeature.TARGET_TEMPERATURE | ||||
|         | ClimateEntityFeature.TURN_ON | ||||
|         | ClimateEntityFeature.TURN_OFF | ||||
|     ) | ||||
|  | ||||
|     def __init__( | ||||
|         self, | ||||
|         coordinator: ActronAirSystemCoordinator, | ||||
|         zone: ActronAirNeoZone, | ||||
|     ) -> None: | ||||
|         """Initialize an Actron Air unit.""" | ||||
|         super().__init__(coordinator, zone.title) | ||||
|         serial_number = coordinator.serial_number | ||||
|         self._zone_id: int = zone.zone_id | ||||
|         self._attr_unique_id: str = f"{serial_number}_zone_{zone.zone_id}" | ||||
|         self._attr_device_info: DeviceInfo = DeviceInfo( | ||||
|             identifiers={(DOMAIN, self._attr_unique_id)}, | ||||
|             name=zone.title, | ||||
|             manufacturer="Actron Air", | ||||
|             model="Zone", | ||||
|             suggested_area=zone.title, | ||||
|             via_device=(DOMAIN, serial_number), | ||||
|         ) | ||||
|  | ||||
|     @property | ||||
|     def min_temp(self) -> float: | ||||
|         """Return the minimum temperature that can be set.""" | ||||
|         return self._zone.min_temp | ||||
|  | ||||
|     @property | ||||
|     def max_temp(self) -> float: | ||||
|         """Return the maximum temperature that can be set.""" | ||||
|         return self._zone.max_temp | ||||
|  | ||||
|     @property | ||||
|     def _zone(self) -> ActronAirNeoZone: | ||||
|         """Get the current zone data from the coordinator.""" | ||||
|         status = self.coordinator.data | ||||
|         return status.zones[self._zone_id] | ||||
|  | ||||
|     @property | ||||
|     def hvac_mode(self) -> HVACMode | None: | ||||
|         """Return the current HVAC mode.""" | ||||
|         if self._zone.is_active: | ||||
|             mode = self._zone.hvac_mode | ||||
|             return HVAC_MODE_MAPPING_ACTRONAIR_TO_HA.get(mode) | ||||
|         return HVACMode.OFF | ||||
|  | ||||
|     @property | ||||
|     def current_humidity(self) -> float | None: | ||||
|         """Return the current humidity.""" | ||||
|         return self._zone.humidity | ||||
|  | ||||
|     @property | ||||
|     def current_temperature(self) -> float | None: | ||||
|         """Return the current temperature.""" | ||||
|         return self._zone.live_temp_c | ||||
|  | ||||
|     @property | ||||
|     def target_temperature(self) -> float | None: | ||||
|         """Return the target temperature.""" | ||||
|         return self._zone.temperature_setpoint_cool_c | ||||
|  | ||||
|     async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None: | ||||
|         """Set the HVAC mode.""" | ||||
|         is_enabled = hvac_mode != HVACMode.OFF | ||||
|         await self._zone.enable(is_enabled) | ||||
|  | ||||
|     async def async_set_temperature(self, **kwargs: Any) -> None: | ||||
|         """Set the temperature.""" | ||||
|         await self._zone.set_temperature(temperature=kwargs["temperature"]) | ||||
| @@ -1,132 +0,0 @@ | ||||
| """Setup config flow for Actron Air integration.""" | ||||
|  | ||||
| import asyncio | ||||
| from typing import Any | ||||
|  | ||||
| from actron_neo_api import ActronNeoAPI, ActronNeoAuthError | ||||
|  | ||||
| from homeassistant.config_entries import ConfigFlow, ConfigFlowResult | ||||
| from homeassistant.const import CONF_API_TOKEN | ||||
| from homeassistant.exceptions import HomeAssistantError | ||||
|  | ||||
| from .const import _LOGGER, DOMAIN | ||||
|  | ||||
|  | ||||
| class ActronAirConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|     """Handle a config flow for Actron Air.""" | ||||
|  | ||||
|     def __init__(self) -> None: | ||||
|         """Initialize the config flow.""" | ||||
|         self._api: ActronNeoAPI | None = None | ||||
|         self._device_code: str | None = None | ||||
|         self._user_code: str = "" | ||||
|         self._verification_uri: str = "" | ||||
|         self._expires_minutes: str = "30" | ||||
|         self.login_task: asyncio.Task | None = None | ||||
|  | ||||
|     async def async_step_user( | ||||
|         self, user_input: dict[str, Any] | None = None | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Handle the initial step.""" | ||||
|         if self._api is None: | ||||
|             _LOGGER.debug("Initiating device authorization") | ||||
|             self._api = ActronNeoAPI() | ||||
|             try: | ||||
|                 device_code_response = await self._api.request_device_code() | ||||
|             except ActronNeoAuthError as err: | ||||
|                 _LOGGER.error("OAuth2 flow failed: %s", err) | ||||
|                 return self.async_abort(reason="oauth2_error") | ||||
|  | ||||
|             self._device_code = device_code_response["device_code"] | ||||
|             self._user_code = device_code_response["user_code"] | ||||
|             self._verification_uri = device_code_response["verification_uri_complete"] | ||||
|             self._expires_minutes = str(device_code_response["expires_in"] // 60) | ||||
|  | ||||
|         async def _wait_for_authorization() -> None: | ||||
|             """Wait for the user to authorize the device.""" | ||||
|             assert self._api is not None | ||||
|             assert self._device_code is not None | ||||
|             _LOGGER.debug("Waiting for device authorization") | ||||
|             try: | ||||
|                 await self._api.poll_for_token(self._device_code) | ||||
|                 _LOGGER.debug("Authorization successful") | ||||
|             except ActronNeoAuthError as ex: | ||||
|                 _LOGGER.exception("Error while waiting for device authorization") | ||||
|                 raise CannotConnect from ex | ||||
|  | ||||
|         _LOGGER.debug("Checking login task") | ||||
|         if self.login_task is None: | ||||
|             _LOGGER.debug("Creating task for device authorization") | ||||
|             self.login_task = self.hass.async_create_task(_wait_for_authorization()) | ||||
|  | ||||
|         if self.login_task.done(): | ||||
|             _LOGGER.debug("Login task is done, checking results") | ||||
|             if exception := self.login_task.exception(): | ||||
|                 if isinstance(exception, CannotConnect): | ||||
|                     return self.async_show_progress_done( | ||||
|                         next_step_id="connection_error" | ||||
|                     ) | ||||
|                 return self.async_show_progress_done(next_step_id="timeout") | ||||
|             return self.async_show_progress_done(next_step_id="finish_login") | ||||
|  | ||||
|         return self.async_show_progress( | ||||
|             step_id="user", | ||||
|             progress_action="wait_for_authorization", | ||||
|             description_placeholders={ | ||||
|                 "user_code": self._user_code, | ||||
|                 "verification_uri": self._verification_uri, | ||||
|                 "expires_minutes": self._expires_minutes, | ||||
|             }, | ||||
|             progress_task=self.login_task, | ||||
|         ) | ||||
|  | ||||
|     async def async_step_finish_login( | ||||
|         self, user_input: dict[str, Any] | None = None | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Handle the finalization of login.""" | ||||
|         _LOGGER.debug("Finalizing authorization") | ||||
|         assert self._api is not None | ||||
|  | ||||
|         try: | ||||
|             user_data = await self._api.get_user_info() | ||||
|         except ActronNeoAuthError as err: | ||||
|             _LOGGER.error("Error getting user info: %s", err) | ||||
|             return self.async_abort(reason="oauth2_error") | ||||
|  | ||||
|         unique_id = str(user_data["id"]) | ||||
|         await self.async_set_unique_id(unique_id) | ||||
|         self._abort_if_unique_id_configured() | ||||
|  | ||||
|         return self.async_create_entry( | ||||
|             title=user_data["email"], | ||||
|             data={CONF_API_TOKEN: self._api.refresh_token_value}, | ||||
|         ) | ||||
|  | ||||
|     async def async_step_timeout( | ||||
|         self, | ||||
|         user_input: dict[str, Any] | None = None, | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Handle issues that need transition await from progress step.""" | ||||
|         if user_input is None: | ||||
|             return self.async_show_form( | ||||
|                 step_id="timeout", | ||||
|             ) | ||||
|         del self.login_task | ||||
|         return await self.async_step_user() | ||||
|  | ||||
|     async def async_step_connection_error( | ||||
|         self, user_input: dict[str, Any] | None = None | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Handle connection error from progress step.""" | ||||
|         if user_input is None: | ||||
|             return self.async_show_form(step_id="connection_error") | ||||
|  | ||||
|         # Reset state and try again | ||||
|         self._api = None | ||||
|         self._device_code = None | ||||
|         self.login_task = None | ||||
|         return await self.async_step_user() | ||||
|  | ||||
|  | ||||
| class CannotConnect(HomeAssistantError): | ||||
|     """Error to indicate we cannot connect.""" | ||||
| @@ -1,6 +0,0 @@ | ||||
| """Constants used by Actron Air integration.""" | ||||
|  | ||||
| import logging | ||||
|  | ||||
| _LOGGER = logging.getLogger(__package__) | ||||
| DOMAIN = "actron_air" | ||||
| @@ -1,69 +0,0 @@ | ||||
| """Coordinator for Actron Air integration.""" | ||||
|  | ||||
| from __future__ import annotations | ||||
|  | ||||
| from dataclasses import dataclass | ||||
| from datetime import timedelta | ||||
|  | ||||
| from actron_neo_api import ActronAirNeoACSystem, ActronAirNeoStatus, ActronNeoAPI | ||||
|  | ||||
| from homeassistant.config_entries import ConfigEntry | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.update_coordinator import DataUpdateCoordinator | ||||
| from homeassistant.util import dt as dt_util | ||||
|  | ||||
| from .const import _LOGGER | ||||
|  | ||||
| STALE_DEVICE_TIMEOUT = timedelta(hours=24) | ||||
| ERROR_NO_SYSTEMS_FOUND = "no_systems_found" | ||||
| ERROR_UNKNOWN = "unknown_error" | ||||
|  | ||||
|  | ||||
| @dataclass | ||||
| class ActronAirRuntimeData: | ||||
|     """Runtime data for the Actron Air integration.""" | ||||
|  | ||||
|     api: ActronNeoAPI | ||||
|     system_coordinators: dict[str, ActronAirSystemCoordinator] | ||||
|  | ||||
|  | ||||
| type ActronAirConfigEntry = ConfigEntry[ActronAirRuntimeData] | ||||
|  | ||||
| AUTH_ERROR_THRESHOLD = 3 | ||||
| SCAN_INTERVAL = timedelta(seconds=30) | ||||
|  | ||||
|  | ||||
| class ActronAirSystemCoordinator(DataUpdateCoordinator[ActronAirNeoACSystem]): | ||||
|     """System coordinator for Actron Air integration.""" | ||||
|  | ||||
|     def __init__( | ||||
|         self, | ||||
|         hass: HomeAssistant, | ||||
|         entry: ActronAirConfigEntry, | ||||
|         api: ActronNeoAPI, | ||||
|         system: ActronAirNeoACSystem, | ||||
|     ) -> None: | ||||
|         """Initialize the coordinator.""" | ||||
|         super().__init__( | ||||
|             hass, | ||||
|             _LOGGER, | ||||
|             name="Actron Air Status", | ||||
|             update_interval=SCAN_INTERVAL, | ||||
|             config_entry=entry, | ||||
|         ) | ||||
|         self.system = system | ||||
|         self.serial_number = system["serial"] | ||||
|         self.api = api | ||||
|         self.status = self.api.state_manager.get_status(self.serial_number) | ||||
|         self.last_seen = dt_util.utcnow() | ||||
|  | ||||
|     async def _async_update_data(self) -> ActronAirNeoStatus: | ||||
|         """Fetch updates and merge incremental changes into the full state.""" | ||||
|         await self.api.update_status() | ||||
|         self.status = self.api.state_manager.get_status(self.serial_number) | ||||
|         self.last_seen = dt_util.utcnow() | ||||
|         return self.status | ||||
|  | ||||
|     def is_device_stale(self) -> bool: | ||||
|         """Check if a device is stale (not seen for a while).""" | ||||
|         return (dt_util.utcnow() - self.last_seen) > STALE_DEVICE_TIMEOUT | ||||
| @@ -1,16 +0,0 @@ | ||||
| { | ||||
|   "domain": "actron_air", | ||||
|   "name": "Actron Air", | ||||
|   "codeowners": ["@kclif9", "@JagadishDhanamjayam"], | ||||
|   "config_flow": true, | ||||
|   "dhcp": [ | ||||
|     { | ||||
|       "hostname": "neo-*", | ||||
|       "macaddress": "FC0FE7*" | ||||
|     } | ||||
|   ], | ||||
|   "documentation": "https://www.home-assistant.io/integrations/actron_air", | ||||
|   "iot_class": "cloud_polling", | ||||
|   "quality_scale": "bronze", | ||||
|   "requirements": ["actron-neo-api==0.1.84"] | ||||
| } | ||||
| @@ -1,78 +0,0 @@ | ||||
| rules: | ||||
|   # Bronze | ||||
|   action-setup: | ||||
|     status: exempt | ||||
|     comment: This integration does not have custom service actions. | ||||
|   appropriate-polling: done | ||||
|   brands: done | ||||
|   common-modules: done | ||||
|   config-flow-test-coverage: done | ||||
|   config-flow: done | ||||
|   dependency-transparency: done | ||||
|   docs-actions: | ||||
|     status: exempt | ||||
|     comment: This integration does not have custom service actions. | ||||
|   docs-high-level-description: done | ||||
|   docs-installation-instructions: done | ||||
|   docs-removal-instructions: done | ||||
|   entity-event-setup: | ||||
|     status: exempt | ||||
|     comment: This integration does not subscribe to external events. | ||||
|   entity-unique-id: done | ||||
|   has-entity-name: done | ||||
|   runtime-data: done | ||||
|   test-before-configure: done | ||||
|   test-before-setup: done | ||||
|   unique-config-entry: done | ||||
|  | ||||
|   # Silver | ||||
|   action-exceptions: todo | ||||
|   config-entry-unloading: done | ||||
|   docs-configuration-parameters: | ||||
|     status: exempt | ||||
|     comment: No options flow | ||||
|   docs-installation-parameters: done | ||||
|   entity-unavailable: done | ||||
|   integration-owner: done | ||||
|   log-when-unavailable: done | ||||
|   parallel-updates: done | ||||
|   reauthentication-flow: todo | ||||
|   test-coverage: todo | ||||
|  | ||||
|   # Gold | ||||
|   devices: done | ||||
|   diagnostics: todo | ||||
|   discovery-update-info: | ||||
|     status: exempt | ||||
|     comment: This integration uses DHCP discovery, however is cloud polling. Therefore there is no information to update. | ||||
|   discovery: done | ||||
|   docs-data-update: done | ||||
|   docs-examples: done | ||||
|   docs-known-limitations: done | ||||
|   docs-supported-devices: done | ||||
|   docs-supported-functions: done | ||||
|   docs-troubleshooting: done | ||||
|   docs-use-cases: done | ||||
|   dynamic-devices: todo | ||||
|   entity-category: | ||||
|     status: exempt | ||||
|     comment: This integration does not use entity categories. | ||||
|   entity-device-class: | ||||
|     status: exempt | ||||
|     comment: This integration does not use entity device classes. | ||||
|   entity-disabled-by-default: | ||||
|     status: exempt | ||||
|     comment: Not required for this integration at this stage. | ||||
|   entity-translations: todo | ||||
|   exception-translations: todo | ||||
|   icon-translations: todo | ||||
|   reconfiguration-flow: todo | ||||
|   repair-issues: | ||||
|     status: exempt | ||||
|     comment: This integration does not have any known issues that require repair. | ||||
|   stale-devices: todo | ||||
|  | ||||
|   # Platinum | ||||
|   async-dependency: done | ||||
|   inject-websession: todo | ||||
|   strict-typing: todo | ||||
| @@ -1,29 +0,0 @@ | ||||
| { | ||||
|   "config": { | ||||
|     "step": { | ||||
|       "user": { | ||||
|         "title": "Actron Air OAuth2 Authorization" | ||||
|       }, | ||||
|       "timeout": { | ||||
|         "title": "Authorization timeout", | ||||
|         "description": "The authorization process timed out. Please try again.", | ||||
|         "data": {} | ||||
|       }, | ||||
|       "connection_error": { | ||||
|         "title": "Connection error", | ||||
|         "description": "Failed to connect to Actron Air. Please check your internet connection and try again.", | ||||
|         "data": {} | ||||
|       } | ||||
|     }, | ||||
|     "progress": { | ||||
|       "wait_for_authorization": "To authenticate, open the following URL and login at Actron Air:\n{verification_uri}\nIf the code is not automatically copied, paste the following code to authorize the integration:\n\n```{user_code}```\n\n\nThe login attempt will time out after {expires_minutes} minutes." | ||||
|     }, | ||||
|     "error": { | ||||
|       "oauth2_error": "Failed to start OAuth2 flow. Please try again later." | ||||
|     }, | ||||
|     "abort": { | ||||
|       "oauth2_error": "Failed to start OAuth2 flow", | ||||
|       "already_configured": "[%key:common::config_flow::abort::already_configured_account%]" | ||||
|     } | ||||
|   } | ||||
| } | ||||
| @@ -71,14 +71,7 @@ class AemetConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             } | ||||
|         ) | ||||
|  | ||||
|         return self.async_show_form( | ||||
|             step_id="user", | ||||
|             data_schema=schema, | ||||
|             errors=errors, | ||||
|             description_placeholders={ | ||||
|                 "api_key_url": "https://opendata.aemet.es/centrodedescargas/altaUsuario" | ||||
|             }, | ||||
|         ) | ||||
|         return self.async_show_form(step_id="user", data_schema=schema, errors=errors) | ||||
|  | ||||
|     @staticmethod | ||||
|     @callback | ||||
|   | ||||
| @@ -14,7 +14,7 @@ | ||||
|           "longitude": "[%key:common::config_flow::data::longitude%]", | ||||
|           "name": "Name of the integration" | ||||
|         }, | ||||
|         "description": "To generate API key go to {api_key_url}" | ||||
|         "description": "To generate API key go to https://opendata.aemet.es/centrodedescargas/altaUsuario" | ||||
|       } | ||||
|     } | ||||
|   }, | ||||
|   | ||||
| @@ -1,9 +1,7 @@ | ||||
| """Airgradient Update platform.""" | ||||
|  | ||||
| from datetime import timedelta | ||||
| import logging | ||||
|  | ||||
| from airgradient import AirGradientConnectionError | ||||
| from propcache.api import cached_property | ||||
|  | ||||
| from homeassistant.components.update import UpdateDeviceClass, UpdateEntity | ||||
| @@ -15,7 +13,6 @@ from .entity import AirGradientEntity | ||||
|  | ||||
| PARALLEL_UPDATES = 1 | ||||
| SCAN_INTERVAL = timedelta(hours=1) | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
|  | ||||
| async def async_setup_entry( | ||||
| @@ -34,7 +31,6 @@ class AirGradientUpdate(AirGradientEntity, UpdateEntity): | ||||
|     """Representation of Airgradient Update.""" | ||||
|  | ||||
|     _attr_device_class = UpdateDeviceClass.FIRMWARE | ||||
|     _server_unreachable_logged = False | ||||
|  | ||||
|     def __init__(self, coordinator: AirGradientCoordinator) -> None: | ||||
|         """Initialize the entity.""" | ||||
| @@ -51,27 +47,10 @@ class AirGradientUpdate(AirGradientEntity, UpdateEntity): | ||||
|         """Return the installed version of the entity.""" | ||||
|         return self.coordinator.data.measures.firmware_version | ||||
|  | ||||
|     @property | ||||
|     def available(self) -> bool: | ||||
|         """Return if entity is available.""" | ||||
|         return super().available and self._attr_available | ||||
|  | ||||
|     async def async_update(self) -> None: | ||||
|         """Update the entity.""" | ||||
|         try: | ||||
|         self._attr_latest_version = ( | ||||
|             await self.coordinator.client.get_latest_firmware_version( | ||||
|                 self.coordinator.serial_number | ||||
|             ) | ||||
|         ) | ||||
|         except AirGradientConnectionError: | ||||
|             self._attr_latest_version = None | ||||
|             self._attr_available = False | ||||
|             if not self._server_unreachable_logged: | ||||
|                 _LOGGER.error( | ||||
|                     "Unable to connect to AirGradient server to check for updates" | ||||
|                 ) | ||||
|                 self._server_unreachable_logged = True | ||||
|         else: | ||||
|             self._server_unreachable_logged = False | ||||
|             self._attr_available = True | ||||
|   | ||||
| @@ -18,10 +18,6 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession | ||||
|  | ||||
| from .const import CONF_USE_NEAREST, DOMAIN, NO_AIRLY_SENSORS | ||||
|  | ||||
| DESCRIPTION_PLACEHOLDERS = { | ||||
|     "developer_registration_url": "https://developer.airly.eu/register", | ||||
| } | ||||
|  | ||||
|  | ||||
| class AirlyFlowHandler(ConfigFlow, domain=DOMAIN): | ||||
|     """Config flow for Airly.""" | ||||
| @@ -89,7 +85,6 @@ class AirlyFlowHandler(ConfigFlow, domain=DOMAIN): | ||||
|                 } | ||||
|             ), | ||||
|             errors=errors, | ||||
|             description_placeholders=DESCRIPTION_PLACEHOLDERS, | ||||
|         ) | ||||
|  | ||||
|  | ||||
|   | ||||
| @@ -2,7 +2,7 @@ | ||||
|   "config": { | ||||
|     "step": { | ||||
|       "user": { | ||||
|         "description": "To generate API key go to {developer_registration_url}", | ||||
|         "description": "To generate API key go to https://developer.airly.eu/register", | ||||
|         "data": { | ||||
|           "name": "[%key:common::config_flow::data::name%]", | ||||
|           "api_key": "[%key:common::config_flow::data::api_key%]", | ||||
|   | ||||
| @@ -2,8 +2,6 @@ | ||||
|  | ||||
| from __future__ import annotations | ||||
|  | ||||
| import logging | ||||
|  | ||||
| from airos.airos8 import AirOS8 | ||||
|  | ||||
| from homeassistant.const import ( | ||||
| @@ -14,11 +12,10 @@ from homeassistant.const import ( | ||||
|     CONF_VERIFY_SSL, | ||||
|     Platform, | ||||
| ) | ||||
| from homeassistant.core import HomeAssistant, callback | ||||
| from homeassistant.helpers import device_registry as dr, entity_registry as er | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.aiohttp_client import async_get_clientsession | ||||
|  | ||||
| from .const import DEFAULT_SSL, DEFAULT_VERIFY_SSL, DOMAIN, SECTION_ADVANCED_SETTINGS | ||||
| from .const import DEFAULT_SSL, DEFAULT_VERIFY_SSL, SECTION_ADVANCED_SETTINGS | ||||
| from .coordinator import AirOSConfigEntry, AirOSDataUpdateCoordinator | ||||
|  | ||||
| _PLATFORMS: list[Platform] = [ | ||||
| @@ -26,8 +23,6 @@ _PLATFORMS: list[Platform] = [ | ||||
|     Platform.SENSOR, | ||||
| ] | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
|  | ||||
| async def async_setup_entry(hass: HomeAssistant, entry: AirOSConfigEntry) -> bool: | ||||
|     """Set up Ubiquiti airOS from a config entry.""" | ||||
| @@ -59,13 +54,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: AirOSConfigEntry) -> boo | ||||
| async def async_migrate_entry(hass: HomeAssistant, entry: AirOSConfigEntry) -> bool: | ||||
|     """Migrate old config entry.""" | ||||
|  | ||||
|     if entry.version > 1: | ||||
|         # This means the user has downgraded from a future version | ||||
|     if entry.version > 2: | ||||
|         return False | ||||
|  | ||||
|     # 1.1 Migrate config_entry to add advanced ssl settings | ||||
|     if entry.version == 1 and entry.minor_version == 1: | ||||
|         new_minor_version = 2 | ||||
|         new_data = {**entry.data} | ||||
|         advanced_data = { | ||||
|             CONF_SSL: DEFAULT_SSL, | ||||
| @@ -76,52 +69,7 @@ async def async_migrate_entry(hass: HomeAssistant, entry: AirOSConfigEntry) -> b | ||||
|         hass.config_entries.async_update_entry( | ||||
|             entry, | ||||
|             data=new_data, | ||||
|             minor_version=new_minor_version, | ||||
|         ) | ||||
|  | ||||
|     # 2.1 Migrate binary_sensor entity unique_id from device_id to mac_address | ||||
|     #     Step 1 - migrate binary_sensor entity unique_id | ||||
|     #     Step 2 - migrate device entity identifier | ||||
|     if entry.version == 1: | ||||
|         new_version = 2 | ||||
|         new_minor_version = 1 | ||||
|  | ||||
|         mac_adress = dr.format_mac(entry.unique_id) | ||||
|  | ||||
|         device_registry = dr.async_get(hass) | ||||
|         if device_entry := device_registry.async_get_device( | ||||
|             connections={(dr.CONNECTION_NETWORK_MAC, mac_adress)} | ||||
|         ): | ||||
|             old_device_id = next( | ||||
|                 ( | ||||
|                     device_id | ||||
|                     for domain, device_id in device_entry.identifiers | ||||
|                     if domain == DOMAIN | ||||
|                 ), | ||||
|             ) | ||||
|  | ||||
|             @callback | ||||
|             def update_unique_id( | ||||
|                 entity_entry: er.RegistryEntry, | ||||
|             ) -> dict[str, str] | None: | ||||
|                 """Update unique id from device_id to mac address.""" | ||||
|                 if old_device_id and entity_entry.unique_id.startswith(old_device_id): | ||||
|                     suffix = entity_entry.unique_id.removeprefix(old_device_id) | ||||
|                     new_unique_id = f"{mac_adress}{suffix}" | ||||
|                     return {"new_unique_id": new_unique_id} | ||||
|                 return None | ||||
|  | ||||
|             await er.async_migrate_entries(hass, entry.entry_id, update_unique_id) | ||||
|  | ||||
|             new_identifiers = device_entry.identifiers.copy() | ||||
|             new_identifiers.discard((DOMAIN, old_device_id)) | ||||
|             new_identifiers.add((DOMAIN, mac_adress)) | ||||
|             device_registry.async_update_device( | ||||
|                 device_entry.id, new_identifiers=new_identifiers | ||||
|             ) | ||||
|  | ||||
|         hass.config_entries.async_update_entry( | ||||
|             entry, version=new_version, minor_version=new_minor_version | ||||
|             minor_version=2, | ||||
|         ) | ||||
|  | ||||
|     return True | ||||
|   | ||||
| @@ -98,7 +98,7 @@ class AirOSBinarySensor(AirOSEntity, BinarySensorEntity): | ||||
|         super().__init__(coordinator) | ||||
|  | ||||
|         self.entity_description = description | ||||
|         self._attr_unique_id = f"{coordinator.data.derived.mac}_{description.key}" | ||||
|         self._attr_unique_id = f"{coordinator.data.host.device_id}_{description.key}" | ||||
|  | ||||
|     @property | ||||
|     def is_on(self) -> bool: | ||||
|   | ||||
| @@ -15,12 +15,7 @@ from airos.exceptions import ( | ||||
| ) | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.config_entries import ( | ||||
|     SOURCE_REAUTH, | ||||
|     SOURCE_RECONFIGURE, | ||||
|     ConfigFlow, | ||||
|     ConfigFlowResult, | ||||
| ) | ||||
| from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlow, ConfigFlowResult | ||||
| from homeassistant.const import ( | ||||
|     CONF_HOST, | ||||
|     CONF_PASSWORD, | ||||
| @@ -62,8 +57,8 @@ STEP_USER_DATA_SCHEMA = vol.Schema( | ||||
| class AirOSConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|     """Handle a config flow for Ubiquiti airOS.""" | ||||
|  | ||||
|     VERSION = 2 | ||||
|     MINOR_VERSION = 1 | ||||
|     VERSION = 1 | ||||
|     MINOR_VERSION = 2 | ||||
|  | ||||
|     def __init__(self) -> None: | ||||
|         """Initialize the config flow.""" | ||||
| @@ -124,7 +119,7 @@ class AirOSConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|         else: | ||||
|             await self.async_set_unique_id(airos_data.derived.mac) | ||||
|  | ||||
|             if self.source in [SOURCE_REAUTH, SOURCE_RECONFIGURE]: | ||||
|             if self.source == SOURCE_REAUTH: | ||||
|                 self._abort_if_unique_id_mismatch() | ||||
|             else: | ||||
|                 self._abort_if_unique_id_configured() | ||||
| @@ -169,54 +164,3 @@ class AirOSConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             ), | ||||
|             errors=self.errors, | ||||
|         ) | ||||
|  | ||||
|     async def async_step_reconfigure( | ||||
|         self, | ||||
|         user_input: Mapping[str, Any] | None = None, | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Handle reconfiguration of airOS.""" | ||||
|         self.errors = {} | ||||
|         entry = self._get_reconfigure_entry() | ||||
|         current_data = entry.data | ||||
|  | ||||
|         if user_input is not None: | ||||
|             validate_data = {**current_data, **user_input} | ||||
|             if await self._validate_and_get_device_info(config_data=validate_data): | ||||
|                 return self.async_update_reload_and_abort( | ||||
|                     entry, | ||||
|                     data_updates=validate_data, | ||||
|                 ) | ||||
|  | ||||
|         return self.async_show_form( | ||||
|             step_id="reconfigure", | ||||
|             data_schema=vol.Schema( | ||||
|                 { | ||||
|                     vol.Required(CONF_PASSWORD): TextSelector( | ||||
|                         TextSelectorConfig( | ||||
|                             type=TextSelectorType.PASSWORD, | ||||
|                             autocomplete="current-password", | ||||
|                         ) | ||||
|                     ), | ||||
|                     vol.Required(SECTION_ADVANCED_SETTINGS): section( | ||||
|                         vol.Schema( | ||||
|                             { | ||||
|                                 vol.Required( | ||||
|                                     CONF_SSL, | ||||
|                                     default=current_data[SECTION_ADVANCED_SETTINGS][ | ||||
|                                         CONF_SSL | ||||
|                                     ], | ||||
|                                 ): bool, | ||||
|                                 vol.Required( | ||||
|                                     CONF_VERIFY_SSL, | ||||
|                                     default=current_data[SECTION_ADVANCED_SETTINGS][ | ||||
|                                         CONF_VERIFY_SSL | ||||
|                                     ], | ||||
|                                 ): bool, | ||||
|                             } | ||||
|                         ), | ||||
|                         {"collapsed": True}, | ||||
|                     ), | ||||
|                 } | ||||
|             ), | ||||
|             errors=self.errors, | ||||
|         ) | ||||
|   | ||||
| @@ -33,14 +33,9 @@ class AirOSEntity(CoordinatorEntity[AirOSDataUpdateCoordinator]): | ||||
|         self._attr_device_info = DeviceInfo( | ||||
|             connections={(CONNECTION_NETWORK_MAC, airos_data.derived.mac)}, | ||||
|             configuration_url=configuration_url, | ||||
|             identifiers={(DOMAIN, airos_data.derived.mac)}, | ||||
|             identifiers={(DOMAIN, str(airos_data.host.device_id))}, | ||||
|             manufacturer=MANUFACTURER, | ||||
|             model=airos_data.host.devmodel, | ||||
|             model_id=( | ||||
|                 sku | ||||
|                 if (sku := airos_data.derived.sku) not in ["UNKNOWN", "AMBIGUOUS"] | ||||
|                 else None | ||||
|             ), | ||||
|             name=airos_data.host.hostname, | ||||
|             sw_version=airos_data.host.fwversion, | ||||
|         ) | ||||
|   | ||||
| @@ -4,8 +4,7 @@ | ||||
|   "codeowners": ["@CoMPaTech"], | ||||
|   "config_flow": true, | ||||
|   "documentation": "https://www.home-assistant.io/integrations/airos", | ||||
|   "integration_type": "device", | ||||
|   "iot_class": "local_polling", | ||||
|   "quality_scale": "silver", | ||||
|   "requirements": ["airos==0.5.6"] | ||||
|   "quality_scale": "bronze", | ||||
|   "requirements": ["airos==0.5.4"] | ||||
| } | ||||
|   | ||||
| @@ -32,11 +32,11 @@ rules: | ||||
|   config-entry-unloading: done | ||||
|   docs-configuration-parameters: done | ||||
|   docs-installation-parameters: done | ||||
|   entity-unavailable: done | ||||
|   entity-unavailable: todo | ||||
|   integration-owner: done | ||||
|   log-when-unavailable: done | ||||
|   parallel-updates: done | ||||
|   reauthentication-flow: done | ||||
|   log-when-unavailable: todo | ||||
|   parallel-updates: todo | ||||
|   reauthentication-flow: todo | ||||
|   test-coverage: done | ||||
|  | ||||
|   # Gold | ||||
| @@ -48,9 +48,9 @@ rules: | ||||
|   docs-examples: todo | ||||
|   docs-known-limitations: done | ||||
|   docs-supported-devices: done | ||||
|   docs-supported-functions: done | ||||
|   docs-supported-functions: todo | ||||
|   docs-troubleshooting: done | ||||
|   docs-use-cases: done | ||||
|   docs-use-cases: todo | ||||
|   dynamic-devices: todo | ||||
|   entity-category: done | ||||
|   entity-device-class: done | ||||
| @@ -60,7 +60,7 @@ rules: | ||||
|   icon-translations: | ||||
|     status: exempt | ||||
|     comment: no (custom) icons used or envisioned | ||||
|   reconfiguration-flow: done | ||||
|   reconfiguration-flow: todo | ||||
|   repair-issues: todo | ||||
|   stale-devices: todo | ||||
|  | ||||
|   | ||||
| @@ -10,27 +10,6 @@ | ||||
|           "password": "[%key:component::airos::config::step::user::data_description::password%]" | ||||
|         } | ||||
|       }, | ||||
|       "reconfigure": { | ||||
|         "data": { | ||||
|           "password": "[%key:common::config_flow::data::password%]" | ||||
|         }, | ||||
|         "data_description": { | ||||
|           "password": "[%key:component::airos::config::step::user::data_description::password%]" | ||||
|         }, | ||||
|         "sections": { | ||||
|           "advanced_settings": { | ||||
|             "name": "[%key:component::airos::config::step::user::sections::advanced_settings::name%]", | ||||
|             "data": { | ||||
|               "ssl": "[%key:component::airos::config::step::user::sections::advanced_settings::data::ssl%]", | ||||
|               "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]" | ||||
|             }, | ||||
|             "data_description": { | ||||
|               "ssl": "[%key:component::airos::config::step::user::sections::advanced_settings::data_description::ssl%]", | ||||
|               "verify_ssl": "[%key:component::airos::config::step::user::sections::advanced_settings::data_description::verify_ssl%]" | ||||
|             } | ||||
|           } | ||||
|         } | ||||
|       }, | ||||
|       "user": { | ||||
|         "data": { | ||||
|           "host": "[%key:common::config_flow::data::host%]", | ||||
| @@ -44,7 +23,6 @@ | ||||
|         }, | ||||
|         "sections": { | ||||
|           "advanced_settings": { | ||||
|             "name": "Advanced settings", | ||||
|             "data": { | ||||
|               "ssl": "Use HTTPS", | ||||
|               "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]" | ||||
| @@ -66,7 +44,6 @@ | ||||
|     "abort": { | ||||
|       "already_configured": "[%key:common::config_flow::abort::already_configured_device%]", | ||||
|       "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]", | ||||
|       "reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]", | ||||
|       "unique_id_mismatch": "Re-authentication should be used for the same device not a new one" | ||||
|     } | ||||
|   }, | ||||
|   | ||||
| @@ -7,5 +7,5 @@ | ||||
|   "integration_type": "hub", | ||||
|   "iot_class": "local_polling", | ||||
|   "loggers": ["aioairq"], | ||||
|   "requirements": ["aioairq==0.4.7"] | ||||
|   "requirements": ["aioairq==0.4.6"] | ||||
| } | ||||
|   | ||||
| @@ -6,13 +6,8 @@ import dataclasses | ||||
| import logging | ||||
| from typing import Any | ||||
|  | ||||
| from airthings_ble import ( | ||||
|     AirthingsBluetoothDeviceData, | ||||
|     AirthingsDevice, | ||||
|     UnsupportedDeviceError, | ||||
| ) | ||||
| from airthings_ble import AirthingsBluetoothDeviceData, AirthingsDevice | ||||
| from bleak import BleakError | ||||
| from habluetooth import BluetoothServiceInfoBleak | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.components import bluetooth | ||||
| @@ -32,7 +27,6 @@ SERVICE_UUIDS = [ | ||||
|     "b42e4a8e-ade7-11e4-89d3-123b93f75cba", | ||||
|     "b42e1c08-ade7-11e4-89d3-123b93f75cba", | ||||
|     "b42e3882-ade7-11e4-89d3-123b93f75cba", | ||||
|     "b42e90a2-ade7-11e4-89d3-123b93f75cba", | ||||
| ] | ||||
|  | ||||
|  | ||||
| @@ -43,7 +37,6 @@ class Discovery: | ||||
|     name: str | ||||
|     discovery_info: BluetoothServiceInfo | ||||
|     device: AirthingsDevice | ||||
|     data: AirthingsBluetoothDeviceData | ||||
|  | ||||
|  | ||||
| def get_name(device: AirthingsDevice) -> str: | ||||
| @@ -51,7 +44,7 @@ def get_name(device: AirthingsDevice) -> str: | ||||
|  | ||||
|     name = device.friendly_name() | ||||
|     if identifier := device.identifier: | ||||
|         name += f" ({device.model.value}{identifier})" | ||||
|         name += f" ({identifier})" | ||||
|     return name | ||||
|  | ||||
|  | ||||
| @@ -69,8 +62,8 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|         self._discovered_device: Discovery | None = None | ||||
|         self._discovered_devices: dict[str, Discovery] = {} | ||||
|  | ||||
|     async def _get_device( | ||||
|         self, data: AirthingsBluetoothDeviceData, discovery_info: BluetoothServiceInfo | ||||
|     async def _get_device_data( | ||||
|         self, discovery_info: BluetoothServiceInfo | ||||
|     ) -> AirthingsDevice: | ||||
|         ble_device = bluetooth.async_ble_device_from_address( | ||||
|             self.hass, discovery_info.address | ||||
| @@ -79,8 +72,10 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             _LOGGER.debug("no ble_device in _get_device_data") | ||||
|             raise AirthingsDeviceUpdateError("No ble_device") | ||||
|  | ||||
|         airthings = AirthingsBluetoothDeviceData(_LOGGER) | ||||
|  | ||||
|         try: | ||||
|             device = await data.update_device(ble_device) | ||||
|             data = await airthings.update_device(ble_device) | ||||
|         except BleakError as err: | ||||
|             _LOGGER.error( | ||||
|                 "Error connecting to and getting data from %s: %s", | ||||
| @@ -88,15 +83,12 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|                 err, | ||||
|             ) | ||||
|             raise AirthingsDeviceUpdateError("Failed getting device data") from err | ||||
|         except UnsupportedDeviceError: | ||||
|             _LOGGER.debug("Skipping unsupported device: %s", discovery_info.name) | ||||
|             raise | ||||
|         except Exception as err: | ||||
|             _LOGGER.error( | ||||
|                 "Unknown error occurred from %s: %s", discovery_info.address, err | ||||
|             ) | ||||
|             raise | ||||
|         return device | ||||
|         return data | ||||
|  | ||||
|     async def async_step_bluetooth( | ||||
|         self, discovery_info: BluetoothServiceInfo | ||||
| @@ -106,21 +98,17 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|         await self.async_set_unique_id(discovery_info.address) | ||||
|         self._abort_if_unique_id_configured() | ||||
|  | ||||
|         data = AirthingsBluetoothDeviceData(logger=_LOGGER) | ||||
|  | ||||
|         try: | ||||
|             device = await self._get_device(data=data, discovery_info=discovery_info) | ||||
|             device = await self._get_device_data(discovery_info) | ||||
|         except AirthingsDeviceUpdateError: | ||||
|             return self.async_abort(reason="cannot_connect") | ||||
|         except UnsupportedDeviceError: | ||||
|             return self.async_abort(reason="unsupported_device") | ||||
|         except Exception: | ||||
|             _LOGGER.exception("Unknown error occurred") | ||||
|             return self.async_abort(reason="unknown") | ||||
|  | ||||
|         name = get_name(device) | ||||
|         self.context["title_placeholders"] = {"name": name} | ||||
|         self._discovered_device = Discovery(name, discovery_info, device, data=data) | ||||
|         self._discovered_device = Discovery(name, discovery_info, device) | ||||
|  | ||||
|         return await self.async_step_bluetooth_confirm() | ||||
|  | ||||
| @@ -129,12 +117,6 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Confirm discovery.""" | ||||
|         if user_input is not None: | ||||
|             if ( | ||||
|                 self._discovered_device is not None | ||||
|                 and self._discovered_device.device.firmware.need_firmware_upgrade | ||||
|             ): | ||||
|                 return self.async_abort(reason="firmware_upgrade_required") | ||||
|  | ||||
|             return self.async_create_entry( | ||||
|                 title=self.context["title_placeholders"]["name"], data={} | ||||
|             ) | ||||
| @@ -155,9 +137,6 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             self._abort_if_unique_id_configured() | ||||
|             discovery = self._discovered_devices[address] | ||||
|  | ||||
|             if discovery.device.firmware.need_firmware_upgrade: | ||||
|                 return self.async_abort(reason="firmware_upgrade_required") | ||||
|  | ||||
|             self.context["title_placeholders"] = { | ||||
|                 "name": discovery.name, | ||||
|             } | ||||
| @@ -167,47 +146,26 @@ class AirthingsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             return self.async_create_entry(title=discovery.name, data={}) | ||||
|  | ||||
|         current_addresses = self._async_current_ids(include_ignore=False) | ||||
|         devices: list[BluetoothServiceInfoBleak] = [] | ||||
|         for discovery_info in async_discovered_service_info(self.hass): | ||||
|             address = discovery_info.address | ||||
|             if address in current_addresses or address in self._discovered_devices: | ||||
|                 continue | ||||
|  | ||||
|             if MFCT_ID not in discovery_info.manufacturer_data: | ||||
|                 continue | ||||
|             if not any(uuid in SERVICE_UUIDS for uuid in discovery_info.service_uuids): | ||||
|                 _LOGGER.debug( | ||||
|                     "Skipping unsupported device: %s (%s)", discovery_info.name, address | ||||
|                 ) | ||||
|                 continue | ||||
|             devices.append(discovery_info) | ||||
|  | ||||
|         for discovery_info in devices: | ||||
|             address = discovery_info.address | ||||
|             data = AirthingsBluetoothDeviceData(logger=_LOGGER) | ||||
|             if not any(uuid in SERVICE_UUIDS for uuid in discovery_info.service_uuids): | ||||
|                 continue | ||||
|  | ||||
|             try: | ||||
|                 device = await self._get_device(data, discovery_info) | ||||
|                 device = await self._get_device_data(discovery_info) | ||||
|             except AirthingsDeviceUpdateError: | ||||
|                 _LOGGER.error( | ||||
|                     "Error connecting to and getting data from %s (%s)", | ||||
|                     discovery_info.name, | ||||
|                     discovery_info.address, | ||||
|                 ) | ||||
|                 continue | ||||
|             except UnsupportedDeviceError: | ||||
|                 _LOGGER.debug( | ||||
|                     "Skipping unsupported device: %s (%s)", | ||||
|                     discovery_info.name, | ||||
|                     discovery_info.address, | ||||
|                 ) | ||||
|                 continue | ||||
|                 return self.async_abort(reason="cannot_connect") | ||||
|             except Exception: | ||||
|                 _LOGGER.exception("Unknown error occurred") | ||||
|                 return self.async_abort(reason="unknown") | ||||
|             name = get_name(device) | ||||
|             _LOGGER.debug("Discovered Airthings device: %s (%s)", name, address) | ||||
|             self._discovered_devices[address] = Discovery( | ||||
|                 name, discovery_info, device, data | ||||
|             ) | ||||
|             self._discovered_devices[address] = Discovery(name, discovery_info, device) | ||||
|  | ||||
|         if not self._discovered_devices: | ||||
|             return self.async_abort(reason="no_devices_found") | ||||
|   | ||||
| @@ -17,10 +17,6 @@ | ||||
|     { | ||||
|       "manufacturer_id": 820, | ||||
|       "service_uuid": "b42e3882-ade7-11e4-89d3-123b93f75cba" | ||||
|     }, | ||||
|     { | ||||
|       "manufacturer_id": 820, | ||||
|       "service_uuid": "b42e90a2-ade7-11e4-89d3-123b93f75cba" | ||||
|     } | ||||
|   ], | ||||
|   "codeowners": ["@vincegio", "@LaStrada"], | ||||
| @@ -28,5 +24,5 @@ | ||||
|   "dependencies": ["bluetooth_adapters"], | ||||
|   "documentation": "https://www.home-assistant.io/integrations/airthings_ble", | ||||
|   "iot_class": "local_polling", | ||||
|   "requirements": ["airthings-ble==1.1.1"] | ||||
|   "requirements": ["airthings-ble==0.9.2"] | ||||
| } | ||||
|   | ||||
| @@ -16,12 +16,10 @@ from homeassistant.components.sensor import ( | ||||
| from homeassistant.const import ( | ||||
|     CONCENTRATION_PARTS_PER_BILLION, | ||||
|     CONCENTRATION_PARTS_PER_MILLION, | ||||
|     LIGHT_LUX, | ||||
|     PERCENTAGE, | ||||
|     EntityCategory, | ||||
|     Platform, | ||||
|     UnitOfPressure, | ||||
|     UnitOfSoundPressure, | ||||
|     UnitOfTemperature, | ||||
| ) | ||||
| from homeassistant.core import HomeAssistant, callback | ||||
| @@ -114,21 +112,6 @@ SENSORS_MAPPING_TEMPLATE: dict[str, SensorEntityDescription] = { | ||||
|         state_class=SensorStateClass.MEASUREMENT, | ||||
|         suggested_display_precision=0, | ||||
|     ), | ||||
|     "lux": SensorEntityDescription( | ||||
|         key="lux", | ||||
|         device_class=SensorDeviceClass.ILLUMINANCE, | ||||
|         native_unit_of_measurement=LIGHT_LUX, | ||||
|         state_class=SensorStateClass.MEASUREMENT, | ||||
|         suggested_display_precision=0, | ||||
|     ), | ||||
|     "noise": SensorEntityDescription( | ||||
|         key="noise", | ||||
|         translation_key="ambient_noise", | ||||
|         device_class=SensorDeviceClass.SOUND_PRESSURE, | ||||
|         native_unit_of_measurement=UnitOfSoundPressure.WEIGHTED_DECIBEL_A, | ||||
|         state_class=SensorStateClass.MEASUREMENT, | ||||
|         suggested_display_precision=0, | ||||
|     ), | ||||
| } | ||||
|  | ||||
| PARALLEL_UPDATES = 0 | ||||
|   | ||||
| @@ -20,8 +20,6 @@ | ||||
|       "already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]", | ||||
|       "already_configured": "[%key:common::config_flow::abort::already_configured_device%]", | ||||
|       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]", | ||||
|       "firmware_upgrade_required": "Your device requires a firmware upgrade. Please use the Airthings app (Android/iOS) to upgrade it.", | ||||
|       "unsupported_device": "Unsupported device", | ||||
|       "unknown": "[%key:common::config_flow::error::unknown%]" | ||||
|     } | ||||
|   }, | ||||
| @@ -41,9 +39,6 @@ | ||||
|       }, | ||||
|       "illuminance": { | ||||
|         "name": "[%key:component::sensor::entity_component::illuminance::name%]" | ||||
|       }, | ||||
|       "ambient_noise": { | ||||
|         "name": "Ambient noise" | ||||
|       } | ||||
|     } | ||||
|   } | ||||
|   | ||||
| @@ -2,9 +2,10 @@ | ||||
|  | ||||
| from __future__ import annotations | ||||
|  | ||||
| import asyncio | ||||
| from datetime import timedelta | ||||
| import logging | ||||
| from typing import Any, Final, final | ||||
| from typing import TYPE_CHECKING, Any, Final, final | ||||
|  | ||||
| from propcache.api import cached_property | ||||
| import voluptuous as vol | ||||
| @@ -27,6 +28,8 @@ from homeassistant.helpers import config_validation as cv | ||||
| from homeassistant.helpers.config_validation import make_entity_service_schema | ||||
| from homeassistant.helpers.entity import Entity, EntityDescription | ||||
| from homeassistant.helpers.entity_component import EntityComponent | ||||
| from homeassistant.helpers.entity_platform import EntityPlatform | ||||
| from homeassistant.helpers.frame import ReportBehavior, report_usage | ||||
| from homeassistant.helpers.typing import ConfigType | ||||
| from homeassistant.util.hass_dict import HassKey | ||||
|  | ||||
| @@ -146,11 +149,68 @@ class AlarmControlPanelEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_A | ||||
|     ) | ||||
|     _alarm_control_panel_option_default_code: str | None = None | ||||
|  | ||||
|     __alarm_legacy_state: bool = False | ||||
|  | ||||
|     def __init_subclass__(cls, **kwargs: Any) -> None: | ||||
|         """Post initialisation processing.""" | ||||
|         super().__init_subclass__(**kwargs) | ||||
|         if any(method in cls.__dict__ for method in ("_attr_state", "state")): | ||||
|             # Integrations should use the 'alarm_state' property instead of | ||||
|             # setting the state directly. | ||||
|             cls.__alarm_legacy_state = True | ||||
|  | ||||
|     def __setattr__(self, name: str, value: Any, /) -> None: | ||||
|         """Set attribute. | ||||
|  | ||||
|         Deprecation warning if setting '_attr_state' directly | ||||
|         unless already reported. | ||||
|         """ | ||||
|         if name == "_attr_state": | ||||
|             self._report_deprecated_alarm_state_handling() | ||||
|         return super().__setattr__(name, value) | ||||
|  | ||||
|     @callback | ||||
|     def add_to_platform_start( | ||||
|         self, | ||||
|         hass: HomeAssistant, | ||||
|         platform: EntityPlatform, | ||||
|         parallel_updates: asyncio.Semaphore | None, | ||||
|     ) -> None: | ||||
|         """Start adding an entity to a platform.""" | ||||
|         super().add_to_platform_start(hass, platform, parallel_updates) | ||||
|         if self.__alarm_legacy_state: | ||||
|             self._report_deprecated_alarm_state_handling() | ||||
|  | ||||
|     @callback | ||||
|     def _report_deprecated_alarm_state_handling(self) -> None: | ||||
|         """Report on deprecated handling of alarm state. | ||||
|  | ||||
|         Integrations should implement alarm_state instead of using state directly. | ||||
|         """ | ||||
|         report_usage( | ||||
|             "is setting state directly." | ||||
|             f" Entity {self.entity_id} ({type(self)}) should implement the 'alarm_state'" | ||||
|             " property and return its state using the AlarmControlPanelState enum", | ||||
|             core_integration_behavior=ReportBehavior.ERROR, | ||||
|             custom_integration_behavior=ReportBehavior.LOG, | ||||
|             breaks_in_ha_version="2025.11", | ||||
|             integration_domain=self.platform.platform_name if self.platform else None, | ||||
|             exclude_integrations={DOMAIN}, | ||||
|         ) | ||||
|  | ||||
|     @final | ||||
|     @property | ||||
|     def state(self) -> str | None: | ||||
|         """Return the current state.""" | ||||
|         return self.alarm_state | ||||
|         if (alarm_state := self.alarm_state) is not None: | ||||
|             return alarm_state | ||||
|         if self._attr_state is not None: | ||||
|             # Backwards compatibility for integrations that set state directly | ||||
|             # Should be removed in 2025.11 | ||||
|             if TYPE_CHECKING: | ||||
|                 assert isinstance(self._attr_state, str) | ||||
|             return self._attr_state | ||||
|         return None | ||||
|  | ||||
|     @cached_property | ||||
|     def alarm_state(self) -> AlarmControlPanelState | None: | ||||
|   | ||||
| @@ -1472,10 +1472,10 @@ class AlexaModeController(AlexaCapability): | ||||
|             # Return state instead of position when using ModeController. | ||||
|             mode = self.entity.state | ||||
|             if mode in ( | ||||
|                 cover.CoverState.OPEN, | ||||
|                 cover.CoverState.OPENING, | ||||
|                 cover.CoverState.CLOSED, | ||||
|                 cover.CoverState.CLOSING, | ||||
|                 cover.STATE_OPEN, | ||||
|                 cover.STATE_OPENING, | ||||
|                 cover.STATE_CLOSED, | ||||
|                 cover.STATE_CLOSING, | ||||
|                 STATE_UNKNOWN, | ||||
|             ): | ||||
|                 return f"{cover.ATTR_POSITION}.{mode}" | ||||
| @@ -1594,11 +1594,11 @@ class AlexaModeController(AlexaCapability): | ||||
|                 ["Position", AlexaGlobalCatalog.SETTING_OPENING], False | ||||
|             ) | ||||
|             self._resource.add_mode( | ||||
|                 f"{cover.ATTR_POSITION}.{cover.CoverState.OPEN}", | ||||
|                 f"{cover.ATTR_POSITION}.{cover.STATE_OPEN}", | ||||
|                 [AlexaGlobalCatalog.VALUE_OPEN], | ||||
|             ) | ||||
|             self._resource.add_mode( | ||||
|                 f"{cover.ATTR_POSITION}.{cover.CoverState.CLOSED}", | ||||
|                 f"{cover.ATTR_POSITION}.{cover.STATE_CLOSED}", | ||||
|                 [AlexaGlobalCatalog.VALUE_CLOSE], | ||||
|             ) | ||||
|             self._resource.add_mode( | ||||
| @@ -1651,22 +1651,22 @@ class AlexaModeController(AlexaCapability): | ||||
|                 raise_labels.append(AlexaSemantics.ACTION_OPEN) | ||||
|                 self._semantics.add_states_to_value( | ||||
|                     [AlexaSemantics.STATES_CLOSED], | ||||
|                     f"{cover.ATTR_POSITION}.{cover.CoverState.CLOSED}", | ||||
|                     f"{cover.ATTR_POSITION}.{cover.STATE_CLOSED}", | ||||
|                 ) | ||||
|                 self._semantics.add_states_to_value( | ||||
|                     [AlexaSemantics.STATES_OPEN], | ||||
|                     f"{cover.ATTR_POSITION}.{cover.CoverState.OPEN}", | ||||
|                     f"{cover.ATTR_POSITION}.{cover.STATE_OPEN}", | ||||
|                 ) | ||||
|  | ||||
|             self._semantics.add_action_to_directive( | ||||
|                 lower_labels, | ||||
|                 "SetMode", | ||||
|                 {"mode": f"{cover.ATTR_POSITION}.{cover.CoverState.CLOSED}"}, | ||||
|                 {"mode": f"{cover.ATTR_POSITION}.{cover.STATE_CLOSED}"}, | ||||
|             ) | ||||
|             self._semantics.add_action_to_directive( | ||||
|                 raise_labels, | ||||
|                 "SetMode", | ||||
|                 {"mode": f"{cover.ATTR_POSITION}.{cover.CoverState.OPEN}"}, | ||||
|                 {"mode": f"{cover.ATTR_POSITION}.{cover.STATE_OPEN}"}, | ||||
|             ) | ||||
|  | ||||
|             return self._semantics.serialize_semantics() | ||||
|   | ||||
| @@ -1261,9 +1261,9 @@ async def async_api_set_mode( | ||||
|     elif instance == f"{cover.DOMAIN}.{cover.ATTR_POSITION}": | ||||
|         position = mode.split(".")[1] | ||||
|  | ||||
|         if position == cover.CoverState.CLOSED: | ||||
|         if position == cover.STATE_CLOSED: | ||||
|             service = cover.SERVICE_CLOSE_COVER | ||||
|         elif position == cover.CoverState.OPEN: | ||||
|         elif position == cover.STATE_OPEN: | ||||
|             service = cover.SERVICE_OPEN_COVER | ||||
|         elif position == "custom": | ||||
|             service = cover.SERVICE_STOP_COVER | ||||
|   | ||||
| @@ -18,9 +18,7 @@ from homeassistant.components.binary_sensor import ( | ||||
| from homeassistant.const import EntityCategory | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
| import homeassistant.helpers.entity_registry as er | ||||
|  | ||||
| from .const import _LOGGER, DOMAIN | ||||
| from .coordinator import AmazonConfigEntry | ||||
| from .entity import AmazonEntity | ||||
| from .utils import async_update_unique_id | ||||
| @@ -53,47 +51,11 @@ BINARY_SENSORS: Final = ( | ||||
|         ), | ||||
|         is_supported=lambda device, key: device.sensors.get(key) is not None, | ||||
|         is_available_fn=lambda device, key: ( | ||||
|             device.online | ||||
|             and (sensor := device.sensors.get(key)) is not None | ||||
|             and sensor.error is False | ||||
|             device.online and device.sensors[key].error is False | ||||
|         ), | ||||
|     ), | ||||
| ) | ||||
|  | ||||
| DEPRECATED_BINARY_SENSORS: Final = ( | ||||
|     AmazonBinarySensorEntityDescription( | ||||
|         key="bluetooth", | ||||
|         entity_category=EntityCategory.DIAGNOSTIC, | ||||
|         translation_key="bluetooth", | ||||
|         is_on_fn=lambda device, key: False, | ||||
|     ), | ||||
|     AmazonBinarySensorEntityDescription( | ||||
|         key="babyCryDetectionState", | ||||
|         translation_key="baby_cry_detection", | ||||
|         is_on_fn=lambda device, key: False, | ||||
|     ), | ||||
|     AmazonBinarySensorEntityDescription( | ||||
|         key="beepingApplianceDetectionState", | ||||
|         translation_key="beeping_appliance_detection", | ||||
|         is_on_fn=lambda device, key: False, | ||||
|     ), | ||||
|     AmazonBinarySensorEntityDescription( | ||||
|         key="coughDetectionState", | ||||
|         translation_key="cough_detection", | ||||
|         is_on_fn=lambda device, key: False, | ||||
|     ), | ||||
|     AmazonBinarySensorEntityDescription( | ||||
|         key="dogBarkDetectionState", | ||||
|         translation_key="dog_bark_detection", | ||||
|         is_on_fn=lambda device, key: False, | ||||
|     ), | ||||
|     AmazonBinarySensorEntityDescription( | ||||
|         key="waterSoundsDetectionState", | ||||
|         translation_key="water_sounds_detection", | ||||
|         is_on_fn=lambda device, key: False, | ||||
|     ), | ||||
| ) | ||||
|  | ||||
|  | ||||
| async def async_setup_entry( | ||||
|     hass: HomeAssistant, | ||||
| @@ -104,8 +66,6 @@ async def async_setup_entry( | ||||
|  | ||||
|     coordinator = entry.runtime_data | ||||
|  | ||||
|     entity_registry = er.async_get(hass) | ||||
|  | ||||
|     # Replace unique id for "detectionState" binary sensor | ||||
|     await async_update_unique_id( | ||||
|         hass, | ||||
| @@ -115,16 +75,6 @@ async def async_setup_entry( | ||||
|         "detectionState", | ||||
|     ) | ||||
|  | ||||
|     # Clean up deprecated sensors | ||||
|     for sensor_desc in DEPRECATED_BINARY_SENSORS: | ||||
|         for serial_num in coordinator.data: | ||||
|             unique_id = f"{serial_num}-{sensor_desc.key}" | ||||
|             if entity_id := entity_registry.async_get_entity_id( | ||||
|                 BINARY_SENSOR_DOMAIN, DOMAIN, unique_id | ||||
|             ): | ||||
|                 _LOGGER.debug("Removing deprecated entity %s", entity_id) | ||||
|                 entity_registry.async_remove(entity_id) | ||||
|  | ||||
|     known_devices: set[str] = set() | ||||
|  | ||||
|     def _check_device() -> None: | ||||
|   | ||||
| @@ -8,5 +8,5 @@ | ||||
|   "iot_class": "cloud_polling", | ||||
|   "loggers": ["aioamazondevices"], | ||||
|   "quality_scale": "platinum", | ||||
|   "requirements": ["aioamazondevices==6.4.4"] | ||||
|   "requirements": ["aioamazondevices==6.2.7"] | ||||
| } | ||||
|   | ||||
| @@ -32,9 +32,7 @@ class AmazonSensorEntityDescription(SensorEntityDescription): | ||||
|  | ||||
|     native_unit_of_measurement_fn: Callable[[AmazonDevice, str], str] | None = None | ||||
|     is_available_fn: Callable[[AmazonDevice, str], bool] = lambda device, key: ( | ||||
|         device.online | ||||
|         and (sensor := device.sensors.get(key)) is not None | ||||
|         and sensor.error is False | ||||
|         device.online and device.sensors[key].error is False | ||||
|     ) | ||||
|  | ||||
|  | ||||
| @@ -42,9 +40,9 @@ SENSORS: Final = ( | ||||
|     AmazonSensorEntityDescription( | ||||
|         key="temperature", | ||||
|         device_class=SensorDeviceClass.TEMPERATURE, | ||||
|         native_unit_of_measurement_fn=lambda device, key: ( | ||||
|         native_unit_of_measurement_fn=lambda device, _key: ( | ||||
|             UnitOfTemperature.CELSIUS | ||||
|             if key in device.sensors and device.sensors[key].scale == "CELSIUS" | ||||
|             if device.sensors[_key].scale == "CELSIUS" | ||||
|             else UnitOfTemperature.FAHRENHEIT | ||||
|         ), | ||||
|         state_class=SensorStateClass.MEASUREMENT, | ||||
|   | ||||
| @@ -18,11 +18,7 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
|  | ||||
| from .coordinator import AmazonConfigEntry | ||||
| from .entity import AmazonEntity | ||||
| from .utils import ( | ||||
|     alexa_api_call, | ||||
|     async_remove_dnd_from_virtual_group, | ||||
|     async_update_unique_id, | ||||
| ) | ||||
| from .utils import alexa_api_call, async_update_unique_id | ||||
|  | ||||
| PARALLEL_UPDATES = 1 | ||||
|  | ||||
| @@ -33,9 +29,7 @@ class AmazonSwitchEntityDescription(SwitchEntityDescription): | ||||
|  | ||||
|     is_on_fn: Callable[[AmazonDevice], bool] | ||||
|     is_available_fn: Callable[[AmazonDevice, str], bool] = lambda device, key: ( | ||||
|         device.online | ||||
|         and (sensor := device.sensors.get(key)) is not None | ||||
|         and sensor.error is False | ||||
|         device.online and device.sensors[key].error is False | ||||
|     ) | ||||
|     method: str | ||||
|  | ||||
| @@ -64,9 +58,6 @@ async def async_setup_entry( | ||||
|         hass, coordinator, SWITCH_DOMAIN, "do_not_disturb", "dnd" | ||||
|     ) | ||||
|  | ||||
|     # Remove DND switch from virtual groups | ||||
|     await async_remove_dnd_from_virtual_group(hass, coordinator) | ||||
|  | ||||
|     known_devices: set[str] = set() | ||||
|  | ||||
|     def _check_device() -> None: | ||||
|   | ||||
| @@ -4,10 +4,8 @@ from collections.abc import Awaitable, Callable, Coroutine | ||||
| from functools import wraps | ||||
| from typing import Any, Concatenate | ||||
|  | ||||
| from aioamazondevices.const import SPEAKER_GROUP_FAMILY | ||||
| from aioamazondevices.exceptions import CannotConnect, CannotRetrieveData | ||||
|  | ||||
| from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.exceptions import HomeAssistantError | ||||
| import homeassistant.helpers.entity_registry as er | ||||
| @@ -63,21 +61,3 @@ async def async_update_unique_id( | ||||
|  | ||||
|             # Update the registry with the new unique_id | ||||
|             entity_registry.async_update_entity(entity_id, new_unique_id=new_unique_id) | ||||
|  | ||||
|  | ||||
| async def async_remove_dnd_from_virtual_group( | ||||
|     hass: HomeAssistant, | ||||
|     coordinator: AmazonDevicesCoordinator, | ||||
| ) -> None: | ||||
|     """Remove entity DND from virtual group.""" | ||||
|     entity_registry = er.async_get(hass) | ||||
|  | ||||
|     for serial_num in coordinator.data: | ||||
|         unique_id = f"{serial_num}-do_not_disturb" | ||||
|         entity_id = entity_registry.async_get_entity_id( | ||||
|             DOMAIN, SWITCH_DOMAIN, unique_id | ||||
|         ) | ||||
|         is_group = coordinator.data[serial_num].device_family == SPEAKER_GROUP_FAMILY | ||||
|         if entity_id and is_group: | ||||
|             entity_registry.async_remove(entity_id) | ||||
|             _LOGGER.debug("Removed DND switch from virtual group %s", entity_id) | ||||
|   | ||||
| @@ -65,31 +65,6 @@ SENSOR_DESCRIPTIONS = [ | ||||
|         suggested_display_precision=2, | ||||
|         translation_placeholders={"sensor_name": "BME280"}, | ||||
|     ), | ||||
|     AltruistSensorEntityDescription( | ||||
|         device_class=SensorDeviceClass.HUMIDITY, | ||||
|         key="BME680_humidity", | ||||
|         translation_key="humidity", | ||||
|         native_unit_of_measurement=PERCENTAGE, | ||||
|         suggested_display_precision=2, | ||||
|         translation_placeholders={"sensor_name": "BME680"}, | ||||
|     ), | ||||
|     AltruistSensorEntityDescription( | ||||
|         device_class=SensorDeviceClass.PRESSURE, | ||||
|         key="BME680_pressure", | ||||
|         translation_key="pressure", | ||||
|         native_unit_of_measurement=UnitOfPressure.PA, | ||||
|         suggested_unit_of_measurement=UnitOfPressure.MMHG, | ||||
|         suggested_display_precision=0, | ||||
|         translation_placeholders={"sensor_name": "BME680"}, | ||||
|     ), | ||||
|     AltruistSensorEntityDescription( | ||||
|         device_class=SensorDeviceClass.TEMPERATURE, | ||||
|         key="BME680_temperature", | ||||
|         translation_key="temperature", | ||||
|         native_unit_of_measurement=UnitOfTemperature.CELSIUS, | ||||
|         suggested_display_precision=2, | ||||
|         translation_placeholders={"sensor_name": "BME680"}, | ||||
|     ), | ||||
|     AltruistSensorEntityDescription( | ||||
|         device_class=SensorDeviceClass.PRESSURE, | ||||
|         key="BMP_pressure", | ||||
|   | ||||
| @@ -4,15 +4,12 @@ from __future__ import annotations | ||||
|  | ||||
| from collections.abc import Mapping | ||||
| from functools import partial | ||||
| import json | ||||
| import logging | ||||
| from typing import Any, cast | ||||
|  | ||||
| import anthropic | ||||
| import voluptuous as vol | ||||
| from voluptuous_openapi import convert | ||||
|  | ||||
| from homeassistant.components.zone import ENTITY_ID_HOME | ||||
| from homeassistant.config_entries import ( | ||||
|     ConfigEntry, | ||||
|     ConfigEntryState, | ||||
| @@ -21,13 +18,7 @@ from homeassistant.config_entries import ( | ||||
|     ConfigSubentryFlow, | ||||
|     SubentryFlowResult, | ||||
| ) | ||||
| from homeassistant.const import ( | ||||
|     ATTR_LATITUDE, | ||||
|     ATTR_LONGITUDE, | ||||
|     CONF_API_KEY, | ||||
|     CONF_LLM_HASS_API, | ||||
|     CONF_NAME, | ||||
| ) | ||||
| from homeassistant.const import CONF_API_KEY, CONF_LLM_HASS_API, CONF_NAME | ||||
| from homeassistant.core import HomeAssistant, callback | ||||
| from homeassistant.helpers import llm | ||||
| from homeassistant.helpers.selector import ( | ||||
| @@ -46,23 +37,12 @@ from .const import ( | ||||
|     CONF_RECOMMENDED, | ||||
|     CONF_TEMPERATURE, | ||||
|     CONF_THINKING_BUDGET, | ||||
|     CONF_WEB_SEARCH, | ||||
|     CONF_WEB_SEARCH_CITY, | ||||
|     CONF_WEB_SEARCH_COUNTRY, | ||||
|     CONF_WEB_SEARCH_MAX_USES, | ||||
|     CONF_WEB_SEARCH_REGION, | ||||
|     CONF_WEB_SEARCH_TIMEZONE, | ||||
|     CONF_WEB_SEARCH_USER_LOCATION, | ||||
|     DEFAULT_CONVERSATION_NAME, | ||||
|     DOMAIN, | ||||
|     RECOMMENDED_CHAT_MODEL, | ||||
|     RECOMMENDED_MAX_TOKENS, | ||||
|     RECOMMENDED_TEMPERATURE, | ||||
|     RECOMMENDED_THINKING_BUDGET, | ||||
|     RECOMMENDED_WEB_SEARCH, | ||||
|     RECOMMENDED_WEB_SEARCH_MAX_USES, | ||||
|     RECOMMENDED_WEB_SEARCH_USER_LOCATION, | ||||
|     WEB_SEARCH_UNSUPPORTED_MODELS, | ||||
| ) | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
| @@ -188,14 +168,6 @@ class ConversationSubentryFlowHandler(ConfigSubentryFlow): | ||||
|                 CONF_THINKING_BUDGET, RECOMMENDED_THINKING_BUDGET | ||||
|             ) >= user_input.get(CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS): | ||||
|                 errors[CONF_THINKING_BUDGET] = "thinking_budget_too_large" | ||||
|             if user_input.get(CONF_WEB_SEARCH, RECOMMENDED_WEB_SEARCH): | ||||
|                 model = user_input.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL) | ||||
|                 if model.startswith(tuple(WEB_SEARCH_UNSUPPORTED_MODELS)): | ||||
|                     errors[CONF_WEB_SEARCH] = "web_search_unsupported_model" | ||||
|                 elif user_input.get( | ||||
|                     CONF_WEB_SEARCH_USER_LOCATION, RECOMMENDED_WEB_SEARCH_USER_LOCATION | ||||
|                 ): | ||||
|                     user_input.update(await self._get_location_data()) | ||||
|  | ||||
|             if not errors: | ||||
|                 if self._is_new: | ||||
| @@ -243,68 +215,6 @@ class ConversationSubentryFlowHandler(ConfigSubentryFlow): | ||||
|             errors=errors or None, | ||||
|         ) | ||||
|  | ||||
|     async def _get_location_data(self) -> dict[str, str]: | ||||
|         """Get approximate location data of the user.""" | ||||
|         location_data: dict[str, str] = {} | ||||
|         zone_home = self.hass.states.get(ENTITY_ID_HOME) | ||||
|         if zone_home is not None: | ||||
|             client = await self.hass.async_add_executor_job( | ||||
|                 partial( | ||||
|                     anthropic.AsyncAnthropic, | ||||
|                     api_key=self._get_entry().data[CONF_API_KEY], | ||||
|                 ) | ||||
|             ) | ||||
|             location_schema = vol.Schema( | ||||
|                 { | ||||
|                     vol.Optional( | ||||
|                         CONF_WEB_SEARCH_CITY, | ||||
|                         description="Free text input for the city, e.g. `San Francisco`", | ||||
|                     ): str, | ||||
|                     vol.Optional( | ||||
|                         CONF_WEB_SEARCH_REGION, | ||||
|                         description="Free text input for the region, e.g. `California`", | ||||
|                     ): str, | ||||
|                 } | ||||
|             ) | ||||
|             response = await client.messages.create( | ||||
|                 model=RECOMMENDED_CHAT_MODEL, | ||||
|                 messages=[ | ||||
|                     { | ||||
|                         "role": "user", | ||||
|                         "content": "Where are the following coordinates located: " | ||||
|                         f"({zone_home.attributes[ATTR_LATITUDE]}," | ||||
|                         f" {zone_home.attributes[ATTR_LONGITUDE]})? Please respond " | ||||
|                         "only with a JSON object using the following schema:\n" | ||||
|                         f"{convert(location_schema)}", | ||||
|                     }, | ||||
|                     { | ||||
|                         "role": "assistant", | ||||
|                         "content": "{",  # hints the model to skip any preamble | ||||
|                     }, | ||||
|                 ], | ||||
|                 max_tokens=RECOMMENDED_MAX_TOKENS, | ||||
|             ) | ||||
|             _LOGGER.debug("Model response: %s", response.content) | ||||
|             location_data = location_schema( | ||||
|                 json.loads( | ||||
|                     "{" | ||||
|                     + "".join( | ||||
|                         block.text | ||||
|                         for block in response.content | ||||
|                         if isinstance(block, anthropic.types.TextBlock) | ||||
|                     ) | ||||
|                 ) | ||||
|                 or {} | ||||
|             ) | ||||
|  | ||||
|         if self.hass.config.country: | ||||
|             location_data[CONF_WEB_SEARCH_COUNTRY] = self.hass.config.country | ||||
|         location_data[CONF_WEB_SEARCH_TIMEZONE] = self.hass.config.time_zone | ||||
|  | ||||
|         _LOGGER.debug("Location data: %s", location_data) | ||||
|  | ||||
|         return location_data | ||||
|  | ||||
|     async_step_user = async_step_set_options | ||||
|     async_step_reconfigure = async_step_set_options | ||||
|  | ||||
| @@ -363,18 +273,6 @@ def anthropic_config_option_schema( | ||||
|                 CONF_THINKING_BUDGET, | ||||
|                 default=RECOMMENDED_THINKING_BUDGET, | ||||
|             ): int, | ||||
|             vol.Optional( | ||||
|                 CONF_WEB_SEARCH, | ||||
|                 default=RECOMMENDED_WEB_SEARCH, | ||||
|             ): bool, | ||||
|             vol.Optional( | ||||
|                 CONF_WEB_SEARCH_MAX_USES, | ||||
|                 default=RECOMMENDED_WEB_SEARCH_MAX_USES, | ||||
|             ): int, | ||||
|             vol.Optional( | ||||
|                 CONF_WEB_SEARCH_USER_LOCATION, | ||||
|                 default=RECOMMENDED_WEB_SEARCH_USER_LOCATION, | ||||
|             ): bool, | ||||
|         } | ||||
|     ) | ||||
|     return schema | ||||
|   | ||||
| @@ -18,26 +18,10 @@ RECOMMENDED_TEMPERATURE = 1.0 | ||||
| CONF_THINKING_BUDGET = "thinking_budget" | ||||
| RECOMMENDED_THINKING_BUDGET = 0 | ||||
| MIN_THINKING_BUDGET = 1024 | ||||
| CONF_WEB_SEARCH = "web_search" | ||||
| RECOMMENDED_WEB_SEARCH = False | ||||
| CONF_WEB_SEARCH_USER_LOCATION = "user_location" | ||||
| RECOMMENDED_WEB_SEARCH_USER_LOCATION = False | ||||
| CONF_WEB_SEARCH_MAX_USES = "web_search_max_uses" | ||||
| RECOMMENDED_WEB_SEARCH_MAX_USES = 5 | ||||
| CONF_WEB_SEARCH_CITY = "city" | ||||
| CONF_WEB_SEARCH_REGION = "region" | ||||
| CONF_WEB_SEARCH_COUNTRY = "country" | ||||
| CONF_WEB_SEARCH_TIMEZONE = "timezone" | ||||
|  | ||||
| NON_THINKING_MODELS = [ | ||||
|     "claude-3-5",  # Both sonnet and haiku | ||||
|     "claude-3-opus", | ||||
|     "claude-3-haiku", | ||||
| ] | ||||
|  | ||||
| WEB_SEARCH_UNSUPPORTED_MODELS = [ | ||||
|     "claude-3-haiku", | ||||
|     "claude-3-opus", | ||||
|     "claude-3-5-sonnet-20240620", | ||||
|     "claude-3-5-sonnet-20241022", | ||||
| THINKING_MODELS = [ | ||||
|     "claude-3-7-sonnet", | ||||
|     "claude-sonnet-4-0", | ||||
|     "claude-opus-4-0", | ||||
|     "claude-opus-4-1", | ||||
| ] | ||||
|   | ||||
| @@ -1,17 +1,12 @@ | ||||
| """Base entity for Anthropic.""" | ||||
|  | ||||
| from collections.abc import AsyncGenerator, Callable, Iterable | ||||
| from dataclasses import dataclass, field | ||||
| import json | ||||
| from typing import Any | ||||
|  | ||||
| import anthropic | ||||
| from anthropic import AsyncStream | ||||
| from anthropic.types import ( | ||||
|     CitationsDelta, | ||||
|     CitationsWebSearchResultLocation, | ||||
|     CitationWebSearchResultLocationParam, | ||||
|     ContentBlockParam, | ||||
|     InputJSONDelta, | ||||
|     MessageDeltaUsage, | ||||
|     MessageParam, | ||||
| @@ -21,16 +16,11 @@ from anthropic.types import ( | ||||
|     RawContentBlockStopEvent, | ||||
|     RawMessageDeltaEvent, | ||||
|     RawMessageStartEvent, | ||||
|     RawMessageStopEvent, | ||||
|     RedactedThinkingBlock, | ||||
|     RedactedThinkingBlockParam, | ||||
|     ServerToolUseBlock, | ||||
|     ServerToolUseBlockParam, | ||||
|     SignatureDelta, | ||||
|     TextBlock, | ||||
|     TextBlockParam, | ||||
|     TextCitation, | ||||
|     TextCitationParam, | ||||
|     TextDelta, | ||||
|     ThinkingBlock, | ||||
|     ThinkingBlockParam, | ||||
| @@ -39,15 +29,9 @@ from anthropic.types import ( | ||||
|     ThinkingDelta, | ||||
|     ToolParam, | ||||
|     ToolResultBlockParam, | ||||
|     ToolUnionParam, | ||||
|     ToolUseBlock, | ||||
|     ToolUseBlockParam, | ||||
|     Usage, | ||||
|     WebSearchTool20250305Param, | ||||
|     WebSearchToolRequestErrorParam, | ||||
|     WebSearchToolResultBlock, | ||||
|     WebSearchToolResultBlockParam, | ||||
|     WebSearchToolResultError, | ||||
| ) | ||||
| from anthropic.types.message_create_params import MessageCreateParamsStreaming | ||||
| from voluptuous_openapi import convert | ||||
| @@ -64,21 +48,14 @@ from .const import ( | ||||
|     CONF_MAX_TOKENS, | ||||
|     CONF_TEMPERATURE, | ||||
|     CONF_THINKING_BUDGET, | ||||
|     CONF_WEB_SEARCH, | ||||
|     CONF_WEB_SEARCH_CITY, | ||||
|     CONF_WEB_SEARCH_COUNTRY, | ||||
|     CONF_WEB_SEARCH_MAX_USES, | ||||
|     CONF_WEB_SEARCH_REGION, | ||||
|     CONF_WEB_SEARCH_TIMEZONE, | ||||
|     CONF_WEB_SEARCH_USER_LOCATION, | ||||
|     DOMAIN, | ||||
|     LOGGER, | ||||
|     MIN_THINKING_BUDGET, | ||||
|     NON_THINKING_MODELS, | ||||
|     RECOMMENDED_CHAT_MODEL, | ||||
|     RECOMMENDED_MAX_TOKENS, | ||||
|     RECOMMENDED_TEMPERATURE, | ||||
|     RECOMMENDED_THINKING_BUDGET, | ||||
|     THINKING_MODELS, | ||||
| ) | ||||
|  | ||||
| # Max number of back and forth with the LLM to generate a response | ||||
| @@ -96,69 +73,6 @@ def _format_tool( | ||||
|     ) | ||||
|  | ||||
|  | ||||
| @dataclass(slots=True) | ||||
| class CitationDetails: | ||||
|     """Citation details for a content part.""" | ||||
|  | ||||
|     index: int = 0 | ||||
|     """Start position of the text.""" | ||||
|  | ||||
|     length: int = 0 | ||||
|     """Length of the relevant data.""" | ||||
|  | ||||
|     citations: list[TextCitationParam] = field(default_factory=list) | ||||
|     """Citations for the content part.""" | ||||
|  | ||||
|  | ||||
| @dataclass(slots=True) | ||||
| class ContentDetails: | ||||
|     """Native data for AssistantContent.""" | ||||
|  | ||||
|     citation_details: list[CitationDetails] = field(default_factory=list) | ||||
|  | ||||
|     def has_content(self) -> bool: | ||||
|         """Check if there is any content.""" | ||||
|         return any(detail.length > 0 for detail in self.citation_details) | ||||
|  | ||||
|     def has_citations(self) -> bool: | ||||
|         """Check if there are any citations.""" | ||||
|         return any(detail.citations for detail in self.citation_details) | ||||
|  | ||||
|     def add_citation_detail(self) -> None: | ||||
|         """Add a new citation detail.""" | ||||
|         if not self.citation_details or self.citation_details[-1].length > 0: | ||||
|             self.citation_details.append( | ||||
|                 CitationDetails( | ||||
|                     index=self.citation_details[-1].index | ||||
|                     + self.citation_details[-1].length | ||||
|                     if self.citation_details | ||||
|                     else 0 | ||||
|                 ) | ||||
|             ) | ||||
|  | ||||
|     def add_citation(self, citation: TextCitation) -> None: | ||||
|         """Add a citation to the current detail.""" | ||||
|         if not self.citation_details: | ||||
|             self.citation_details.append(CitationDetails()) | ||||
|         citation_param: TextCitationParam | None = None | ||||
|         if isinstance(citation, CitationsWebSearchResultLocation): | ||||
|             citation_param = CitationWebSearchResultLocationParam( | ||||
|                 type="web_search_result_location", | ||||
|                 title=citation.title, | ||||
|                 url=citation.url, | ||||
|                 cited_text=citation.cited_text, | ||||
|                 encrypted_index=citation.encrypted_index, | ||||
|             ) | ||||
|         if citation_param: | ||||
|             self.citation_details[-1].citations.append(citation_param) | ||||
|  | ||||
|     def delete_empty(self) -> None: | ||||
|         """Delete empty citation details.""" | ||||
|         self.citation_details = [ | ||||
|             detail for detail in self.citation_details if detail.citations | ||||
|         ] | ||||
|  | ||||
|  | ||||
| def _convert_content( | ||||
|     chat_content: Iterable[conversation.Content], | ||||
| ) -> list[MessageParam]: | ||||
| @@ -167,31 +81,15 @@ def _convert_content( | ||||
|  | ||||
|     for content in chat_content: | ||||
|         if isinstance(content, conversation.ToolResultContent): | ||||
|             if content.tool_name == "web_search": | ||||
|                 tool_result_block: ContentBlockParam = WebSearchToolResultBlockParam( | ||||
|                     type="web_search_tool_result", | ||||
|                     tool_use_id=content.tool_call_id, | ||||
|                     content=content.tool_result["content"] | ||||
|                     if "content" in content.tool_result | ||||
|                     else WebSearchToolRequestErrorParam( | ||||
|                         type="web_search_tool_result_error", | ||||
|                         error_code=content.tool_result.get("error_code", "unavailable"),  # type: ignore[typeddict-item] | ||||
|                     ), | ||||
|                 ) | ||||
|                 external_tool = True | ||||
|             else: | ||||
|             tool_result_block = ToolResultBlockParam( | ||||
|                 type="tool_result", | ||||
|                 tool_use_id=content.tool_call_id, | ||||
|                 content=json.dumps(content.tool_result), | ||||
|             ) | ||||
|                 external_tool = False | ||||
|             if not messages or messages[-1]["role"] != ( | ||||
|                 "assistant" if external_tool else "user" | ||||
|             ): | ||||
|             if not messages or messages[-1]["role"] != "user": | ||||
|                 messages.append( | ||||
|                     MessageParam( | ||||
|                         role="assistant" if external_tool else "user", | ||||
|                         role="user", | ||||
|                         content=[tool_result_block], | ||||
|                     ) | ||||
|                 ) | ||||
| @@ -253,56 +151,13 @@ def _convert_content( | ||||
|                         redacted_thinking_block | ||||
|                     ) | ||||
|             if content.content: | ||||
|                 current_index = 0 | ||||
|                 for detail in ( | ||||
|                     content.native.citation_details | ||||
|                     if isinstance(content.native, ContentDetails) | ||||
|                     else [CitationDetails(length=len(content.content))] | ||||
|                 ): | ||||
|                     if detail.index > current_index: | ||||
|                         # Add text block for any text without citations | ||||
|                 messages[-1]["content"].append(  # type: ignore[union-attr] | ||||
|                             TextBlockParam( | ||||
|                                 type="text", | ||||
|                                 text=content.content[current_index : detail.index], | ||||
|                             ) | ||||
|                         ) | ||||
|                     messages[-1]["content"].append(  # type: ignore[union-attr] | ||||
|                         TextBlockParam( | ||||
|                             type="text", | ||||
|                             text=content.content[ | ||||
|                                 detail.index : detail.index + detail.length | ||||
|                             ], | ||||
|                             citations=detail.citations, | ||||
|                         ) | ||||
|                         if detail.citations | ||||
|                         else TextBlockParam( | ||||
|                             type="text", | ||||
|                             text=content.content[ | ||||
|                                 detail.index : detail.index + detail.length | ||||
|                             ], | ||||
|                         ) | ||||
|                     ) | ||||
|                     current_index = detail.index + detail.length | ||||
|                 if current_index < len(content.content): | ||||
|                     # Add text block for any remaining text without citations | ||||
|                     messages[-1]["content"].append(  # type: ignore[union-attr] | ||||
|                         TextBlockParam( | ||||
|                             type="text", | ||||
|                             text=content.content[current_index:], | ||||
|                         ) | ||||
|                     TextBlockParam(type="text", text=content.content) | ||||
|                 ) | ||||
|             if content.tool_calls: | ||||
|                 messages[-1]["content"].extend(  # type: ignore[union-attr] | ||||
|                     [ | ||||
|                         ServerToolUseBlockParam( | ||||
|                             type="server_tool_use", | ||||
|                             id=tool_call.id, | ||||
|                             name="web_search", | ||||
|                             input=tool_call.tool_args, | ||||
|                         ) | ||||
|                         if tool_call.external and tool_call.tool_name == "web_search" | ||||
|                         else ToolUseBlockParam( | ||||
|                         ToolUseBlockParam( | ||||
|                             type="tool_use", | ||||
|                             id=tool_call.id, | ||||
|                             name=tool_call.tool_name, | ||||
| @@ -318,12 +173,10 @@ def _convert_content( | ||||
|     return messages | ||||
|  | ||||
|  | ||||
| async def _transform_stream(  # noqa: C901 - This is complex, but better to have it in one place | ||||
| async def _transform_stream( | ||||
|     chat_log: conversation.ChatLog, | ||||
|     stream: AsyncStream[MessageStreamEvent], | ||||
| ) -> AsyncGenerator[ | ||||
|     conversation.AssistantContentDeltaDict | conversation.ToolResultContentDeltaDict | ||||
| ]: | ||||
| ) -> AsyncGenerator[conversation.AssistantContentDeltaDict]: | ||||
|     """Transform the response stream into HA format. | ||||
|  | ||||
|     A typical stream of responses might look something like the following: | ||||
| @@ -356,13 +209,11 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|     if stream is None: | ||||
|         raise TypeError("Expected a stream of messages") | ||||
|  | ||||
|     current_tool_block: ToolUseBlockParam | ServerToolUseBlockParam | None = None | ||||
|     current_tool_block: ToolUseBlockParam | None = None | ||||
|     current_tool_args: str | ||||
|     content_details = ContentDetails() | ||||
|     content_details.add_citation_detail() | ||||
|     input_usage: Usage | None = None | ||||
|     has_content = False | ||||
|     has_native = False | ||||
|     first_block: bool | ||||
|  | ||||
|     async for response in stream: | ||||
|         LOGGER.debug("Received response: %s", response) | ||||
| @@ -371,7 +222,6 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|             if response.message.role != "assistant": | ||||
|                 raise ValueError("Unexpected message role") | ||||
|             input_usage = response.message.usage | ||||
|             first_block = True | ||||
|         elif isinstance(response, RawContentBlockStartEvent): | ||||
|             if isinstance(response.content_block, ToolUseBlock): | ||||
|                 current_tool_block = ToolUseBlockParam( | ||||
| @@ -382,37 +232,17 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|                 ) | ||||
|                 current_tool_args = "" | ||||
|             elif isinstance(response.content_block, TextBlock): | ||||
|                 if (  # Do not start a new assistant content just for citations, concatenate consecutive blocks with citations instead. | ||||
|                     first_block | ||||
|                     or ( | ||||
|                         not content_details.has_citations() | ||||
|                         and response.content_block.citations is None | ||||
|                         and content_details.has_content() | ||||
|                     ) | ||||
|                 ): | ||||
|                     if content_details.has_citations(): | ||||
|                         content_details.delete_empty() | ||||
|                         yield {"native": content_details} | ||||
|                     content_details = ContentDetails() | ||||
|                 if has_content: | ||||
|                     yield {"role": "assistant"} | ||||
|                     has_native = False | ||||
|                     first_block = False | ||||
|                 content_details.add_citation_detail() | ||||
|                 has_content = True | ||||
|                 if response.content_block.text: | ||||
|                     content_details.citation_details[-1].length += len( | ||||
|                         response.content_block.text | ||||
|                     ) | ||||
|                     yield {"content": response.content_block.text} | ||||
|             elif isinstance(response.content_block, ThinkingBlock): | ||||
|                 if first_block or has_native: | ||||
|                     if content_details.has_citations(): | ||||
|                         content_details.delete_empty() | ||||
|                         yield {"native": content_details} | ||||
|                     content_details = ContentDetails() | ||||
|                     content_details.add_citation_detail() | ||||
|                 if has_native: | ||||
|                     yield {"role": "assistant"} | ||||
|                     has_native = False | ||||
|                     first_block = False | ||||
|                     has_content = False | ||||
|             elif isinstance(response.content_block, RedactedThinkingBlock): | ||||
|                 LOGGER.debug( | ||||
|                     "Some of Claude’s internal reasoning has been automatically " | ||||
| @@ -420,60 +250,15 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|                     "responses" | ||||
|                 ) | ||||
|                 if has_native: | ||||
|                     if content_details.has_citations(): | ||||
|                         content_details.delete_empty() | ||||
|                         yield {"native": content_details} | ||||
|                     content_details = ContentDetails() | ||||
|                     content_details.add_citation_detail() | ||||
|                     yield {"role": "assistant"} | ||||
|                     has_native = False | ||||
|                     first_block = False | ||||
|                     has_content = False | ||||
|                 yield {"native": response.content_block} | ||||
|                 has_native = True | ||||
|             elif isinstance(response.content_block, ServerToolUseBlock): | ||||
|                 current_tool_block = ServerToolUseBlockParam( | ||||
|                     type="server_tool_use", | ||||
|                     id=response.content_block.id, | ||||
|                     name=response.content_block.name, | ||||
|                     input="", | ||||
|                 ) | ||||
|                 current_tool_args = "" | ||||
|             elif isinstance(response.content_block, WebSearchToolResultBlock): | ||||
|                 if content_details.has_citations(): | ||||
|                     content_details.delete_empty() | ||||
|                     yield {"native": content_details} | ||||
|                 content_details = ContentDetails() | ||||
|                 content_details.add_citation_detail() | ||||
|                 yield { | ||||
|                     "role": "tool_result", | ||||
|                     "tool_call_id": response.content_block.tool_use_id, | ||||
|                     "tool_name": "web_search", | ||||
|                     "tool_result": { | ||||
|                         "type": "web_search_tool_result_error", | ||||
|                         "error_code": response.content_block.content.error_code, | ||||
|                     } | ||||
|                     if isinstance( | ||||
|                         response.content_block.content, WebSearchToolResultError | ||||
|                     ) | ||||
|                     else { | ||||
|                         "content": [ | ||||
|                             { | ||||
|                                 "type": "web_search_result", | ||||
|                                 "encrypted_content": block.encrypted_content, | ||||
|                                 "page_age": block.page_age, | ||||
|                                 "title": block.title, | ||||
|                                 "url": block.url, | ||||
|                             } | ||||
|                             for block in response.content_block.content | ||||
|                         ] | ||||
|                     }, | ||||
|                 } | ||||
|                 first_block = True | ||||
|         elif isinstance(response, RawContentBlockDeltaEvent): | ||||
|             if isinstance(response.delta, InputJSONDelta): | ||||
|                 current_tool_args += response.delta.partial_json | ||||
|             elif isinstance(response.delta, TextDelta): | ||||
|                 content_details.citation_details[-1].length += len(response.delta.text) | ||||
|                 yield {"content": response.delta.text} | ||||
|             elif isinstance(response.delta, ThinkingDelta): | ||||
|                 yield {"thinking_content": response.delta.thinking} | ||||
| @@ -486,8 +271,6 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|                     ) | ||||
|                 } | ||||
|                 has_native = True | ||||
|             elif isinstance(response.delta, CitationsDelta): | ||||
|                 content_details.add_citation(response.delta.citation) | ||||
|         elif isinstance(response, RawContentBlockStopEvent): | ||||
|             if current_tool_block is not None: | ||||
|                 tool_args = json.loads(current_tool_args) if current_tool_args else {} | ||||
| @@ -498,7 +281,6 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|                             id=current_tool_block["id"], | ||||
|                             tool_name=current_tool_block["name"], | ||||
|                             tool_args=tool_args, | ||||
|                             external=current_tool_block["type"] == "server_tool_use", | ||||
|                         ) | ||||
|                     ] | ||||
|                 } | ||||
| @@ -508,12 +290,6 @@ async def _transform_stream(  # noqa: C901 - This is complex, but better to have | ||||
|                 chat_log.async_trace(_create_token_stats(input_usage, usage)) | ||||
|             if response.delta.stop_reason == "refusal": | ||||
|                 raise HomeAssistantError("Potential policy violation detected") | ||||
|         elif isinstance(response, RawMessageStopEvent): | ||||
|             if content_details.has_citations(): | ||||
|                 content_details.delete_empty() | ||||
|                 yield {"native": content_details} | ||||
|             content_details = ContentDetails() | ||||
|             content_details.add_citation_detail() | ||||
|  | ||||
|  | ||||
| def _create_token_stats( | ||||
| @@ -561,11 +337,21 @@ class AnthropicBaseLLMEntity(Entity): | ||||
|         """Generate an answer for the chat log.""" | ||||
|         options = self.subentry.data | ||||
|  | ||||
|         tools: list[ToolParam] | None = None | ||||
|         if chat_log.llm_api: | ||||
|             tools = [ | ||||
|                 _format_tool(tool, chat_log.llm_api.custom_serializer) | ||||
|                 for tool in chat_log.llm_api.tools | ||||
|             ] | ||||
|  | ||||
|         system = chat_log.content[0] | ||||
|         if not isinstance(system, conversation.SystemContent): | ||||
|             raise TypeError("First message must be a system message") | ||||
|         messages = _convert_content(chat_log.content[1:]) | ||||
|  | ||||
|         client = self.entry.runtime_data | ||||
|  | ||||
|         thinking_budget = options.get(CONF_THINKING_BUDGET, RECOMMENDED_THINKING_BUDGET) | ||||
|         model = options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL) | ||||
|  | ||||
|         model_args = MessageCreateParamsStreaming( | ||||
| @@ -575,10 +361,10 @@ class AnthropicBaseLLMEntity(Entity): | ||||
|             system=system.content, | ||||
|             stream=True, | ||||
|         ) | ||||
|  | ||||
|         thinking_budget = options.get(CONF_THINKING_BUDGET, RECOMMENDED_THINKING_BUDGET) | ||||
|         if tools: | ||||
|             model_args["tools"] = tools | ||||
|         if ( | ||||
|             not model.startswith(tuple(NON_THINKING_MODELS)) | ||||
|             model.startswith(tuple(THINKING_MODELS)) | ||||
|             and thinking_budget >= MIN_THINKING_BUDGET | ||||
|         ): | ||||
|             model_args["thinking"] = ThinkingConfigEnabledParam( | ||||
| @@ -590,34 +376,6 @@ class AnthropicBaseLLMEntity(Entity): | ||||
|                 CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE | ||||
|             ) | ||||
|  | ||||
|         tools: list[ToolUnionParam] = [] | ||||
|         if chat_log.llm_api: | ||||
|             tools = [ | ||||
|                 _format_tool(tool, chat_log.llm_api.custom_serializer) | ||||
|                 for tool in chat_log.llm_api.tools | ||||
|             ] | ||||
|  | ||||
|         if options.get(CONF_WEB_SEARCH): | ||||
|             web_search = WebSearchTool20250305Param( | ||||
|                 name="web_search", | ||||
|                 type="web_search_20250305", | ||||
|                 max_uses=options.get(CONF_WEB_SEARCH_MAX_USES), | ||||
|             ) | ||||
|             if options.get(CONF_WEB_SEARCH_USER_LOCATION): | ||||
|                 web_search["user_location"] = { | ||||
|                     "type": "approximate", | ||||
|                     "city": options.get(CONF_WEB_SEARCH_CITY, ""), | ||||
|                     "region": options.get(CONF_WEB_SEARCH_REGION, ""), | ||||
|                     "country": options.get(CONF_WEB_SEARCH_COUNTRY, ""), | ||||
|                     "timezone": options.get(CONF_WEB_SEARCH_TIMEZONE, ""), | ||||
|                 } | ||||
|             tools.append(web_search) | ||||
|  | ||||
|         if tools: | ||||
|             model_args["tools"] = tools | ||||
|  | ||||
|         client = self.entry.runtime_data | ||||
|  | ||||
|         # To prevent infinite loops, we limit the number of iterations | ||||
|         for _iteration in range(MAX_TOOL_ITERATIONS): | ||||
|             try: | ||||
|   | ||||
| @@ -8,5 +8,5 @@ | ||||
|   "documentation": "https://www.home-assistant.io/integrations/anthropic", | ||||
|   "integration_type": "service", | ||||
|   "iot_class": "cloud_polling", | ||||
|   "requirements": ["anthropic==0.69.0"] | ||||
|   "requirements": ["anthropic==0.62.0"] | ||||
| } | ||||
|   | ||||
| @@ -35,17 +35,11 @@ | ||||
|             "temperature": "Temperature", | ||||
|             "llm_hass_api": "[%key:common::config_flow::data::llm_hass_api%]", | ||||
|             "recommended": "Recommended model settings", | ||||
|             "thinking_budget": "Thinking budget", | ||||
|             "web_search": "Enable web search", | ||||
|             "web_search_max_uses": "Maximum web searches", | ||||
|             "user_location": "Include home location" | ||||
|             "thinking_budget_tokens": "Thinking budget" | ||||
|           }, | ||||
|           "data_description": { | ||||
|             "prompt": "Instruct how the LLM should respond. This can be a template.", | ||||
|             "thinking_budget": "The number of tokens the model can use to think about the response out of the total maximum number of tokens. Set to 1024 or greater to enable extended thinking.", | ||||
|             "web_search": "The web search tool gives Claude direct access to real-time web content, allowing it to answer questions with up-to-date information beyond its knowledge cutoff", | ||||
|             "web_search_max_uses": "Limit the number of searches performed per response", | ||||
|             "user_location": "Localize search results based on home location" | ||||
|             "thinking_budget_tokens": "The number of tokens the model can use to think about the response out of the total maximum number of tokens. Set to 1024 or greater to enable extended thinking." | ||||
|           } | ||||
|         } | ||||
|       }, | ||||
| @@ -54,8 +48,7 @@ | ||||
|         "entry_not_loaded": "Cannot add things while the configuration is disabled." | ||||
|       }, | ||||
|       "error": { | ||||
|         "thinking_budget_too_large": "Maximum tokens must be greater than the thinking budget.", | ||||
|         "web_search_unsupported_model": "Web search is not supported by the selected model. Please choose a compatible model or disable web search." | ||||
|         "thinking_budget_too_large": "Maximum tokens must be greater than the thinking budget." | ||||
|       } | ||||
|     } | ||||
|   } | ||||
|   | ||||
| @@ -5,9 +5,14 @@ from __future__ import annotations | ||||
| import asyncio | ||||
| import logging | ||||
| from random import randrange | ||||
| import sys | ||||
| from typing import Any, cast | ||||
|  | ||||
| from pyatv import connect, exceptions, scan | ||||
| from pyatv.conf import AppleTV | ||||
| from pyatv.const import DeviceModel, Protocol | ||||
| from pyatv.convert import model_str | ||||
| from pyatv.interface import AppleTV as AppleTVInterface, DeviceListener | ||||
|  | ||||
| from homeassistant.components import zeroconf | ||||
| from homeassistant.config_entries import ConfigEntry | ||||
| from homeassistant.const import ( | ||||
| @@ -24,11 +29,7 @@ from homeassistant.const import ( | ||||
|     Platform, | ||||
| ) | ||||
| from homeassistant.core import Event, HomeAssistant, callback | ||||
| from homeassistant.exceptions import ( | ||||
|     ConfigEntryAuthFailed, | ||||
|     ConfigEntryNotReady, | ||||
|     HomeAssistantError, | ||||
| ) | ||||
| from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady | ||||
| from homeassistant.helpers import device_registry as dr | ||||
| from homeassistant.helpers.aiohttp_client import async_get_clientsession | ||||
| from homeassistant.helpers.dispatcher import async_dispatcher_send | ||||
| @@ -42,18 +43,6 @@ from .const import ( | ||||
|     SIGNAL_DISCONNECTED, | ||||
| ) | ||||
|  | ||||
| if sys.version_info < (3, 14): | ||||
|     from pyatv import connect, exceptions, scan | ||||
|     from pyatv.conf import AppleTV | ||||
|     from pyatv.const import DeviceModel, Protocol | ||||
|     from pyatv.convert import model_str | ||||
|     from pyatv.interface import AppleTV as AppleTVInterface, DeviceListener | ||||
| else: | ||||
|  | ||||
|     class DeviceListener: | ||||
|         """Dummy class.""" | ||||
|  | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
| DEFAULT_NAME_TV = "Apple TV" | ||||
| @@ -64,41 +53,31 @@ BACKOFF_TIME_UPPER_LIMIT = 300  # Five minutes | ||||
|  | ||||
| PLATFORMS = [Platform.MEDIA_PLAYER, Platform.REMOTE] | ||||
|  | ||||
| if sys.version_info < (3, 14): | ||||
|     AUTH_EXCEPTIONS = ( | ||||
| AUTH_EXCEPTIONS = ( | ||||
|     exceptions.AuthenticationError, | ||||
|     exceptions.InvalidCredentialsError, | ||||
|     exceptions.NoCredentialsError, | ||||
|     ) | ||||
|     CONNECTION_TIMEOUT_EXCEPTIONS = ( | ||||
| ) | ||||
| CONNECTION_TIMEOUT_EXCEPTIONS = ( | ||||
|     OSError, | ||||
|     asyncio.CancelledError, | ||||
|     TimeoutError, | ||||
|     exceptions.ConnectionLostError, | ||||
|     exceptions.ConnectionFailedError, | ||||
|     ) | ||||
|     DEVICE_EXCEPTIONS = ( | ||||
| ) | ||||
| DEVICE_EXCEPTIONS = ( | ||||
|     exceptions.ProtocolError, | ||||
|     exceptions.NoServiceError, | ||||
|     exceptions.PairingError, | ||||
|     exceptions.BackOffError, | ||||
|     exceptions.DeviceIdMissingError, | ||||
|     ) | ||||
| else: | ||||
|     AUTH_EXCEPTIONS = () | ||||
|     CONNECTION_TIMEOUT_EXCEPTIONS = () | ||||
|     DEVICE_EXCEPTIONS = () | ||||
|  | ||||
| ) | ||||
|  | ||||
| type AppleTvConfigEntry = ConfigEntry[AppleTVManager] | ||||
|  | ||||
|  | ||||
| async def async_setup_entry(hass: HomeAssistant, entry: AppleTvConfigEntry) -> bool: | ||||
|     """Set up a config entry for Apple TV.""" | ||||
|     if sys.version_info >= (3, 14): | ||||
|         raise HomeAssistantError( | ||||
|             "Apple TV is not supported on Python 3.14. Please use Python 3.13." | ||||
|         ) | ||||
|     manager = AppleTVManager(hass, entry) | ||||
|  | ||||
|     if manager.is_on: | ||||
|   | ||||
| @@ -7,7 +7,7 @@ | ||||
|   "documentation": "https://www.home-assistant.io/integrations/apple_tv", | ||||
|   "iot_class": "local_push", | ||||
|   "loggers": ["pyatv", "srptools"], | ||||
|   "requirements": ["pyatv==0.16.1;python_version<'3.14'"], | ||||
|   "requirements": ["pyatv==0.16.1"], | ||||
|   "zeroconf": [ | ||||
|     "_mediaremotetv._tcp.local.", | ||||
|     "_companion-link._tcp.local.", | ||||
|   | ||||
| @@ -7,8 +7,6 @@ from typing import Any | ||||
| from pyaprilaire.const import Attribute | ||||
|  | ||||
| from homeassistant.components.climate import ( | ||||
|     ATTR_TARGET_TEMP_HIGH, | ||||
|     ATTR_TARGET_TEMP_LOW, | ||||
|     FAN_AUTO, | ||||
|     FAN_ON, | ||||
|     PRESET_AWAY, | ||||
| @@ -18,12 +16,7 @@ from homeassistant.components.climate import ( | ||||
|     HVACAction, | ||||
|     HVACMode, | ||||
| ) | ||||
| from homeassistant.const import ( | ||||
|     ATTR_TEMPERATURE, | ||||
|     PRECISION_HALVES, | ||||
|     PRECISION_WHOLE, | ||||
|     UnitOfTemperature, | ||||
| ) | ||||
| from homeassistant.const import PRECISION_HALVES, PRECISION_WHOLE, UnitOfTemperature | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
|  | ||||
| @@ -239,15 +232,15 @@ class AprilaireClimate(BaseAprilaireEntity, ClimateEntity): | ||||
|         cool_setpoint = 0 | ||||
|         heat_setpoint = 0 | ||||
|  | ||||
|         if temperature := kwargs.get(ATTR_TEMPERATURE): | ||||
|         if temperature := kwargs.get("temperature"): | ||||
|             if self.coordinator.data.get(Attribute.MODE) == 3: | ||||
|                 cool_setpoint = temperature | ||||
|             else: | ||||
|                 heat_setpoint = temperature | ||||
|         else: | ||||
|             if target_temp_low := kwargs.get(ATTR_TARGET_TEMP_LOW): | ||||
|             if target_temp_low := kwargs.get("target_temp_low"): | ||||
|                 heat_setpoint = target_temp_low | ||||
|             if target_temp_high := kwargs.get(ATTR_TARGET_TEMP_HIGH): | ||||
|             if target_temp_high := kwargs.get("target_temp_high"): | ||||
|                 cool_setpoint = target_temp_high | ||||
|  | ||||
|         if cool_setpoint == 0 and heat_setpoint == 0: | ||||
|   | ||||
| @@ -3,17 +3,17 @@ | ||||
| from __future__ import annotations | ||||
|  | ||||
| from abc import ABC, abstractmethod | ||||
| from collections import namedtuple | ||||
| from collections.abc import Awaitable, Callable, Coroutine | ||||
| import functools | ||||
| import logging | ||||
| from typing import Any, NamedTuple | ||||
| from typing import Any, cast | ||||
|  | ||||
| from aioasuswrt.asuswrt import AsusWrt as AsusWrtLegacy | ||||
| from aiohttp import ClientSession | ||||
| from asusrouter import AsusRouter, AsusRouterError | ||||
| from asusrouter.config import ARConfigKey | ||||
| from asusrouter.modules.client import AsusClient | ||||
| from asusrouter.modules.connection import ConnectionState | ||||
| from asusrouter.modules.data import AsusData | ||||
| from asusrouter.modules.homeassistant import convert_to_ha_data, convert_to_ha_sensors | ||||
| from asusrouter.tools.connection import get_cookie_jar | ||||
| @@ -61,14 +61,7 @@ SENSORS_TYPE_RATES = "sensors_rates" | ||||
| SENSORS_TYPE_TEMPERATURES = "sensors_temperatures" | ||||
| SENSORS_TYPE_UPTIME = "sensors_uptime" | ||||
|  | ||||
|  | ||||
| class WrtDevice(NamedTuple): | ||||
|     """WrtDevice structure.""" | ||||
|  | ||||
|     ip: str | None | ||||
|     name: str | None | ||||
|     conneted_to: str | None | ||||
|  | ||||
| WrtDevice = namedtuple("WrtDevice", ["ip", "name", "connected_to"])  # noqa: PYI024 | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
| @@ -87,7 +80,7 @@ def handle_errors_and_zip[_AsusWrtBridgeT: AsusWrtBridge]( | ||||
|         """Run library methods and zip results or manage exceptions.""" | ||||
|  | ||||
|         @functools.wraps(func) | ||||
|         async def _wrapper(self: _AsusWrtBridgeT) -> dict[str, str]: | ||||
|         async def _wrapper(self: _AsusWrtBridgeT) -> dict[str, Any]: | ||||
|             try: | ||||
|                 data = await func(self) | ||||
|             except exceptions as exc: | ||||
| @@ -226,7 +219,7 @@ class AsusWrtLegacyBridge(AsusWrtBridge): | ||||
|     @property | ||||
|     def is_connected(self) -> bool: | ||||
|         """Get connected status.""" | ||||
|         return self._api.is_connected | ||||
|         return cast(bool, self._api.is_connected) | ||||
|  | ||||
|     async def async_connect(self) -> None: | ||||
|         """Connect to the device.""" | ||||
| @@ -242,7 +235,8 @@ class AsusWrtLegacyBridge(AsusWrtBridge): | ||||
|  | ||||
|     async def async_disconnect(self) -> None: | ||||
|         """Disconnect to the device.""" | ||||
|         await self._api.async_disconnect() | ||||
|         if self._api is not None and self._protocol == PROTOCOL_TELNET: | ||||
|             self._api.connection.disconnect() | ||||
|  | ||||
|     async def async_get_connected_devices(self) -> dict[str, WrtDevice]: | ||||
|         """Get list of connected devices.""" | ||||
| @@ -443,7 +437,6 @@ class AsusWrtHttpBridge(AsusWrtBridge): | ||||
|             if dev.connection is not None | ||||
|             and dev.description is not None | ||||
|             and dev.connection.ip_address is not None | ||||
|             and dev.state is ConnectionState.CONNECTED | ||||
|         } | ||||
|  | ||||
|     async def async_get_available_sensors(self) -> dict[str, dict[str, Any]]: | ||||
|   | ||||
| @@ -10,6 +10,8 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
| from . import AsusWrtConfigEntry | ||||
| from .router import AsusWrtDevInfo, AsusWrtRouter | ||||
|  | ||||
| ATTR_LAST_TIME_REACHABLE = "last_time_reachable" | ||||
|  | ||||
| DEFAULT_DEVICE_NAME = "Unknown device" | ||||
|  | ||||
|  | ||||
| @@ -56,6 +58,8 @@ def add_entities( | ||||
| class AsusWrtDevice(ScannerEntity): | ||||
|     """Representation of a AsusWrt device.""" | ||||
|  | ||||
|     _unrecorded_attributes = frozenset({ATTR_LAST_TIME_REACHABLE}) | ||||
|  | ||||
|     _attr_should_poll = False | ||||
|  | ||||
|     def __init__(self, router: AsusWrtRouter, device: AsusWrtDevInfo) -> None: | ||||
| @@ -93,6 +97,11 @@ class AsusWrtDevice(ScannerEntity): | ||||
|     def async_on_demand_update(self) -> None: | ||||
|         """Update state.""" | ||||
|         self._device = self._router.devices[self._device.mac] | ||||
|         self._attr_extra_state_attributes = {} | ||||
|         if self._device.last_activity: | ||||
|             self._attr_extra_state_attributes[ATTR_LAST_TIME_REACHABLE] = ( | ||||
|                 self._device.last_activity.isoformat(timespec="seconds") | ||||
|             ) | ||||
|         self.async_write_ha_state() | ||||
|  | ||||
|     async def async_added_to_hass(self) -> None: | ||||
|   | ||||
| @@ -7,5 +7,5 @@ | ||||
|   "integration_type": "hub", | ||||
|   "iot_class": "local_polling", | ||||
|   "loggers": ["aioasuswrt", "asusrouter", "asyncssh"], | ||||
|   "requirements": ["aioasuswrt==1.5.1", "asusrouter==1.21.0"] | ||||
|   "requirements": ["aioasuswrt==1.4.0", "asusrouter==1.21.0"] | ||||
| } | ||||
|   | ||||
| @@ -36,14 +36,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: AugustConfigEntry) -> bo | ||||
|         raise ConfigEntryAuthFailed("Migration to OAuth required") | ||||
|  | ||||
|     session = async_create_august_clientsession(hass) | ||||
|     try: | ||||
|     implementation = ( | ||||
|         await config_entry_oauth2_flow.async_get_config_entry_implementation( | ||||
|             hass, entry | ||||
|         ) | ||||
|     ) | ||||
|     except ValueError as err: | ||||
|         raise ConfigEntryNotReady("OAuth implementation not available") from err | ||||
|     oauth_session = config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation) | ||||
|     august_gateway = AugustGateway(Path(hass.config.config_dir), session, oauth_session) | ||||
|     try: | ||||
|   | ||||
| @@ -5,5 +5,5 @@ | ||||
|   "config_flow": true, | ||||
|   "documentation": "https://www.home-assistant.io/integrations/autarco", | ||||
|   "iot_class": "cloud_polling", | ||||
|   "requirements": ["autarco==3.2.0"] | ||||
|   "requirements": ["autarco==3.1.0"] | ||||
| } | ||||
|   | ||||
| @@ -136,8 +136,8 @@ class WellKnownOAuthInfoView(HomeAssistantView): | ||||
|             url_prefix = get_url(hass, require_current_request=True) | ||||
|         except NoURLAvailableError: | ||||
|             url_prefix = "" | ||||
|  | ||||
|         metadata = { | ||||
|         return self.json( | ||||
|             { | ||||
|                 "authorization_endpoint": f"{url_prefix}/auth/authorize", | ||||
|                 "token_endpoint": f"{url_prefix}/auth/token", | ||||
|                 "revocation_endpoint": f"{url_prefix}/auth/revoke", | ||||
| @@ -146,12 +146,7 @@ class WellKnownOAuthInfoView(HomeAssistantView): | ||||
|                     "https://developers.home-assistant.io/docs/auth_api" | ||||
|                 ), | ||||
|             } | ||||
|  | ||||
|         # Add issuer only when we have a valid base URL (RFC 8414 compliance) | ||||
|         if url_prefix: | ||||
|             metadata["issuer"] = url_prefix | ||||
|  | ||||
|         return self.json(metadata) | ||||
|         ) | ||||
|  | ||||
|  | ||||
| class AuthProvidersView(HomeAssistantView): | ||||
|   | ||||
| @@ -57,7 +57,6 @@ from .api import ( | ||||
|     _get_manager, | ||||
|     async_address_present, | ||||
|     async_ble_device_from_address, | ||||
|     async_clear_address_from_match_history, | ||||
|     async_current_scanners, | ||||
|     async_discovered_service_info, | ||||
|     async_get_advertisement_callback, | ||||
| @@ -116,7 +115,6 @@ __all__ = [ | ||||
|     "HomeAssistantRemoteScanner", | ||||
|     "async_address_present", | ||||
|     "async_ble_device_from_address", | ||||
|     "async_clear_address_from_match_history", | ||||
|     "async_current_scanners", | ||||
|     "async_discovered_service_info", | ||||
|     "async_get_advertisement_callback", | ||||
|   | ||||
| @@ -193,20 +193,6 @@ def async_rediscover_address(hass: HomeAssistant, address: str) -> None: | ||||
|     _get_manager(hass).async_rediscover_address(address) | ||||
|  | ||||
|  | ||||
| @hass_callback | ||||
| def async_clear_address_from_match_history(hass: HomeAssistant, address: str) -> None: | ||||
|     """Clear an address from the integration matcher history. | ||||
|  | ||||
|     This allows future advertisements from this address to trigger discovery | ||||
|     even if the advertisement content has changed but the service data UUIDs | ||||
|     remain the same. | ||||
|  | ||||
|     Unlike async_rediscover_address, this does not immediately re-trigger | ||||
|     discovery with the current advertisement in history. | ||||
|     """ | ||||
|     _get_manager(hass).async_clear_address_from_match_history(address) | ||||
|  | ||||
|  | ||||
| @hass_callback | ||||
| def async_register_scanner( | ||||
|     hass: HomeAssistant, | ||||
|   | ||||
| @@ -120,19 +120,6 @@ class HomeAssistantBluetoothManager(BluetoothManager): | ||||
|         if service_info := self._all_history.get(address): | ||||
|             self._async_trigger_matching_discovery(service_info) | ||||
|  | ||||
|     @hass_callback | ||||
|     def async_clear_address_from_match_history(self, address: str) -> None: | ||||
|         """Clear an address from the integration matcher history. | ||||
|  | ||||
|         This allows future advertisements from this address to trigger discovery | ||||
|         even if the advertisement content has changed but the service data UUIDs | ||||
|         remain the same. | ||||
|  | ||||
|         Unlike async_rediscover_address, this does not immediately re-trigger | ||||
|         discovery with the current advertisement in history. | ||||
|         """ | ||||
|         self._integration_matcher.async_clear_address(address) | ||||
|  | ||||
|     def _discover_service_info(self, service_info: BluetoothServiceInfoBleak) -> None: | ||||
|         matched_domains = self._integration_matcher.match_domains(service_info) | ||||
|         if self._debug: | ||||
|   | ||||
| @@ -19,8 +19,8 @@ | ||||
|     "bleak-retry-connector==4.4.3", | ||||
|     "bluetooth-adapters==2.1.0", | ||||
|     "bluetooth-auto-recovery==1.5.3", | ||||
|     "bluetooth-data-tools==1.28.3", | ||||
|     "dbus-fast==2.44.5", | ||||
|     "habluetooth==5.7.0" | ||||
|     "bluetooth-data-tools==1.28.2", | ||||
|     "dbus-fast==2.44.3", | ||||
|     "habluetooth==5.6.4" | ||||
|   ] | ||||
| } | ||||
|   | ||||
| @@ -68,17 +68,12 @@ class IntegrationMatchHistory: | ||||
|     manufacturer_data: bool | ||||
|     service_data: set[str] | ||||
|     service_uuids: set[str] | ||||
|     name: str | ||||
|  | ||||
|  | ||||
| def seen_all_fields( | ||||
|     previous_match: IntegrationMatchHistory, | ||||
|     advertisement_data: AdvertisementData, | ||||
|     name: str, | ||||
|     previous_match: IntegrationMatchHistory, advertisement_data: AdvertisementData | ||||
| ) -> bool: | ||||
|     """Return if we have seen all fields.""" | ||||
|     if previous_match.name != name: | ||||
|         return False | ||||
|     if not previous_match.manufacturer_data and advertisement_data.manufacturer_data: | ||||
|         return False | ||||
|     if advertisement_data.service_data and ( | ||||
| @@ -127,11 +122,10 @@ class IntegrationMatcher: | ||||
|         device = service_info.device | ||||
|         advertisement_data = service_info.advertisement | ||||
|         connectable = service_info.connectable | ||||
|         name = service_info.name | ||||
|         matched = self._matched_connectable if connectable else self._matched | ||||
|         matched_domains: set[str] = set() | ||||
|         if (previous_match := matched.get(device.address)) and seen_all_fields( | ||||
|             previous_match, advertisement_data, name | ||||
|             previous_match, advertisement_data | ||||
|         ): | ||||
|             # We have seen all fields so we can skip the rest of the matchers | ||||
|             return matched_domains | ||||
| @@ -146,13 +140,11 @@ class IntegrationMatcher: | ||||
|             ) | ||||
|             previous_match.service_data |= set(advertisement_data.service_data) | ||||
|             previous_match.service_uuids |= set(advertisement_data.service_uuids) | ||||
|             previous_match.name = name | ||||
|         else: | ||||
|             matched[device.address] = IntegrationMatchHistory( | ||||
|                 manufacturer_data=bool(advertisement_data.manufacturer_data), | ||||
|                 service_data=set(advertisement_data.service_data), | ||||
|                 service_uuids=set(advertisement_data.service_uuids), | ||||
|                 name=name, | ||||
|             ) | ||||
|         return matched_domains | ||||
|  | ||||
|   | ||||
| @@ -8,7 +8,7 @@ | ||||
|   "integration_type": "device", | ||||
|   "iot_class": "local_polling", | ||||
|   "loggers": ["brother", "pyasn1", "pysmi", "pysnmp"], | ||||
|   "requirements": ["brother==5.1.1"], | ||||
|   "requirements": ["brother==5.1.0"], | ||||
|   "zeroconf": [ | ||||
|     { | ||||
|       "type": "_printer._tcp.local.", | ||||
|   | ||||
| @@ -7,14 +7,12 @@ from typing import Any | ||||
| from evolutionhttp import BryantEvolutionLocalClient | ||||
|  | ||||
| from homeassistant.components.climate import ( | ||||
|     ATTR_TARGET_TEMP_HIGH, | ||||
|     ATTR_TARGET_TEMP_LOW, | ||||
|     ClimateEntity, | ||||
|     ClimateEntityFeature, | ||||
|     HVACAction, | ||||
|     HVACMode, | ||||
| ) | ||||
| from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature | ||||
| from homeassistant.const import UnitOfTemperature | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.exceptions import HomeAssistantError | ||||
| from homeassistant.helpers.device_registry import DeviceInfo | ||||
| @@ -210,24 +208,24 @@ class BryantEvolutionClimate(ClimateEntity): | ||||
|  | ||||
|     async def async_set_temperature(self, **kwargs: Any) -> None: | ||||
|         """Set new target temperature.""" | ||||
|         if value := kwargs.get(ATTR_TARGET_TEMP_HIGH): | ||||
|             temp = int(value) | ||||
|         if kwargs.get("target_temp_high"): | ||||
|             temp = int(kwargs["target_temp_high"]) | ||||
|             if not await self._client.set_cooling_setpoint(temp): | ||||
|                 raise HomeAssistantError( | ||||
|                     translation_domain=DOMAIN, translation_key="failed_to_set_clsp" | ||||
|                 ) | ||||
|             self._attr_target_temperature_high = temp | ||||
|  | ||||
|         if value := kwargs.get(ATTR_TARGET_TEMP_LOW): | ||||
|             temp = int(value) | ||||
|         if kwargs.get("target_temp_low"): | ||||
|             temp = int(kwargs["target_temp_low"]) | ||||
|             if not await self._client.set_heating_setpoint(temp): | ||||
|                 raise HomeAssistantError( | ||||
|                     translation_domain=DOMAIN, translation_key="failed_to_set_htsp" | ||||
|                 ) | ||||
|             self._attr_target_temperature_low = temp | ||||
|  | ||||
|         if value := kwargs.get(ATTR_TEMPERATURE): | ||||
|             temp = int(value) | ||||
|         if kwargs.get("temperature"): | ||||
|             temp = int(kwargs["temperature"]) | ||||
|             fn = ( | ||||
|                 self._client.set_heating_setpoint | ||||
|                 if self.hvac_mode == HVACMode.HEAT | ||||
|   | ||||
| @@ -3,20 +3,15 @@ | ||||
| from __future__ import annotations | ||||
|  | ||||
| from datetime import datetime | ||||
| from functools import partial | ||||
| import logging | ||||
| from typing import Any | ||||
|  | ||||
| import caldav | ||||
| from caldav.lib.error import DAVError | ||||
| import requests | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.components.calendar import ( | ||||
|     ENTITY_ID_FORMAT, | ||||
|     PLATFORM_SCHEMA as CALENDAR_PLATFORM_SCHEMA, | ||||
|     CalendarEntity, | ||||
|     CalendarEntityFeature, | ||||
|     CalendarEvent, | ||||
|     is_offset_reached, | ||||
| ) | ||||
| @@ -28,7 +23,6 @@ from homeassistant.const import ( | ||||
|     CONF_VERIFY_SSL, | ||||
| ) | ||||
| from homeassistant.core import HomeAssistant, callback | ||||
| from homeassistant.exceptions import HomeAssistantError | ||||
| from homeassistant.helpers import config_validation as cv | ||||
| from homeassistant.helpers.entity import async_generate_entity_id | ||||
| from homeassistant.helpers.entity_platform import ( | ||||
| @@ -181,8 +175,6 @@ async def async_setup_entry( | ||||
| class WebDavCalendarEntity(CoordinatorEntity[CalDavUpdateCoordinator], CalendarEntity): | ||||
|     """A device for getting the next Task from a WebDav Calendar.""" | ||||
|  | ||||
|     _attr_supported_features = CalendarEntityFeature.CREATE_EVENT | ||||
|  | ||||
|     def __init__( | ||||
|         self, | ||||
|         name: str | None, | ||||
| @@ -211,31 +203,6 @@ class WebDavCalendarEntity(CoordinatorEntity[CalDavUpdateCoordinator], CalendarE | ||||
|         """Get all events in a specific time frame.""" | ||||
|         return await self.coordinator.async_get_events(hass, start_date, end_date) | ||||
|  | ||||
|     async def async_create_event(self, **kwargs: Any) -> None: | ||||
|         """Create a new event in the calendar.""" | ||||
|         _LOGGER.debug("Event: %s", kwargs) | ||||
|  | ||||
|         item_data: dict[str, Any] = { | ||||
|             "summary": kwargs["summary"], | ||||
|             "dtstart": kwargs["dtstart"], | ||||
|             "dtend": kwargs["dtend"], | ||||
|         } | ||||
|         if description := kwargs.get("description"): | ||||
|             item_data["description"] = description | ||||
|         if location := kwargs.get("location"): | ||||
|             item_data["location"] = location | ||||
|         if rrule := kwargs.get("rrule"): | ||||
|             item_data["rrule"] = rrule | ||||
|  | ||||
|         _LOGGER.debug("ICS data %s", item_data) | ||||
|  | ||||
|         try: | ||||
|             await self.hass.async_add_executor_job( | ||||
|                 partial(self.coordinator.calendar.add_event, **item_data), | ||||
|             ) | ||||
|         except (requests.ConnectionError, DAVError) as err: | ||||
|             raise HomeAssistantError(f"CalDAV save error: {err}") from err | ||||
|  | ||||
|     @callback | ||||
|     def _handle_coordinator_update(self) -> None: | ||||
|         """Update event data.""" | ||||
|   | ||||
| @@ -169,7 +169,7 @@ class CalendarEventListener: | ||||
|     def __init__( | ||||
|         self, | ||||
|         hass: HomeAssistant, | ||||
|         job: HassJob[..., Coroutine[Any, Any, None] | Any], | ||||
|         job: HassJob[..., Coroutine[Any, Any, None]], | ||||
|         trigger_data: dict[str, Any], | ||||
|         fetcher: QueuedEventFetcher, | ||||
|     ) -> None: | ||||
|   | ||||
| @@ -31,7 +31,7 @@ async def async_setup_entry( | ||||
|         for location_id, location in coordinator.data["locations"].items() | ||||
|     ] | ||||
|  | ||||
|     async_add_entities(alarms) | ||||
|     async_add_entities(alarms, True) | ||||
|  | ||||
|  | ||||
| class CanaryAlarm( | ||||
|   | ||||
| @@ -68,7 +68,8 @@ async def async_setup_entry( | ||||
|             for location_id, location in coordinator.data["locations"].items() | ||||
|             for device in location.devices | ||||
|             if device.is_online | ||||
|         ) | ||||
|         ), | ||||
|         True, | ||||
|     ) | ||||
|  | ||||
|  | ||||
|   | ||||
| @@ -80,7 +80,7 @@ async def async_setup_entry( | ||||
|                     if device_type.get("name") in sensor_type[4] | ||||
|                 ) | ||||
|  | ||||
|     async_add_entities(sensors) | ||||
|     async_add_entities(sensors, True) | ||||
|  | ||||
|  | ||||
| class CanarySensor(CoordinatorEntity[CanaryDataUpdateCoordinator], SensorEntity): | ||||
|   | ||||
| @@ -4,6 +4,5 @@ | ||||
|   "codeowners": [], | ||||
|   "documentation": "https://www.home-assistant.io/integrations/citybikes", | ||||
|   "iot_class": "cloud_polling", | ||||
|   "quality_scale": "legacy", | ||||
|   "requirements": ["python-citybikes==0.3.3"] | ||||
|   "quality_scale": "legacy" | ||||
| } | ||||
|   | ||||
| @@ -5,11 +5,8 @@ from __future__ import annotations | ||||
| import asyncio | ||||
| from datetime import timedelta | ||||
| import logging | ||||
| import sys | ||||
|  | ||||
| import aiohttp | ||||
| from citybikes import __version__ as CITYBIKES_CLIENT_VERSION | ||||
| from citybikes.asyncio import Client as CitybikesClient | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.components.sensor import ( | ||||
| @@ -18,18 +15,21 @@ from homeassistant.components.sensor import ( | ||||
|     SensorEntity, | ||||
| ) | ||||
| from homeassistant.const import ( | ||||
|     APPLICATION_NAME, | ||||
|     ATTR_ID, | ||||
|     ATTR_LATITUDE, | ||||
|     ATTR_LOCATION, | ||||
|     ATTR_LONGITUDE, | ||||
|     ATTR_NAME, | ||||
|     CONF_LATITUDE, | ||||
|     CONF_LONGITUDE, | ||||
|     CONF_NAME, | ||||
|     CONF_RADIUS, | ||||
|     EVENT_HOMEASSISTANT_CLOSE, | ||||
|     UnitOfLength, | ||||
|     __version__, | ||||
| ) | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.exceptions import PlatformNotReady | ||||
| from homeassistant.helpers import config_validation as cv | ||||
| from homeassistant.helpers.aiohttp_client import async_get_clientsession | ||||
| from homeassistant.helpers.entity import async_generate_entity_id | ||||
| from homeassistant.helpers.entity_platform import AddEntitiesCallback | ||||
| from homeassistant.helpers.event import async_track_time_interval | ||||
| @@ -40,33 +40,31 @@ from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
| HA_USER_AGENT = ( | ||||
|     f"{APPLICATION_NAME}/{__version__} " | ||||
|     f"python-citybikes/{CITYBIKES_CLIENT_VERSION} " | ||||
|     f"Python/{sys.version_info[0]}.{sys.version_info[1]}" | ||||
| ) | ||||
|  | ||||
| ATTR_UID = "uid" | ||||
| ATTR_LATITUDE = "latitude" | ||||
| ATTR_LONGITUDE = "longitude" | ||||
| ATTR_EMPTY_SLOTS = "empty_slots" | ||||
| ATTR_EXTRA = "extra" | ||||
| ATTR_FREE_BIKES = "free_bikes" | ||||
| ATTR_NETWORK = "network" | ||||
| ATTR_NETWORKS_LIST = "networks" | ||||
| ATTR_STATIONS_LIST = "stations" | ||||
| ATTR_TIMESTAMP = "timestamp" | ||||
| ATTR_UID = "uid" | ||||
|  | ||||
| CONF_NETWORK = "network" | ||||
| CONF_STATIONS_LIST = "stations" | ||||
|  | ||||
| DEFAULT_ENDPOINT = "https://api.citybik.es/{uri}" | ||||
| PLATFORM = "citybikes" | ||||
|  | ||||
| MONITORED_NETWORKS = "monitored-networks" | ||||
|  | ||||
| DATA_CLIENT = "client" | ||||
|  | ||||
| NETWORKS_URI = "v2/networks" | ||||
|  | ||||
| REQUEST_TIMEOUT = aiohttp.ClientTimeout(total=5) | ||||
| REQUEST_TIMEOUT = 5  # In seconds; argument to asyncio.timeout | ||||
|  | ||||
| SCAN_INTERVAL = timedelta(minutes=5)  # Timely, and doesn't suffocate the API | ||||
|  | ||||
| STATIONS_URI = "v2/networks/{uid}?fields=network.stations" | ||||
|  | ||||
| CITYBIKES_ATTRIBUTION = ( | ||||
|     "Information provided by the CityBikes Project (https://citybik.es/#about)" | ||||
| ) | ||||
| @@ -89,6 +87,72 @@ PLATFORM_SCHEMA = vol.All( | ||||
|     ), | ||||
| ) | ||||
|  | ||||
| NETWORK_SCHEMA = vol.Schema( | ||||
|     { | ||||
|         vol.Required(ATTR_ID): cv.string, | ||||
|         vol.Required(ATTR_NAME): cv.string, | ||||
|         vol.Required(ATTR_LOCATION): vol.Schema( | ||||
|             { | ||||
|                 vol.Required(ATTR_LATITUDE): cv.latitude, | ||||
|                 vol.Required(ATTR_LONGITUDE): cv.longitude, | ||||
|             }, | ||||
|             extra=vol.REMOVE_EXTRA, | ||||
|         ), | ||||
|     }, | ||||
|     extra=vol.REMOVE_EXTRA, | ||||
| ) | ||||
|  | ||||
| NETWORKS_RESPONSE_SCHEMA = vol.Schema( | ||||
|     {vol.Required(ATTR_NETWORKS_LIST): [NETWORK_SCHEMA]} | ||||
| ) | ||||
|  | ||||
| STATION_SCHEMA = vol.Schema( | ||||
|     { | ||||
|         vol.Required(ATTR_FREE_BIKES): cv.positive_int, | ||||
|         vol.Required(ATTR_EMPTY_SLOTS): vol.Any(cv.positive_int, None), | ||||
|         vol.Required(ATTR_LATITUDE): cv.latitude, | ||||
|         vol.Required(ATTR_LONGITUDE): cv.longitude, | ||||
|         vol.Required(ATTR_ID): cv.string, | ||||
|         vol.Required(ATTR_NAME): cv.string, | ||||
|         vol.Required(ATTR_TIMESTAMP): cv.string, | ||||
|         vol.Optional(ATTR_EXTRA): vol.Schema( | ||||
|             {vol.Optional(ATTR_UID): cv.string}, extra=vol.REMOVE_EXTRA | ||||
|         ), | ||||
|     }, | ||||
|     extra=vol.REMOVE_EXTRA, | ||||
| ) | ||||
|  | ||||
| STATIONS_RESPONSE_SCHEMA = vol.Schema( | ||||
|     { | ||||
|         vol.Required(ATTR_NETWORK): vol.Schema( | ||||
|             {vol.Required(ATTR_STATIONS_LIST): [STATION_SCHEMA]}, extra=vol.REMOVE_EXTRA | ||||
|         ) | ||||
|     } | ||||
| ) | ||||
|  | ||||
|  | ||||
| class CityBikesRequestError(Exception): | ||||
|     """Error to indicate a CityBikes API request has failed.""" | ||||
|  | ||||
|  | ||||
| async def async_citybikes_request(hass, uri, schema): | ||||
|     """Perform a request to CityBikes API endpoint, and parse the response.""" | ||||
|     try: | ||||
|         session = async_get_clientsession(hass) | ||||
|  | ||||
|         async with asyncio.timeout(REQUEST_TIMEOUT): | ||||
|             req = await session.get(DEFAULT_ENDPOINT.format(uri=uri)) | ||||
|  | ||||
|         json_response = await req.json() | ||||
|         return schema(json_response) | ||||
|     except (TimeoutError, aiohttp.ClientError): | ||||
|         _LOGGER.error("Could not connect to CityBikes API endpoint") | ||||
|     except ValueError: | ||||
|         _LOGGER.error("Received non-JSON data from CityBikes API endpoint") | ||||
|     except vol.Invalid as err: | ||||
|         _LOGGER.error("Received unexpected JSON from CityBikes API endpoint: %s", err) | ||||
|     raise CityBikesRequestError | ||||
|  | ||||
|  | ||||
| async def async_setup_platform( | ||||
|     hass: HomeAssistant, | ||||
| @@ -111,14 +175,6 @@ async def async_setup_platform( | ||||
|             radius, UnitOfLength.FEET, UnitOfLength.METERS | ||||
|         ) | ||||
|  | ||||
|     client = CitybikesClient(user_agent=HA_USER_AGENT, timeout=REQUEST_TIMEOUT) | ||||
|     hass.data[PLATFORM][DATA_CLIENT] = client | ||||
|  | ||||
|     async def _async_close_client(event): | ||||
|         await client.close() | ||||
|  | ||||
|     hass.bus.async_listen_once(EVENT_HOMEASSISTANT_CLOSE, _async_close_client) | ||||
|  | ||||
|     # Create a single instance of CityBikesNetworks. | ||||
|     networks = hass.data.setdefault(CITYBIKES_NETWORKS, CityBikesNetworks(hass)) | ||||
|  | ||||
| @@ -138,10 +194,10 @@ async def async_setup_platform( | ||||
|     devices = [] | ||||
|     for station in network.stations: | ||||
|         dist = location_util.distance( | ||||
|             latitude, longitude, station.latitude, station.longitude | ||||
|             latitude, longitude, station[ATTR_LATITUDE], station[ATTR_LONGITUDE] | ||||
|         ) | ||||
|         station_id = station.id | ||||
|         station_uid = str(station.extra.get(ATTR_UID, "")) | ||||
|         station_id = station[ATTR_ID] | ||||
|         station_uid = str(station.get(ATTR_EXTRA, {}).get(ATTR_UID, "")) | ||||
|  | ||||
|         if radius > dist or stations_list.intersection((station_id, station_uid)): | ||||
|             if name: | ||||
| @@ -160,7 +216,6 @@ class CityBikesNetworks: | ||||
|     def __init__(self, hass): | ||||
|         """Initialize the networks instance.""" | ||||
|         self.hass = hass | ||||
|         self.client = hass.data[PLATFORM][DATA_CLIENT] | ||||
|         self.networks = None | ||||
|         self.networks_loading = asyncio.Condition() | ||||
|  | ||||
| @@ -169,21 +224,24 @@ class CityBikesNetworks: | ||||
|         try: | ||||
|             await self.networks_loading.acquire() | ||||
|             if self.networks is None: | ||||
|                 self.networks = await self.client.networks.fetch() | ||||
|         except aiohttp.ClientError as err: | ||||
|                 networks = await async_citybikes_request( | ||||
|                     self.hass, NETWORKS_URI, NETWORKS_RESPONSE_SCHEMA | ||||
|                 ) | ||||
|                 self.networks = networks[ATTR_NETWORKS_LIST] | ||||
|         except CityBikesRequestError as err: | ||||
|             raise PlatformNotReady from err | ||||
|         else: | ||||
|             result = None | ||||
|             minimum_dist = None | ||||
|             for network in self.networks: | ||||
|                 network_latitude = network.location.latitude | ||||
|                 network_longitude = network.location.longitude | ||||
|                 network_latitude = network[ATTR_LOCATION][ATTR_LATITUDE] | ||||
|                 network_longitude = network[ATTR_LOCATION][ATTR_LONGITUDE] | ||||
|                 dist = location_util.distance( | ||||
|                     latitude, longitude, network_latitude, network_longitude | ||||
|                 ) | ||||
|                 if minimum_dist is None or dist < minimum_dist: | ||||
|                     minimum_dist = dist | ||||
|                     result = network.id | ||||
|                     result = network[ATTR_ID] | ||||
|  | ||||
|             return result | ||||
|         finally: | ||||
| @@ -199,20 +257,22 @@ class CityBikesNetwork: | ||||
|         self.network_id = network_id | ||||
|         self.stations = [] | ||||
|         self.ready = asyncio.Event() | ||||
|         self.client = hass.data[PLATFORM][DATA_CLIENT] | ||||
|  | ||||
|     async def async_refresh(self, now=None): | ||||
|         """Refresh the state of the network.""" | ||||
|         try: | ||||
|             network = await self.client.network(uid=self.network_id).fetch() | ||||
|         except aiohttp.ClientError as err: | ||||
|             if now is None: | ||||
|                 raise PlatformNotReady from err | ||||
|             self.ready.clear() | ||||
|             return | ||||
|  | ||||
|         self.stations = network.stations | ||||
|             network = await async_citybikes_request( | ||||
|                 self.hass, | ||||
|                 STATIONS_URI.format(uid=self.network_id), | ||||
|                 STATIONS_RESPONSE_SCHEMA, | ||||
|             ) | ||||
|             self.stations = network[ATTR_NETWORK][ATTR_STATIONS_LIST] | ||||
|             self.ready.set() | ||||
|         except CityBikesRequestError as err: | ||||
|             if now is not None: | ||||
|                 self.ready.clear() | ||||
|             else: | ||||
|                 raise PlatformNotReady from err | ||||
|  | ||||
|  | ||||
| class CityBikesStation(SensorEntity): | ||||
| @@ -230,13 +290,16 @@ class CityBikesStation(SensorEntity): | ||||
|  | ||||
|     async def async_update(self) -> None: | ||||
|         """Update station state.""" | ||||
|         station = next(s for s in self._network.stations if s.id == self._station_id) | ||||
|         self._attr_name = station.name | ||||
|         self._attr_native_value = station.free_bikes | ||||
|         for station in self._network.stations: | ||||
|             if station[ATTR_ID] == self._station_id: | ||||
|                 station_data = station | ||||
|                 break | ||||
|         self._attr_name = station_data.get(ATTR_NAME) | ||||
|         self._attr_native_value = station_data.get(ATTR_FREE_BIKES) | ||||
|         self._attr_extra_state_attributes = { | ||||
|             ATTR_UID: station.extra.get(ATTR_UID), | ||||
|             ATTR_LATITUDE: station.latitude, | ||||
|             ATTR_LONGITUDE: station.longitude, | ||||
|             ATTR_EMPTY_SLOTS: station.empty_slots, | ||||
|             ATTR_TIMESTAMP: station.timestamp, | ||||
|             ATTR_UID: station_data.get(ATTR_EXTRA, {}).get(ATTR_UID), | ||||
|             ATTR_LATITUDE: station_data.get(ATTR_LATITUDE), | ||||
|             ATTR_LONGITUDE: station_data.get(ATTR_LONGITUDE), | ||||
|             ATTR_EMPTY_SLOTS: station_data.get(ATTR_EMPTY_SLOTS), | ||||
|             ATTR_TIMESTAMP: station_data.get(ATTR_TIMESTAMP), | ||||
|         } | ||||
|   | ||||
| @@ -13,6 +13,6 @@ | ||||
|   "integration_type": "system", | ||||
|   "iot_class": "cloud_push", | ||||
|   "loggers": ["acme", "hass_nabucasa", "snitun"], | ||||
|   "requirements": ["hass-nabucasa==1.3.0"], | ||||
|   "requirements": ["hass-nabucasa==1.2.0"], | ||||
|   "single_config_entry": true | ||||
| } | ||||
|   | ||||
| @@ -38,10 +38,6 @@ TYPE_SPECIFY_COUNTRY = "specify_country_code" | ||||
|  | ||||
| _LOGGER = logging.getLogger(__name__) | ||||
|  | ||||
| DESCRIPTION_PLACEHOLDER = { | ||||
|     "register_link": "https://electricitymaps.com/free-tier", | ||||
| } | ||||
|  | ||||
|  | ||||
| class ElectricityMapsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|     """Handle a config flow for Co2signal.""" | ||||
| @@ -74,7 +70,6 @@ class ElectricityMapsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             return self.async_show_form( | ||||
|                 step_id="user", | ||||
|                 data_schema=data_schema, | ||||
|                 description_placeholders=DESCRIPTION_PLACEHOLDER, | ||||
|             ) | ||||
|  | ||||
|         data = {CONF_API_KEY: user_input[CONF_API_KEY]} | ||||
| @@ -184,5 +179,4 @@ class ElectricityMapsConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             step_id=step_id, | ||||
|             data_schema=data_schema, | ||||
|             errors=errors, | ||||
|             description_placeholders=DESCRIPTION_PLACEHOLDER, | ||||
|         ) | ||||
|   | ||||
| @@ -18,6 +18,7 @@ rules: | ||||
|     status: todo | ||||
|     comment: | | ||||
|       The config flow misses data descriptions. | ||||
|       Remove URLs from data descriptions, they should be replaced with placeholders. | ||||
|       Make use of Electricity Maps zone keys in country code as dropdown. | ||||
|       Make use of location selector for coordinates. | ||||
|   dependency-transparency: done | ||||
|   | ||||
| @@ -6,7 +6,7 @@ | ||||
|           "location": "[%key:common::config_flow::data::location%]", | ||||
|           "api_key": "[%key:common::config_flow::data::access_token%]" | ||||
|         }, | ||||
|         "description": "Visit the [Electricity Maps page]({register_link}) to request a token." | ||||
|         "description": "Visit https://electricitymaps.com/free-tier to request a token." | ||||
|       }, | ||||
|       "coordinates": { | ||||
|         "data": { | ||||
|   | ||||
| @@ -166,7 +166,6 @@ class CoinbaseConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|                 data_schema=STEP_USER_DATA_SCHEMA, | ||||
|                 description_placeholders={ | ||||
|                     "account_name": self.reauth_entry.title, | ||||
|                     "developer_url": "https://www.coinbase.com/developer-platform", | ||||
|                 }, | ||||
|                 errors=errors, | ||||
|             ) | ||||
| @@ -196,7 +195,6 @@ class CoinbaseConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             data_schema=STEP_USER_DATA_SCHEMA, | ||||
|             description_placeholders={ | ||||
|                 "account_name": self.reauth_entry.title, | ||||
|                 "developer_url": "https://www.coinbase.com/developer-platform", | ||||
|             }, | ||||
|             errors=errors, | ||||
|         ) | ||||
|   | ||||
| @@ -11,7 +11,7 @@ | ||||
|       }, | ||||
|       "reauth_confirm": { | ||||
|         "title": "Update Coinbase API credentials", | ||||
|         "description": "Your current Coinbase API key appears to be for the deprecated v2 API. Please reconfigure with a new API key created for the v3 API. Visit the [Developer Platform]({developer_url}) to create new credentials for {account_name}.", | ||||
|         "description": "Your current Coinbase API key appears to be for the deprecated v2 API. Please reconfigure with a new API key created for the v3 API. Visit https://www.coinbase.com/developer-platform  to create new credentials for {account_name}.", | ||||
|         "data": { | ||||
|           "api_key": "[%key:common::config_flow::data::api_key%]", | ||||
|           "api_token": "API secret" | ||||
|   | ||||
| @@ -15,7 +15,6 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
| from homeassistant.helpers.update_coordinator import CoordinatorEntity | ||||
|  | ||||
| from .coordinator import ComelitConfigEntry, ComelitVedoSystem | ||||
| from .utils import DeviceType, new_device_listener | ||||
|  | ||||
| # Coordinator is used to centralize the data updates | ||||
| PARALLEL_UPDATES = 0 | ||||
| @@ -30,19 +29,23 @@ async def async_setup_entry( | ||||
|  | ||||
|     coordinator = cast(ComelitVedoSystem, config_entry.runtime_data) | ||||
|  | ||||
|     def _add_new_entities(new_devices: list[DeviceType], dev_type: str) -> None: | ||||
|         """Add entities for new monitors.""" | ||||
|         entities = [ | ||||
|             ComelitVedoBinarySensorEntity(coordinator, device, config_entry.entry_id) | ||||
|             for device in coordinator.data["alarm_zones"].values() | ||||
|             if device in new_devices | ||||
|         ] | ||||
|         if entities: | ||||
|             async_add_entities(entities) | ||||
|     known_devices: set[int] = set() | ||||
|  | ||||
|     config_entry.async_on_unload( | ||||
|         new_device_listener(coordinator, _add_new_entities, "alarm_zones") | ||||
|     def _check_device() -> None: | ||||
|         current_devices = set(coordinator.data["alarm_zones"]) | ||||
|         new_devices = current_devices - known_devices | ||||
|         if new_devices: | ||||
|             known_devices.update(new_devices) | ||||
|             async_add_entities( | ||||
|                 ComelitVedoBinarySensorEntity( | ||||
|                     coordinator, device, config_entry.entry_id | ||||
|                 ) | ||||
|                 for device in coordinator.data["alarm_zones"].values() | ||||
|                 if device.index in new_devices | ||||
|             ) | ||||
|  | ||||
|     _check_device() | ||||
|     config_entry.async_on_unload(coordinator.async_add_listener(_check_device)) | ||||
|  | ||||
|  | ||||
| class ComelitVedoBinarySensorEntity( | ||||
|   | ||||
| @@ -4,7 +4,6 @@ from __future__ import annotations | ||||
|  | ||||
| from asyncio.exceptions import TimeoutError | ||||
| from collections.abc import Mapping | ||||
| import re | ||||
| from typing import Any | ||||
|  | ||||
| from aiocomelit import ( | ||||
| @@ -28,20 +27,25 @@ from .utils import async_client_session | ||||
| DEFAULT_HOST = "192.168.1.252" | ||||
| DEFAULT_PIN = "111111" | ||||
|  | ||||
|  | ||||
| pin_regex = r"^[0-9]{4,10}$" | ||||
|  | ||||
| USER_SCHEMA = vol.Schema( | ||||
|     { | ||||
|         vol.Required(CONF_HOST, default=DEFAULT_HOST): cv.string, | ||||
|         vol.Required(CONF_PORT, default=DEFAULT_PORT): cv.port, | ||||
|         vol.Optional(CONF_PIN, default=DEFAULT_PIN): cv.string, | ||||
|         vol.Optional(CONF_PIN, default=DEFAULT_PIN): cv.matches_regex(pin_regex), | ||||
|         vol.Required(CONF_TYPE, default=BRIDGE): vol.In(DEVICE_TYPE_LIST), | ||||
|     } | ||||
| ) | ||||
| STEP_REAUTH_DATA_SCHEMA = vol.Schema({vol.Required(CONF_PIN): cv.string}) | ||||
| STEP_REAUTH_DATA_SCHEMA = vol.Schema( | ||||
|     {vol.Required(CONF_PIN): cv.matches_regex(pin_regex)} | ||||
| ) | ||||
| STEP_RECONFIGURE = vol.Schema( | ||||
|     { | ||||
|         vol.Required(CONF_HOST): cv.string, | ||||
|         vol.Required(CONF_PORT): cv.port, | ||||
|         vol.Optional(CONF_PIN, default=DEFAULT_PIN): cv.string, | ||||
|         vol.Optional(CONF_PIN, default=DEFAULT_PIN): cv.matches_regex(pin_regex), | ||||
|     } | ||||
| ) | ||||
|  | ||||
| @@ -51,9 +55,6 @@ async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> dict[str, | ||||
|  | ||||
|     api: ComelitCommonApi | ||||
|  | ||||
|     if not re.fullmatch(r"[0-9]{4,10}", data[CONF_PIN]): | ||||
|         raise InvalidPin | ||||
|  | ||||
|     session = await async_client_session(hass) | ||||
|     if data.get(CONF_TYPE, BRIDGE) == BRIDGE: | ||||
|         api = ComeliteSerialBridgeApi( | ||||
| @@ -104,8 +105,6 @@ class ComelitConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             errors["base"] = "cannot_connect" | ||||
|         except InvalidAuth: | ||||
|             errors["base"] = "invalid_auth" | ||||
|         except InvalidPin: | ||||
|             errors["base"] = "invalid_pin" | ||||
|         except Exception:  # noqa: BLE001 | ||||
|             _LOGGER.exception("Unexpected exception") | ||||
|             errors["base"] = "unknown" | ||||
| @@ -147,8 +146,6 @@ class ComelitConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|                 errors["base"] = "cannot_connect" | ||||
|             except InvalidAuth: | ||||
|                 errors["base"] = "invalid_auth" | ||||
|             except InvalidPin: | ||||
|                 errors["base"] = "invalid_pin" | ||||
|             except Exception:  # noqa: BLE001 | ||||
|                 _LOGGER.exception("Unexpected exception") | ||||
|                 errors["base"] = "unknown" | ||||
| @@ -192,8 +189,6 @@ class ComelitConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|             errors["base"] = "cannot_connect" | ||||
|         except InvalidAuth: | ||||
|             errors["base"] = "invalid_auth" | ||||
|         except InvalidPin: | ||||
|             errors["base"] = "invalid_pin" | ||||
|         except Exception:  # noqa: BLE001 | ||||
|             _LOGGER.exception("Unexpected exception") | ||||
|             errors["base"] = "unknown" | ||||
| @@ -215,7 +210,3 @@ class CannotConnect(HomeAssistantError): | ||||
|  | ||||
| class InvalidAuth(HomeAssistantError): | ||||
|     """Error to indicate there is invalid auth.""" | ||||
|  | ||||
|  | ||||
| class InvalidPin(HomeAssistantError): | ||||
|     """Error to indicate an invalid pin.""" | ||||
|   | ||||
| @@ -161,7 +161,7 @@ class ComelitSerialBridge( | ||||
|         entry: ComelitConfigEntry, | ||||
|         host: str, | ||||
|         port: int, | ||||
|         pin: str, | ||||
|         pin: int, | ||||
|         session: ClientSession, | ||||
|     ) -> None: | ||||
|         """Initialize the scanner.""" | ||||
| @@ -195,7 +195,7 @@ class ComelitVedoSystem(ComelitBaseCoordinator[AlarmDataObject]): | ||||
|         entry: ComelitConfigEntry, | ||||
|         host: str, | ||||
|         port: int, | ||||
|         pin: str, | ||||
|         pin: int, | ||||
|         session: ClientSession, | ||||
|     ) -> None: | ||||
|         """Initialize the scanner.""" | ||||
|   | ||||
| @@ -7,14 +7,14 @@ from typing import Any, cast | ||||
| from aiocomelit import ComelitSerialBridgeObject | ||||
| from aiocomelit.const import COVER, STATE_COVER, STATE_OFF, STATE_ON | ||||
|  | ||||
| from homeassistant.components.cover import CoverDeviceClass, CoverEntity, CoverState | ||||
| from homeassistant.components.cover import CoverDeviceClass, CoverEntity | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
| from homeassistant.helpers.restore_state import RestoreEntity | ||||
|  | ||||
| from .coordinator import ComelitConfigEntry, ComelitSerialBridge | ||||
| from .entity import ComelitBridgeBaseEntity | ||||
| from .utils import DeviceType, bridge_api_call, new_device_listener | ||||
| from .utils import bridge_api_call | ||||
|  | ||||
| # Coordinator is used to centralize the data updates | ||||
| PARALLEL_UPDATES = 0 | ||||
| @@ -29,20 +29,22 @@ async def async_setup_entry( | ||||
|  | ||||
|     coordinator = cast(ComelitSerialBridge, config_entry.runtime_data) | ||||
|  | ||||
|     def _add_new_entities(new_devices: list[DeviceType], dev_type: str) -> None: | ||||
|         """Add entities for new monitors.""" | ||||
|         entities = [ | ||||
|             ComelitCoverEntity(coordinator, device, config_entry.entry_id) | ||||
|             for device in coordinator.data[dev_type].values() | ||||
|             if device in new_devices | ||||
|         ] | ||||
|         if entities: | ||||
|             async_add_entities(entities) | ||||
|     known_devices: set[int] = set() | ||||
|  | ||||
|     config_entry.async_on_unload( | ||||
|         new_device_listener(coordinator, _add_new_entities, COVER) | ||||
|     def _check_device() -> None: | ||||
|         current_devices = set(coordinator.data[COVER]) | ||||
|         new_devices = current_devices - known_devices | ||||
|         if new_devices: | ||||
|             known_devices.update(new_devices) | ||||
|             async_add_entities( | ||||
|                 ComelitCoverEntity(coordinator, device, config_entry.entry_id) | ||||
|                 for device in coordinator.data[COVER].values() | ||||
|                 if device.index in new_devices | ||||
|             ) | ||||
|  | ||||
|     _check_device() | ||||
|     config_entry.async_on_unload(coordinator.async_add_listener(_check_device)) | ||||
|  | ||||
|  | ||||
| class ComelitCoverEntity(ComelitBridgeBaseEntity, RestoreEntity, CoverEntity): | ||||
|     """Cover device.""" | ||||
| @@ -60,6 +62,7 @@ class ComelitCoverEntity(ComelitBridgeBaseEntity, RestoreEntity, CoverEntity): | ||||
|         super().__init__(coordinator, device, config_entry_entry_id) | ||||
|         # Device doesn't provide a status so we assume UNKNOWN at first startup | ||||
|         self._last_action: int | None = None | ||||
|         self._last_state: str | None = None | ||||
|  | ||||
|     def _current_action(self, action: str) -> bool: | ||||
|         """Return the current cover action.""" | ||||
| @@ -95,6 +98,7 @@ class ComelitCoverEntity(ComelitBridgeBaseEntity, RestoreEntity, CoverEntity): | ||||
|     @bridge_api_call | ||||
|     async def _cover_set_state(self, action: int, state: int) -> None: | ||||
|         """Set desired cover state.""" | ||||
|         self._last_state = self.state | ||||
|         await self.coordinator.api.set_device_status(COVER, self._device.index, action) | ||||
|         self.coordinator.data[COVER][self._device.index].status = state | ||||
|         self.async_write_ha_state() | ||||
| @@ -120,10 +124,5 @@ class ComelitCoverEntity(ComelitBridgeBaseEntity, RestoreEntity, CoverEntity): | ||||
|  | ||||
|         await super().async_added_to_hass() | ||||
|  | ||||
|         if (state := await self.async_get_last_state()) is not None: | ||||
|             if state.state == CoverState.CLOSED: | ||||
|                 self._last_action = STATE_COVER.index(CoverState.CLOSING) | ||||
|             if state.state == CoverState.OPEN: | ||||
|                 self._last_action = STATE_COVER.index(CoverState.OPENING) | ||||
|  | ||||
|             self._attr_is_closed = state.state == CoverState.CLOSED | ||||
|         if last_state := await self.async_get_last_state(): | ||||
|             self._last_state = last_state.state | ||||
|   | ||||
| @@ -12,7 +12,7 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
|  | ||||
| from .coordinator import ComelitConfigEntry, ComelitSerialBridge | ||||
| from .entity import ComelitBridgeBaseEntity | ||||
| from .utils import DeviceType, bridge_api_call, new_device_listener | ||||
| from .utils import bridge_api_call | ||||
|  | ||||
| # Coordinator is used to centralize the data updates | ||||
| PARALLEL_UPDATES = 0 | ||||
| @@ -27,20 +27,22 @@ async def async_setup_entry( | ||||
|  | ||||
|     coordinator = cast(ComelitSerialBridge, config_entry.runtime_data) | ||||
|  | ||||
|     def _add_new_entities(new_devices: list[DeviceType], dev_type: str) -> None: | ||||
|         """Add entities for new monitors.""" | ||||
|         entities = [ | ||||
|             ComelitLightEntity(coordinator, device, config_entry.entry_id) | ||||
|             for device in coordinator.data[dev_type].values() | ||||
|             if device in new_devices | ||||
|         ] | ||||
|         if entities: | ||||
|             async_add_entities(entities) | ||||
|     known_devices: set[int] = set() | ||||
|  | ||||
|     config_entry.async_on_unload( | ||||
|         new_device_listener(coordinator, _add_new_entities, LIGHT) | ||||
|     def _check_device() -> None: | ||||
|         current_devices = set(coordinator.data[LIGHT]) | ||||
|         new_devices = current_devices - known_devices | ||||
|         if new_devices: | ||||
|             known_devices.update(new_devices) | ||||
|             async_add_entities( | ||||
|                 ComelitLightEntity(coordinator, device, config_entry.entry_id) | ||||
|                 for device in coordinator.data[LIGHT].values() | ||||
|                 if device.index in new_devices | ||||
|             ) | ||||
|  | ||||
|     _check_device() | ||||
|     config_entry.async_on_unload(coordinator.async_add_listener(_check_device)) | ||||
|  | ||||
|  | ||||
| class ComelitLightEntity(ComelitBridgeBaseEntity, LightEntity): | ||||
|     """Light device.""" | ||||
|   | ||||
| @@ -8,5 +8,5 @@ | ||||
|   "iot_class": "local_polling", | ||||
|   "loggers": ["aiocomelit"], | ||||
|   "quality_scale": "platinum", | ||||
|   "requirements": ["aiocomelit==1.1.2"] | ||||
|   "requirements": ["aiocomelit==0.12.3"] | ||||
| } | ||||
|   | ||||
| @@ -20,7 +20,6 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity | ||||
|  | ||||
| from .coordinator import ComelitConfigEntry, ComelitSerialBridge, ComelitVedoSystem | ||||
| from .entity import ComelitBridgeBaseEntity | ||||
| from .utils import DeviceType, new_device_listener | ||||
|  | ||||
| # Coordinator is used to centralize the data updates | ||||
| PARALLEL_UPDATES = 0 | ||||
| @@ -66,23 +65,25 @@ async def async_setup_bridge_entry( | ||||
|  | ||||
|     coordinator = cast(ComelitSerialBridge, config_entry.runtime_data) | ||||
|  | ||||
|     def _add_new_entities(new_devices: list[DeviceType], dev_type: str) -> None: | ||||
|         """Add entities for new monitors.""" | ||||
|         entities = [ | ||||
|     known_devices: set[int] = set() | ||||
|  | ||||
|     def _check_device() -> None: | ||||
|         current_devices = set(coordinator.data[OTHER]) | ||||
|         new_devices = current_devices - known_devices | ||||
|         if new_devices: | ||||
|             known_devices.update(new_devices) | ||||
|             async_add_entities( | ||||
|                 ComelitBridgeSensorEntity( | ||||
|                     coordinator, device, config_entry.entry_id, sensor_desc | ||||
|                 ) | ||||
|                 for sensor_desc in SENSOR_BRIDGE_TYPES | ||||
|             for device in coordinator.data[dev_type].values() | ||||
|             if device in new_devices | ||||
|         ] | ||||
|         if entities: | ||||
|             async_add_entities(entities) | ||||
|  | ||||
|     config_entry.async_on_unload( | ||||
|         new_device_listener(coordinator, _add_new_entities, OTHER) | ||||
|                 for device in coordinator.data[OTHER].values() | ||||
|                 if device.index in new_devices | ||||
|             ) | ||||
|  | ||||
|     _check_device() | ||||
|     config_entry.async_on_unload(coordinator.async_add_listener(_check_device)) | ||||
|  | ||||
|  | ||||
| async def async_setup_vedo_entry( | ||||
|     hass: HomeAssistant, | ||||
| @@ -93,23 +94,25 @@ async def async_setup_vedo_entry( | ||||
|  | ||||
|     coordinator = cast(ComelitVedoSystem, config_entry.runtime_data) | ||||
|  | ||||
|     def _add_new_entities(new_devices: list[DeviceType], dev_type: str) -> None: | ||||
|         """Add entities for new monitors.""" | ||||
|         entities = [ | ||||
|     known_devices: set[int] = set() | ||||
|  | ||||
|     def _check_device() -> None: | ||||
|         current_devices = set(coordinator.data["alarm_zones"]) | ||||
|         new_devices = current_devices - known_devices | ||||
|         if new_devices: | ||||
|             known_devices.update(new_devices) | ||||
|             async_add_entities( | ||||
|                 ComelitVedoSensorEntity( | ||||
|                     coordinator, device, config_entry.entry_id, sensor_desc | ||||
|                 ) | ||||
|                 for sensor_desc in SENSOR_VEDO_TYPES | ||||
|                 for device in coordinator.data["alarm_zones"].values() | ||||
|             if device in new_devices | ||||
|         ] | ||||
|         if entities: | ||||
|             async_add_entities(entities) | ||||
|  | ||||
|     config_entry.async_on_unload( | ||||
|         new_device_listener(coordinator, _add_new_entities, "alarm_zones") | ||||
|                 if device.index in new_devices | ||||
|             ) | ||||
|  | ||||
|     _check_device() | ||||
|     config_entry.async_on_unload(coordinator.async_add_listener(_check_device)) | ||||
|  | ||||
|  | ||||
| class ComelitBridgeSensorEntity(ComelitBridgeBaseEntity, SensorEntity): | ||||
|     """Sensor device.""" | ||||
|   | ||||
| @@ -43,13 +43,11 @@ | ||||
|       "reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]", | ||||
|       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]", | ||||
|       "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]", | ||||
|       "invalid_pin": "The provided PIN is invalid. It must be a 4-10 digit number.", | ||||
|       "unknown": "[%key:common::config_flow::error::unknown%]" | ||||
|     }, | ||||
|     "error": { | ||||
|       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]", | ||||
|       "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]", | ||||
|       "invalid_pin": "[%key:component::comelit::config::abort::invalid_pin%]", | ||||
|       "unknown": "[%key:common::config_flow::error::unknown%]" | ||||
|     } | ||||
|   }, | ||||
|   | ||||
| @@ -13,7 +13,7 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback | ||||
|  | ||||
| from .coordinator import ComelitConfigEntry, ComelitSerialBridge | ||||
| from .entity import ComelitBridgeBaseEntity | ||||
| from .utils import DeviceType, bridge_api_call, new_device_listener | ||||
| from .utils import bridge_api_call | ||||
|  | ||||
| # Coordinator is used to centralize the data updates | ||||
| PARALLEL_UPDATES = 0 | ||||
| @@ -28,21 +28,36 @@ async def async_setup_entry( | ||||
|  | ||||
|     coordinator = cast(ComelitSerialBridge, config_entry.runtime_data) | ||||
|  | ||||
|     def _add_new_entities(new_devices: list[DeviceType], dev_type: str) -> None: | ||||
|         """Add entities for new monitors.""" | ||||
|         entities = [ | ||||
|     entities: list[ComelitSwitchEntity] = [] | ||||
|     entities.extend( | ||||
|         ComelitSwitchEntity(coordinator, device, config_entry.entry_id) | ||||
|             for device in coordinator.data[dev_type].values() | ||||
|             if device in new_devices | ||||
|         ] | ||||
|         if entities: | ||||
|         for device in coordinator.data[IRRIGATION].values() | ||||
|     ) | ||||
|     entities.extend( | ||||
|         ComelitSwitchEntity(coordinator, device, config_entry.entry_id) | ||||
|         for device in coordinator.data[OTHER].values() | ||||
|     ) | ||||
|     async_add_entities(entities) | ||||
|  | ||||
|     known_devices: dict[str, set[int]] = { | ||||
|         dev_type: set() for dev_type in (IRRIGATION, OTHER) | ||||
|     } | ||||
|  | ||||
|     def _check_device() -> None: | ||||
|         for dev_type in (IRRIGATION, OTHER): | ||||
|         config_entry.async_on_unload( | ||||
|             new_device_listener(coordinator, _add_new_entities, dev_type) | ||||
|             current_devices = set(coordinator.data[dev_type]) | ||||
|             new_devices = current_devices - known_devices[dev_type] | ||||
|             if new_devices: | ||||
|                 known_devices[dev_type].update(new_devices) | ||||
|                 async_add_entities( | ||||
|                     ComelitSwitchEntity(coordinator, device, config_entry.entry_id) | ||||
|                     for device in coordinator.data[dev_type].values() | ||||
|                     if device.index in new_devices | ||||
|                 ) | ||||
|  | ||||
|     _check_device() | ||||
|     config_entry.async_on_unload(coordinator.async_add_listener(_check_device)) | ||||
|  | ||||
|  | ||||
| class ComelitSwitchEntity(ComelitBridgeBaseEntity, SwitchEntity): | ||||
|     """Switch device.""" | ||||
|   | ||||
| @@ -4,11 +4,7 @@ from collections.abc import Awaitable, Callable, Coroutine | ||||
| from functools import wraps | ||||
| from typing import Any, Concatenate | ||||
|  | ||||
| from aiocomelit.api import ( | ||||
|     ComelitSerialBridgeObject, | ||||
|     ComelitVedoAreaObject, | ||||
|     ComelitVedoZoneObject, | ||||
| ) | ||||
| from aiocomelit import ComelitSerialBridgeObject | ||||
| from aiocomelit.exceptions import CannotAuthenticate, CannotConnect, CannotRetrieveData | ||||
| from aiohttp import ClientSession, CookieJar | ||||
|  | ||||
| @@ -23,11 +19,8 @@ from homeassistant.helpers import ( | ||||
| ) | ||||
|  | ||||
| from .const import _LOGGER, DOMAIN | ||||
| from .coordinator import ComelitBaseCoordinator | ||||
| from .entity import ComelitBridgeBaseEntity | ||||
|  | ||||
| DeviceType = ComelitSerialBridgeObject | ComelitVedoAreaObject | ComelitVedoZoneObject | ||||
|  | ||||
|  | ||||
| async def async_client_session(hass: HomeAssistant) -> ClientSession: | ||||
|     """Return a new aiohttp session.""" | ||||
| @@ -120,41 +113,3 @@ def bridge_api_call[_T: ComelitBridgeBaseEntity, **_P]( | ||||
|             self.coordinator.config_entry.async_start_reauth(self.hass) | ||||
|  | ||||
|     return cmd_wrapper | ||||
|  | ||||
|  | ||||
| def new_device_listener( | ||||
|     coordinator: ComelitBaseCoordinator, | ||||
|     new_devices_callback: Callable[ | ||||
|         [ | ||||
|             list[ | ||||
|                 ComelitSerialBridgeObject | ||||
|                 | ComelitVedoAreaObject | ||||
|                 | ComelitVedoZoneObject | ||||
|             ], | ||||
|             str, | ||||
|         ], | ||||
|         None, | ||||
|     ], | ||||
|     data_type: str, | ||||
| ) -> Callable[[], None]: | ||||
|     """Subscribe to coordinator updates to check for new devices.""" | ||||
|     known_devices: dict[str, list[int]] = {} | ||||
|  | ||||
|     def _check_devices() -> None: | ||||
|         """Check for new devices and call callback with any new monitors.""" | ||||
|         if not coordinator.data: | ||||
|             return | ||||
|  | ||||
|         new_devices: list[DeviceType] = [] | ||||
|         for _id in coordinator.data[data_type]: | ||||
|             if _id not in (id_list := known_devices.get(data_type, [])): | ||||
|                 known_devices.update({data_type: [*id_list, _id]}) | ||||
|                 new_devices.append(coordinator.data[data_type][_id]) | ||||
|  | ||||
|         if new_devices: | ||||
|             new_devices_callback(new_devices, data_type) | ||||
|  | ||||
|     # Check for devices immediately | ||||
|     _check_devices() | ||||
|  | ||||
|     return coordinator.async_add_listener(_check_devices) | ||||
|   | ||||
| @@ -6,7 +6,7 @@ | ||||
|   "documentation": "https://www.home-assistant.io/integrations/control4", | ||||
|   "iot_class": "local_polling", | ||||
|   "loggers": ["pyControl4"], | ||||
|   "requirements": ["pyControl4==1.5.0"], | ||||
|   "requirements": ["pyControl4==1.2.0"], | ||||
|   "ssdp": [ | ||||
|     { | ||||
|       "st": "c4:director" | ||||
|   | ||||
| @@ -148,15 +148,6 @@ async def async_setup_entry( | ||||
|                             source_type={dev_type}, idx=dev_id, name=name | ||||
|                         ) | ||||
|  | ||||
|         # Skip rooms with no audio/video sources | ||||
|         if not sources: | ||||
|             _LOGGER.debug( | ||||
|                 "Skipping room '%s' (ID: %s) - no audio/video sources found", | ||||
|                 room.get("name"), | ||||
|                 room_id, | ||||
|             ) | ||||
|             continue | ||||
|  | ||||
|         try: | ||||
|             hidden = room["roomHidden"] | ||||
|             entity_list.append( | ||||
|   | ||||
| @@ -514,7 +514,7 @@ class ChatLog: | ||||
|         """Set the LLM system prompt.""" | ||||
|         llm_api: llm.APIInstance | None = None | ||||
|  | ||||
|         if not user_llm_hass_api: | ||||
|         if user_llm_hass_api is None: | ||||
|             pass | ||||
|         elif isinstance(user_llm_hass_api, llm.API): | ||||
|             llm_api = await user_llm_hass_api.async_get_api_instance(llm_context) | ||||
|   | ||||
| @@ -38,30 +38,22 @@ from home_assistant_intents import ( | ||||
|     ErrorKey, | ||||
|     FuzzyConfig, | ||||
|     FuzzyLanguageResponses, | ||||
|     LanguageScores, | ||||
|     get_fuzzy_config, | ||||
|     get_fuzzy_language, | ||||
|     get_intents, | ||||
|     get_language_scores, | ||||
|     get_languages, | ||||
| ) | ||||
| import yaml | ||||
|  | ||||
| from homeassistant import core | ||||
| from homeassistant.components.homeassistant.exposed_entities import ( | ||||
|     async_listen_entity_updates, | ||||
|     async_should_expose, | ||||
| ) | ||||
| from homeassistant.const import EVENT_STATE_CHANGED, MATCH_ALL | ||||
| from homeassistant.core import ( | ||||
|     Event, | ||||
|     EventStateChangedData, | ||||
|     HomeAssistant, | ||||
|     State, | ||||
|     callback, | ||||
| ) | ||||
| from homeassistant.core import Event, callback | ||||
| from homeassistant.helpers import ( | ||||
|     area_registry as ar, | ||||
|     config_validation as cv, | ||||
|     device_registry as dr, | ||||
|     entity_registry as er, | ||||
|     floor_registry as fr, | ||||
| @@ -200,7 +192,7 @@ class IntentCache: | ||||
|  | ||||
|  | ||||
| async def async_setup_default_agent( | ||||
|     hass: HomeAssistant, | ||||
|     hass: core.HomeAssistant, | ||||
|     entity_component: EntityComponent[ConversationEntity], | ||||
|     config_intents: dict[str, Any], | ||||
| ) -> None: | ||||
| @@ -209,13 +201,15 @@ async def async_setup_default_agent( | ||||
|     await entity_component.async_add_entities([agent]) | ||||
|     await get_agent_manager(hass).async_setup_default_agent(agent) | ||||
|  | ||||
|     @callback | ||||
|     def async_entity_state_listener(event: Event[EventStateChangedData]) -> None: | ||||
|     @core.callback | ||||
|     def async_entity_state_listener( | ||||
|         event: core.Event[core.EventStateChangedData], | ||||
|     ) -> None: | ||||
|         """Set expose flag on new entities.""" | ||||
|         async_should_expose(hass, DOMAIN, event.data["entity_id"]) | ||||
|  | ||||
|     @callback | ||||
|     def async_hass_started(hass: HomeAssistant) -> None: | ||||
|     @core.callback | ||||
|     def async_hass_started(hass: core.HomeAssistant) -> None: | ||||
|         """Set expose flag on all entities.""" | ||||
|         for state in hass.states.async_all(): | ||||
|             async_should_expose(hass, DOMAIN, state.entity_id) | ||||
| @@ -230,7 +224,9 @@ class DefaultAgent(ConversationEntity): | ||||
|     _attr_name = "Home Assistant" | ||||
|     _attr_supported_features = ConversationEntityFeature.CONTROL | ||||
|  | ||||
|     def __init__(self, hass: HomeAssistant, config_intents: dict[str, Any]) -> None: | ||||
|     def __init__( | ||||
|         self, hass: core.HomeAssistant, config_intents: dict[str, Any] | ||||
|     ) -> None: | ||||
|         """Initialize the default agent.""" | ||||
|         self.hass = hass | ||||
|         self._lang_intents: dict[str, LanguageIntents | object] = {} | ||||
| @@ -263,7 +259,7 @@ class DefaultAgent(ConversationEntity): | ||||
|         """Return a list of supported languages.""" | ||||
|         return get_languages() | ||||
|  | ||||
|     @callback | ||||
|     @core.callback | ||||
|     def _filter_entity_registry_changes( | ||||
|         self, event_data: er.EventEntityRegistryUpdatedData | ||||
|     ) -> bool: | ||||
| @@ -272,12 +268,12 @@ class DefaultAgent(ConversationEntity): | ||||
|             field in event_data["changes"] for field in _ENTITY_REGISTRY_UPDATE_FIELDS | ||||
|         ) | ||||
|  | ||||
|     @callback | ||||
|     def _filter_state_changes(self, event_data: EventStateChangedData) -> bool: | ||||
|     @core.callback | ||||
|     def _filter_state_changes(self, event_data: core.EventStateChangedData) -> bool: | ||||
|         """Filter state changed events.""" | ||||
|         return not event_data["old_state"] or not event_data["new_state"] | ||||
|  | ||||
|     @callback | ||||
|     @core.callback | ||||
|     def _listen_clear_slot_list(self) -> None: | ||||
|         """Listen for changes that can invalidate slot list.""" | ||||
|         assert self._unsub_clear_slot_list is None | ||||
| @@ -346,81 +342,6 @@ class DefaultAgent(ConversationEntity): | ||||
|  | ||||
|         return result | ||||
|  | ||||
|     async def async_debug_recognize( | ||||
|         self, user_input: ConversationInput | ||||
|     ) -> dict[str, Any] | None: | ||||
|         """Debug recognize from user input.""" | ||||
|         result_dict: dict[str, Any] | None = None | ||||
|  | ||||
|         if trigger_result := await self.async_recognize_sentence_trigger(user_input): | ||||
|             result_dict = { | ||||
|                 # Matched a user-defined sentence trigger. | ||||
|                 # We can't provide the response here without executing the | ||||
|                 # trigger. | ||||
|                 "match": True, | ||||
|                 "source": "trigger", | ||||
|                 "sentence_template": trigger_result.sentence_template or "", | ||||
|             } | ||||
|         elif intent_result := await self.async_recognize_intent(user_input): | ||||
|             successful_match = not intent_result.unmatched_entities | ||||
|             result_dict = { | ||||
|                 # Name of the matching intent (or the closest) | ||||
|                 "intent": { | ||||
|                     "name": intent_result.intent.name, | ||||
|                 }, | ||||
|                 # Slot values that would be received by the intent | ||||
|                 "slots": {  # direct access to values | ||||
|                     entity_key: entity.text or entity.value | ||||
|                     for entity_key, entity in intent_result.entities.items() | ||||
|                 }, | ||||
|                 # Extra slot details, such as the originally matched text | ||||
|                 "details": { | ||||
|                     entity_key: { | ||||
|                         "name": entity.name, | ||||
|                         "value": entity.value, | ||||
|                         "text": entity.text, | ||||
|                     } | ||||
|                     for entity_key, entity in intent_result.entities.items() | ||||
|                 }, | ||||
|                 # Entities/areas/etc. that would be targeted | ||||
|                 "targets": {}, | ||||
|                 # True if match was successful | ||||
|                 "match": successful_match, | ||||
|                 # Text of the sentence template that matched (or was closest) | ||||
|                 "sentence_template": "", | ||||
|                 # When match is incomplete, this will contain the best slot guesses | ||||
|                 "unmatched_slots": _get_unmatched_slots(intent_result), | ||||
|                 # True if match was not exact | ||||
|                 "fuzzy_match": False, | ||||
|             } | ||||
|  | ||||
|             if successful_match: | ||||
|                 result_dict["targets"] = { | ||||
|                     state.entity_id: {"matched": is_matched} | ||||
|                     for state, is_matched in _get_debug_targets( | ||||
|                         self.hass, intent_result | ||||
|                     ) | ||||
|                 } | ||||
|  | ||||
|             if intent_result.intent_sentence is not None: | ||||
|                 result_dict["sentence_template"] = intent_result.intent_sentence.text | ||||
|  | ||||
|             if intent_result.intent_metadata: | ||||
|                 # Inspect metadata to determine if this matched a custom sentence | ||||
|                 if intent_result.intent_metadata.get(METADATA_CUSTOM_SENTENCE): | ||||
|                     result_dict["source"] = "custom" | ||||
|                     result_dict["file"] = intent_result.intent_metadata.get( | ||||
|                         METADATA_CUSTOM_FILE | ||||
|                     ) | ||||
|                 else: | ||||
|                     result_dict["source"] = "builtin" | ||||
|  | ||||
|                 result_dict["fuzzy_match"] = intent_result.intent_metadata.get( | ||||
|                     METADATA_FUZZY_MATCH, False | ||||
|                 ) | ||||
|  | ||||
|         return result_dict | ||||
|  | ||||
|     async def _async_handle_message( | ||||
|         self, | ||||
|         user_input: ConversationInput, | ||||
| @@ -969,7 +890,7 @@ class DefaultAgent(ConversationEntity): | ||||
|     ) -> str: | ||||
|         # Get first matched or unmatched state. | ||||
|         # This is available in the response template as "state". | ||||
|         state1: State | None = None | ||||
|         state1: core.State | None = None | ||||
|         if intent_response.matched_states: | ||||
|             state1 = intent_response.matched_states[0] | ||||
|         elif intent_response.unmatched_states: | ||||
| @@ -1607,10 +1528,6 @@ class DefaultAgent(ConversationEntity): | ||||
|             return None | ||||
|         return response | ||||
|  | ||||
|     async def async_get_language_scores(self) -> dict[str, LanguageScores]: | ||||
|         """Get support scores per language.""" | ||||
|         return await self.hass.async_add_executor_job(get_language_scores) | ||||
|  | ||||
|  | ||||
| def _make_error_result( | ||||
|     language: str, | ||||
| @@ -1672,7 +1589,7 @@ def _get_unmatched_response(result: RecognizeResult) -> tuple[ErrorKey, dict[str | ||||
|  | ||||
|  | ||||
| def _get_match_error_response( | ||||
|     hass: HomeAssistant, | ||||
|     hass: core.HomeAssistant, | ||||
|     match_error: intent.MatchFailedError, | ||||
| ) -> tuple[ErrorKey, dict[str, Any]]: | ||||
|     """Return key and template arguments for error when target matching fails.""" | ||||
| @@ -1807,75 +1724,3 @@ def _collect_list_references(expression: Expression, list_names: set[str]) -> No | ||||
|     elif isinstance(expression, ListReference): | ||||
|         # {list} | ||||
|         list_names.add(expression.slot_name) | ||||
|  | ||||
|  | ||||
| def _get_debug_targets( | ||||
|     hass: HomeAssistant, | ||||
|     result: RecognizeResult, | ||||
| ) -> Iterable[tuple[State, bool]]: | ||||
|     """Yield state/is_matched pairs for a hassil recognition.""" | ||||
|     entities = result.entities | ||||
|  | ||||
|     name: str | None = None | ||||
|     area_name: str | None = None | ||||
|     domains: set[str] | None = None | ||||
|     device_classes: set[str] | None = None | ||||
|     state_names: set[str] | None = None | ||||
|  | ||||
|     if "name" in entities: | ||||
|         name = str(entities["name"].value) | ||||
|  | ||||
|     if "area" in entities: | ||||
|         area_name = str(entities["area"].value) | ||||
|  | ||||
|     if "domain" in entities: | ||||
|         domains = set(cv.ensure_list(entities["domain"].value)) | ||||
|  | ||||
|     if "device_class" in entities: | ||||
|         device_classes = set(cv.ensure_list(entities["device_class"].value)) | ||||
|  | ||||
|     if "state" in entities: | ||||
|         # HassGetState only | ||||
|         state_names = set(cv.ensure_list(entities["state"].value)) | ||||
|  | ||||
|     if ( | ||||
|         (name is None) | ||||
|         and (area_name is None) | ||||
|         and (not domains) | ||||
|         and (not device_classes) | ||||
|         and (not state_names) | ||||
|     ): | ||||
|         # Avoid "matching" all entities when there is no filter | ||||
|         return | ||||
|  | ||||
|     states = intent.async_match_states( | ||||
|         hass, | ||||
|         name=name, | ||||
|         area_name=area_name, | ||||
|         domains=domains, | ||||
|         device_classes=device_classes, | ||||
|     ) | ||||
|  | ||||
|     for state in states: | ||||
|         # For queries, a target is "matched" based on its state | ||||
|         is_matched = (state_names is None) or (state.state in state_names) | ||||
|         yield state, is_matched | ||||
|  | ||||
|  | ||||
| def _get_unmatched_slots( | ||||
|     result: RecognizeResult, | ||||
| ) -> dict[str, str | int | float]: | ||||
|     """Return a dict of unmatched text/range slot entities.""" | ||||
|     unmatched_slots: dict[str, str | int | float] = {} | ||||
|     for entity in result.unmatched_entities_list: | ||||
|         if isinstance(entity, UnmatchedTextEntity): | ||||
|             if entity.text == MISSING_ENTITY: | ||||
|                 # Don't report <missing> since these are just missing context | ||||
|                 # slots. | ||||
|                 continue | ||||
|  | ||||
|             unmatched_slots[entity.name] = entity.text | ||||
|         elif isinstance(entity, UnmatchedRangeEntity): | ||||
|             unmatched_slots[entity.name] = entity.value | ||||
|  | ||||
|     return unmatched_slots | ||||
|   | ||||
| @@ -2,16 +2,21 @@ | ||||
|  | ||||
| from __future__ import annotations | ||||
|  | ||||
| from collections.abc import Iterable | ||||
| from dataclasses import asdict | ||||
| from typing import Any | ||||
|  | ||||
| from aiohttp import web | ||||
| from hassil.recognize import MISSING_ENTITY, RecognizeResult | ||||
| from hassil.string_matcher import UnmatchedRangeEntity, UnmatchedTextEntity | ||||
| from home_assistant_intents import get_language_scores | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.components import http, websocket_api | ||||
| from homeassistant.components.http.data_validator import RequestDataValidator | ||||
| from homeassistant.const import MATCH_ALL | ||||
| from homeassistant.core import HomeAssistant, callback | ||||
| from homeassistant.core import HomeAssistant, State, callback | ||||
| from homeassistant.helpers import config_validation as cv, intent | ||||
| from homeassistant.util import language as language_util | ||||
|  | ||||
| from .agent_manager import ( | ||||
| @@ -21,6 +26,11 @@ from .agent_manager import ( | ||||
|     get_agent_manager, | ||||
| ) | ||||
| from .const import DATA_COMPONENT | ||||
| from .default_agent import ( | ||||
|     METADATA_CUSTOM_FILE, | ||||
|     METADATA_CUSTOM_SENTENCE, | ||||
|     METADATA_FUZZY_MATCH, | ||||
| ) | ||||
| from .entity import ConversationEntity | ||||
| from .models import ConversationInput | ||||
|  | ||||
| @@ -196,12 +206,150 @@ async def websocket_hass_agent_debug( | ||||
|             language=msg.get("language", hass.config.language), | ||||
|             agent_id=agent.entity_id, | ||||
|         ) | ||||
|         result_dict = await agent.async_debug_recognize(user_input) | ||||
|         result_dict: dict[str, Any] | None = None | ||||
|  | ||||
|         if trigger_result := await agent.async_recognize_sentence_trigger(user_input): | ||||
|             result_dict = { | ||||
|                 # Matched a user-defined sentence trigger. | ||||
|                 # We can't provide the response here without executing the | ||||
|                 # trigger. | ||||
|                 "match": True, | ||||
|                 "source": "trigger", | ||||
|                 "sentence_template": trigger_result.sentence_template or "", | ||||
|             } | ||||
|         elif intent_result := await agent.async_recognize_intent(user_input): | ||||
|             successful_match = not intent_result.unmatched_entities | ||||
|             result_dict = { | ||||
|                 # Name of the matching intent (or the closest) | ||||
|                 "intent": { | ||||
|                     "name": intent_result.intent.name, | ||||
|                 }, | ||||
|                 # Slot values that would be received by the intent | ||||
|                 "slots": {  # direct access to values | ||||
|                     entity_key: entity.text or entity.value | ||||
|                     for entity_key, entity in intent_result.entities.items() | ||||
|                 }, | ||||
|                 # Extra slot details, such as the originally matched text | ||||
|                 "details": { | ||||
|                     entity_key: { | ||||
|                         "name": entity.name, | ||||
|                         "value": entity.value, | ||||
|                         "text": entity.text, | ||||
|                     } | ||||
|                     for entity_key, entity in intent_result.entities.items() | ||||
|                 }, | ||||
|                 # Entities/areas/etc. that would be targeted | ||||
|                 "targets": {}, | ||||
|                 # True if match was successful | ||||
|                 "match": successful_match, | ||||
|                 # Text of the sentence template that matched (or was closest) | ||||
|                 "sentence_template": "", | ||||
|                 # When match is incomplete, this will contain the best slot guesses | ||||
|                 "unmatched_slots": _get_unmatched_slots(intent_result), | ||||
|                 # True if match was not exact | ||||
|                 "fuzzy_match": False, | ||||
|             } | ||||
|  | ||||
|             if successful_match: | ||||
|                 result_dict["targets"] = { | ||||
|                     state.entity_id: {"matched": is_matched} | ||||
|                     for state, is_matched in _get_debug_targets(hass, intent_result) | ||||
|                 } | ||||
|  | ||||
|             if intent_result.intent_sentence is not None: | ||||
|                 result_dict["sentence_template"] = intent_result.intent_sentence.text | ||||
|  | ||||
|             if intent_result.intent_metadata: | ||||
|                 # Inspect metadata to determine if this matched a custom sentence | ||||
|                 if intent_result.intent_metadata.get(METADATA_CUSTOM_SENTENCE): | ||||
|                     result_dict["source"] = "custom" | ||||
|                     result_dict["file"] = intent_result.intent_metadata.get( | ||||
|                         METADATA_CUSTOM_FILE | ||||
|                     ) | ||||
|                 else: | ||||
|                     result_dict["source"] = "builtin" | ||||
|  | ||||
|                 result_dict["fuzzy_match"] = intent_result.intent_metadata.get( | ||||
|                     METADATA_FUZZY_MATCH, False | ||||
|                 ) | ||||
|  | ||||
|         result_dicts.append(result_dict) | ||||
|  | ||||
|     connection.send_result(msg["id"], {"results": result_dicts}) | ||||
|  | ||||
|  | ||||
| def _get_debug_targets( | ||||
|     hass: HomeAssistant, | ||||
|     result: RecognizeResult, | ||||
| ) -> Iterable[tuple[State, bool]]: | ||||
|     """Yield state/is_matched pairs for a hassil recognition.""" | ||||
|     entities = result.entities | ||||
|  | ||||
|     name: str | None = None | ||||
|     area_name: str | None = None | ||||
|     domains: set[str] | None = None | ||||
|     device_classes: set[str] | None = None | ||||
|     state_names: set[str] | None = None | ||||
|  | ||||
|     if "name" in entities: | ||||
|         name = str(entities["name"].value) | ||||
|  | ||||
|     if "area" in entities: | ||||
|         area_name = str(entities["area"].value) | ||||
|  | ||||
|     if "domain" in entities: | ||||
|         domains = set(cv.ensure_list(entities["domain"].value)) | ||||
|  | ||||
|     if "device_class" in entities: | ||||
|         device_classes = set(cv.ensure_list(entities["device_class"].value)) | ||||
|  | ||||
|     if "state" in entities: | ||||
|         # HassGetState only | ||||
|         state_names = set(cv.ensure_list(entities["state"].value)) | ||||
|  | ||||
|     if ( | ||||
|         (name is None) | ||||
|         and (area_name is None) | ||||
|         and (not domains) | ||||
|         and (not device_classes) | ||||
|         and (not state_names) | ||||
|     ): | ||||
|         # Avoid "matching" all entities when there is no filter | ||||
|         return | ||||
|  | ||||
|     states = intent.async_match_states( | ||||
|         hass, | ||||
|         name=name, | ||||
|         area_name=area_name, | ||||
|         domains=domains, | ||||
|         device_classes=device_classes, | ||||
|     ) | ||||
|  | ||||
|     for state in states: | ||||
|         # For queries, a target is "matched" based on its state | ||||
|         is_matched = (state_names is None) or (state.state in state_names) | ||||
|         yield state, is_matched | ||||
|  | ||||
|  | ||||
| def _get_unmatched_slots( | ||||
|     result: RecognizeResult, | ||||
| ) -> dict[str, str | int | float]: | ||||
|     """Return a dict of unmatched text/range slot entities.""" | ||||
|     unmatched_slots: dict[str, str | int | float] = {} | ||||
|     for entity in result.unmatched_entities_list: | ||||
|         if isinstance(entity, UnmatchedTextEntity): | ||||
|             if entity.text == MISSING_ENTITY: | ||||
|                 # Don't report <missing> since these are just missing context | ||||
|                 # slots. | ||||
|                 continue | ||||
|  | ||||
|             unmatched_slots[entity.name] = entity.text | ||||
|         elif isinstance(entity, UnmatchedRangeEntity): | ||||
|             unmatched_slots[entity.name] = entity.value | ||||
|  | ||||
|     return unmatched_slots | ||||
|  | ||||
|  | ||||
| @websocket_api.websocket_command( | ||||
|     { | ||||
|         vol.Required("type"): "conversation/agent/homeassistant/language_scores", | ||||
| @@ -216,13 +364,10 @@ async def websocket_hass_agent_language_scores( | ||||
|     msg: dict[str, Any], | ||||
| ) -> None: | ||||
|     """Get support scores per language.""" | ||||
|     agent = get_agent_manager(hass).default_agent | ||||
|     assert agent is not None | ||||
|  | ||||
|     language = msg.get("language", hass.config.language) | ||||
|     country = msg.get("country", hass.config.country) | ||||
|  | ||||
|     scores = await agent.async_get_language_scores() | ||||
|     scores = await hass.async_add_executor_job(get_language_scores) | ||||
|     matching_langs = language_util.matches(language, scores.keys(), country=country) | ||||
|     preferred_lang = matching_langs[0] if matching_langs else language | ||||
|     result = { | ||||
|   | ||||
| @@ -13,7 +13,7 @@ from propcache.api import cached_property | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.config_entries import ConfigEntry | ||||
| from homeassistant.const import ( | ||||
| from homeassistant.const import (  # noqa: F401 | ||||
|     SERVICE_CLOSE_COVER, | ||||
|     SERVICE_CLOSE_COVER_TILT, | ||||
|     SERVICE_OPEN_COVER, | ||||
| @@ -24,9 +24,19 @@ from homeassistant.const import ( | ||||
|     SERVICE_STOP_COVER_TILT, | ||||
|     SERVICE_TOGGLE, | ||||
|     SERVICE_TOGGLE_COVER_TILT, | ||||
|     STATE_CLOSED, | ||||
|     STATE_CLOSING, | ||||
|     STATE_OPEN, | ||||
|     STATE_OPENING, | ||||
| ) | ||||
| from homeassistant.core import HomeAssistant | ||||
| from homeassistant.helpers import config_validation as cv | ||||
| from homeassistant.helpers.deprecation import ( | ||||
|     DeprecatedConstantEnum, | ||||
|     all_with_deprecated_constants, | ||||
|     check_if_deprecated_constant, | ||||
|     dir_with_deprecated_constants, | ||||
| ) | ||||
| from homeassistant.helpers.entity import Entity, EntityDescription | ||||
| from homeassistant.helpers.entity_component import EntityComponent | ||||
| from homeassistant.helpers.typing import ConfigType | ||||
| @@ -53,6 +63,15 @@ class CoverState(StrEnum): | ||||
|     OPENING = "opening" | ||||
|  | ||||
|  | ||||
| # STATE_* below are deprecated as of 2024.11 | ||||
| # when imported from homeassistant.components.cover | ||||
| # use the CoverState enum instead. | ||||
| _DEPRECATED_STATE_CLOSED = DeprecatedConstantEnum(CoverState.CLOSED, "2025.11") | ||||
| _DEPRECATED_STATE_CLOSING = DeprecatedConstantEnum(CoverState.CLOSING, "2025.11") | ||||
| _DEPRECATED_STATE_OPEN = DeprecatedConstantEnum(CoverState.OPEN, "2025.11") | ||||
| _DEPRECATED_STATE_OPENING = DeprecatedConstantEnum(CoverState.OPENING, "2025.11") | ||||
|  | ||||
|  | ||||
| class CoverDeviceClass(StrEnum): | ||||
|     """Device class for cover.""" | ||||
|  | ||||
| @@ -444,3 +463,11 @@ class CoverEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_): | ||||
|         return ( | ||||
|             fns["close"] if self._cover_is_last_toggle_direction_open else fns["open"] | ||||
|         ) | ||||
|  | ||||
|  | ||||
| # These can be removed if no deprecated constant are in this module anymore | ||||
| __getattr__ = ft.partial(check_if_deprecated_constant, module_globals=globals()) | ||||
| __dir__ = ft.partial( | ||||
|     dir_with_deprecated_constants, module_globals_keys=[*globals().keys()] | ||||
| ) | ||||
| __all__ = all_with_deprecated_constants(globals()) | ||||
|   | ||||
| @@ -2,7 +2,6 @@ | ||||
|  | ||||
| from __future__ import annotations | ||||
|  | ||||
| from collections.abc import Mapping | ||||
| import logging | ||||
| from typing import Any | ||||
|  | ||||
| @@ -10,7 +9,7 @@ from pycync import Auth | ||||
| from pycync.exceptions import AuthFailedError, CyncError, TwoFactorRequiredError | ||||
| import voluptuous as vol | ||||
|  | ||||
| from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlow, ConfigFlowResult | ||||
| from homeassistant.config_entries import ConfigFlow, ConfigFlowResult | ||||
| from homeassistant.const import CONF_ACCESS_TOKEN, CONF_EMAIL, CONF_PASSWORD | ||||
| from homeassistant.helpers.aiohttp_client import async_get_clientsession | ||||
|  | ||||
| @@ -40,7 +39,7 @@ class CyncConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|  | ||||
|     VERSION = 1 | ||||
|  | ||||
|     cync_auth: Auth = None | ||||
|     cync_auth: Auth | ||||
|  | ||||
|     async def async_step_user( | ||||
|         self, user_input: dict[str, Any] | None = None | ||||
| @@ -48,13 +47,28 @@ class CyncConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|         """Attempt login with user credentials.""" | ||||
|         errors: dict[str, str] = {} | ||||
|  | ||||
|         if user_input: | ||||
|         if user_input is None: | ||||
|             return self.async_show_form( | ||||
|                 step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors | ||||
|             ) | ||||
|  | ||||
|         self.cync_auth = Auth( | ||||
|             async_get_clientsession(self.hass), | ||||
|             username=user_input[CONF_EMAIL], | ||||
|             password=user_input[CONF_PASSWORD], | ||||
|         ) | ||||
|         try: | ||||
|                 errors = await self._validate_credentials(user_input) | ||||
|             await self.cync_auth.login() | ||||
|         except AuthFailedError: | ||||
|             errors["base"] = "invalid_auth" | ||||
|         except TwoFactorRequiredError: | ||||
|             return await self.async_step_two_factor() | ||||
|  | ||||
|             if not errors: | ||||
|         except CyncError: | ||||
|             errors["base"] = "cannot_connect" | ||||
|         except Exception: | ||||
|             _LOGGER.exception("Unexpected exception") | ||||
|             errors["base"] = "unknown" | ||||
|         else: | ||||
|             return await self._create_config_entry(self.cync_auth.username) | ||||
|  | ||||
|         return self.async_show_form( | ||||
| @@ -67,65 +81,12 @@ class CyncConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|         """Attempt login with the two factor auth code sent to the user.""" | ||||
|         errors: dict[str, str] = {} | ||||
|  | ||||
|         if user_input: | ||||
|             errors = await self._validate_credentials(user_input) | ||||
|  | ||||
|             if not errors: | ||||
|                 return await self._create_config_entry(self.cync_auth.username) | ||||
|  | ||||
|             return self.async_show_form( | ||||
|                 step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors | ||||
|             ) | ||||
|  | ||||
|         if user_input is None: | ||||
|             return self.async_show_form( | ||||
|                 step_id="two_factor", data_schema=STEP_TWO_FACTOR_SCHEMA, errors=errors | ||||
|             ) | ||||
|  | ||||
|     async def async_step_reauth( | ||||
|         self, entry_data: Mapping[str, Any] | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Perform reauth upon an API authentication error.""" | ||||
|         return await self.async_step_reauth_confirm() | ||||
|  | ||||
|     async def async_step_reauth_confirm( | ||||
|         self, user_input: dict[str, Any] | None = None | ||||
|     ) -> ConfigFlowResult: | ||||
|         """Dialog that informs the user that reauth is required and prompts for their Cync credentials.""" | ||||
|         errors: dict[str, str] = {} | ||||
|  | ||||
|         reauth_entry = self._get_reauth_entry() | ||||
|  | ||||
|         if user_input: | ||||
|         try: | ||||
|                 errors = await self._validate_credentials(user_input) | ||||
|             except TwoFactorRequiredError: | ||||
|                 return await self.async_step_two_factor() | ||||
|  | ||||
|             if not errors: | ||||
|                 return await self._create_config_entry(self.cync_auth.username) | ||||
|  | ||||
|         return self.async_show_form( | ||||
|             step_id="reauth_confirm", | ||||
|             data_schema=STEP_USER_DATA_SCHEMA, | ||||
|             errors=errors, | ||||
|             description_placeholders={CONF_EMAIL: reauth_entry.title}, | ||||
|         ) | ||||
|  | ||||
|     async def _validate_credentials(self, user_input: dict[str, Any]) -> dict[str, str]: | ||||
|         """Attempt to log in with user email and password, and return the error dict.""" | ||||
|         errors: dict[str, str] = {} | ||||
|  | ||||
|         if not self.cync_auth: | ||||
|             self.cync_auth = Auth( | ||||
|                 async_get_clientsession(self.hass), | ||||
|                 username=user_input[CONF_EMAIL], | ||||
|                 password=user_input[CONF_PASSWORD], | ||||
|             ) | ||||
|  | ||||
|         try: | ||||
|             await self.cync_auth.login(user_input.get(CONF_TWO_FACTOR_CODE)) | ||||
|         except TwoFactorRequiredError: | ||||
|             raise | ||||
|             await self.cync_auth.login(user_input[CONF_TWO_FACTOR_CODE]) | ||||
|         except AuthFailedError: | ||||
|             errors["base"] = "invalid_auth" | ||||
|         except CyncError: | ||||
| @@ -133,29 +94,25 @@ class CyncConfigFlow(ConfigFlow, domain=DOMAIN): | ||||
|         except Exception: | ||||
|             _LOGGER.exception("Unexpected exception") | ||||
|             errors["base"] = "unknown" | ||||
|         else: | ||||
|             return await self._create_config_entry(self.cync_auth.username) | ||||
|  | ||||
|         return errors | ||||
|         return self.async_show_form( | ||||
|             step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors | ||||
|         ) | ||||
|  | ||||
|     async def _create_config_entry(self, user_email: str) -> ConfigFlowResult: | ||||
|         """Create the Cync config entry using input user data.""" | ||||
|  | ||||
|         cync_user = self.cync_auth.user | ||||
|         await self.async_set_unique_id(str(cync_user.user_id)) | ||||
|         self._abort_if_unique_id_configured() | ||||
|  | ||||
|         config_data = { | ||||
|         config = { | ||||
|             CONF_USER_ID: cync_user.user_id, | ||||
|             CONF_AUTHORIZE_STRING: cync_user.authorize, | ||||
|             CONF_EXPIRES_AT: cync_user.expires_at, | ||||
|             CONF_ACCESS_TOKEN: cync_user.access_token, | ||||
|             CONF_REFRESH_TOKEN: cync_user.refresh_token, | ||||
|         } | ||||
|  | ||||
|         if self.source == SOURCE_REAUTH: | ||||
|             self._abort_if_unique_id_mismatch() | ||||
|             return self.async_update_reload_and_abort( | ||||
|                 entry=self._get_reauth_entry(), title=user_email, data=config_data | ||||
|             ) | ||||
|  | ||||
|         self._abort_if_unique_id_configured() | ||||
|  | ||||
|         return self.async_create_entry(title=user_email, data=config_data) | ||||
|         return self.async_create_entry(title=user_email, data=config) | ||||
|   | ||||
| @@ -37,7 +37,7 @@ rules: | ||||
|   integration-owner: done | ||||
|   log-when-unavailable: todo | ||||
|   parallel-updates: todo | ||||
|   reauthentication-flow: done | ||||
|   reauthentication-flow: todo | ||||
|   test-coverage: todo | ||||
|  | ||||
|   # Gold | ||||
|   | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user