jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 21:57:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use EntityFeature enum in components (n**) (#69432)

Browse Source
This commit is contained in:
epenet 2022-04-07 09:34:29 +02:00 committed by GitHub
parent 2cc3bfa056
commit 96819fff74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 49 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
homeassistant/components/nad/media_player.py
View File

@ -4,14 +4,10 @@ from __future__ import annotations
from nad_receiver import NADReceiver, NADReceiverTCP, NADReceiverTelnet from nad_receiver import NADReceiver, NADReceiverTCP, NADReceiverTelnet
import voluptuous as vol import voluptuous as vol
from homeassistant.components.media_player import PLATFORM_SCHEMA, MediaPlayerEntity from homeassistant.components.media_player import (
from homeassistant.components.media_player.const import ( PLATFORM_SCHEMA,
SUPPORT_SELECT_SOURCE, MediaPlayerEntity,
SUPPORT_TURN_OFF, MediaPlayerEntityFeature,
SUPPORT_TURN_ON,
SUPPORT_VOLUME_MUTE,
SUPPORT_VOLUME_SET,
SUPPORT_VOLUME_STEP,
) )
from homeassistant.const import ( from homeassistant.const import (
CONF_HOST, CONF_HOST,
@ -35,12 +31,12 @@ DEFAULT_MAX_VOLUME = -20
DEFAULT_VOLUME_STEP = 4 DEFAULT_VOLUME_STEP = 4
SUPPORT_NAD = ( SUPPORT_NAD = (
SUPPORT_VOLUME_SET MediaPlayerEntityFeature.VOLUME_SET
| SUPPORT_VOLUME_MUTE | MediaPlayerEntityFeature.VOLUME_MUTE
| SUPPORT_TURN_ON | MediaPlayerEntityFeature.TURN_ON
| SUPPORT_TURN_OFF | MediaPlayerEntityFeature.TURN_OFF
| SUPPORT_VOLUME_STEP | MediaPlayerEntityFeature.VOLUME_STEP
| SUPPORT_SELECT_SOURCE | MediaPlayerEntityFeature.SELECT_SOURCE
) )
CONF_SERIAL_PORT = "serial_port" # for NADReceiver CONF_SERIAL_PORT = "serial_port" # for NADReceiver
@ -90,6 +86,8 @@ def setup_platform(
class NAD(MediaPlayerEntity): class NAD(MediaPlayerEntity):
"""Representation of a NAD Receiver.""" """Representation of a NAD Receiver."""
_attr_supported_features = SUPPORT_NAD
def __init__(self, config): def __init__(self, config):
"""Initialize the NAD Receiver device.""" """Initialize the NAD Receiver device."""
self.config = config self.config = config
@ -134,11 +132,6 @@ class NAD(MediaPlayerEntity):
"""Boolean if volume is currently muted.""" """Boolean if volume is currently muted."""
return self._mute return self._mute
@property
def supported_features(self):
"""Flag media player features that are supported."""
return SUPPORT_NAD
def turn_off(self): def turn_off(self):
"""Turn the media player off.""" """Turn the media player off."""
self._nad_receiver.main_power("=", "Off") self._nad_receiver.main_power("=", "Off")
@ -227,6 +220,8 @@ class NAD(MediaPlayerEntity):
class NADtcp(MediaPlayerEntity): class NADtcp(MediaPlayerEntity):
"""Representation of a NAD Digital amplifier.""" """Representation of a NAD Digital amplifier."""
_attr_supported_features = SUPPORT_NAD
def __init__(self, config): def __init__(self, config):
"""Initialize the amplifier.""" """Initialize the amplifier."""
self._name = config[CONF_NAME] self._name = config[CONF_NAME]
@ -261,11 +256,6 @@ class NADtcp(MediaPlayerEntity):
"""Boolean if volume is currently muted.""" """Boolean if volume is currently muted."""
return self._mute return self._mute
@property
def supported_features(self):
"""Flag media player features that are supported."""
return SUPPORT_NAD
def turn_off(self): def turn_off(self):
"""Turn the media player off.""" """Turn the media player off."""
self._nad_receiver.power_off() self._nad_receiver.power_off()

17
homeassistant/components/ness_alarm/alarm_control_panel.py
View File

@ -6,11 +6,7 @@ import logging
from nessclient import ArmingState from nessclient import ArmingState
import homeassistant.components.alarm_control_panel as alarm import homeassistant.components.alarm_control_panel as alarm
from homeassistant.components.alarm_control_panel.const import ( from homeassistant.components.alarm_control_panel import AlarmControlPanelEntityFeature
SUPPORT_ALARM_ARM_AWAY,
SUPPORT_ALARM_ARM_HOME,
SUPPORT_ALARM_TRIGGER,
)
from homeassistant.const import ( from homeassistant.const import (
STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_AWAY,
STATE_ALARM_ARMING, STATE_ALARM_ARMING,
@ -45,6 +41,12 @@ async def async_setup_platform(
class NessAlarmPanel(alarm.AlarmControlPanelEntity): class NessAlarmPanel(alarm.AlarmControlPanelEntity):
"""Representation of a Ness alarm panel.""" """Representation of a Ness alarm panel."""
_attr_supported_features = (
AlarmControlPanelEntityFeature.ARM_HOME
| AlarmControlPanelEntityFeature.ARM_AWAY
| AlarmControlPanelEntityFeature.TRIGGER
)
def __init__(self, client, name): def __init__(self, client, name):
"""Initialize the alarm panel.""" """Initialize the alarm panel."""
self._client = client self._client = client
@ -79,11 +81,6 @@ class NessAlarmPanel(alarm.AlarmControlPanelEntity):
"""Return the state of the device.""" """Return the state of the device."""
return self._state return self._state
@property
def supported_features(self) -> int:
"""Return the list of supported features."""
return SUPPORT_ALARM_ARM_HOME | SUPPORT_ALARM_ARM_AWAY | SUPPORT_ALARM_TRIGGER
async def async_alarm_disarm(self, code=None): async def async_alarm_disarm(self, code=None):
"""Send disarm command.""" """Send disarm command."""
await self._client.disarm(code) await self._client.disarm(code)

9
homeassistant/components/netatmo/camera.py
View File

@ -8,7 +8,7 @@ import aiohttp
import pyatmo import pyatmo
import voluptuous as vol import voluptuous as vol
from homeassistant.components.camera import SUPPORT_STREAM, Camera from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError, PlatformNotReady from homeassistant.exceptions import HomeAssistantError, PlatformNotReady
@ -112,6 +112,8 @@ async def async_setup_entry(
class NetatmoCamera(NetatmoBase, Camera): class NetatmoCamera(NetatmoBase, Camera):
"""Representation of a Netatmo camera.""" """Representation of a Netatmo camera."""
_attr_supported_features = CameraEntityFeature.STREAM
def __init__( def __init__(
self, self,
data_handler: NetatmoDataHandler, data_handler: NetatmoDataHandler,
@ -217,11 +219,6 @@ class NetatmoCamera(NetatmoBase, Camera):
"""Return True if entity is available.""" """Return True if entity is available."""
return bool(self._alim_status == "on" or self._status == "disconnected") return bool(self._alim_status == "on" or self._status == "disconnected")
@property
def supported_features(self) -> int:
"""Return supported features."""
return SUPPORT_STREAM
@property @property
def brand(self) -> str: def brand(self) -> str:
"""Return the camera brand.""" """Return the camera brand."""

10
homeassistant/components/netatmo/climate.py
View File

@ -7,7 +7,7 @@ from typing import Any
import pyatmo import pyatmo
import voluptuous as vol import voluptuous as vol
from homeassistant.components.climate import ClimateEntity from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
CURRENT_HVAC_HEAT, CURRENT_HVAC_HEAT,
CURRENT_HVAC_IDLE, CURRENT_HVAC_IDLE,
@ -17,8 +17,6 @@ from homeassistant.components.climate.const import (
HVAC_MODE_OFF, HVAC_MODE_OFF,
PRESET_AWAY, PRESET_AWAY,
PRESET_BOOST, PRESET_BOOST,
SUPPORT_PRESET_MODE,
SUPPORT_TARGET_TEMPERATURE,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
@ -69,8 +67,6 @@ PRESET_FROST_GUARD = "Frost Guard"
PRESET_SCHEDULE = "Schedule" PRESET_SCHEDULE = "Schedule"
PRESET_MANUAL = "Manual" PRESET_MANUAL = "Manual"
SUPPORT_FLAGS = SUPPORT_TARGET_TEMPERATURE | SUPPORT_PRESET_MODE
SUPPORT_HVAC = [HVAC_MODE_HEAT, HVAC_MODE_AUTO, HVAC_MODE_OFF]
SUPPORT_PRESET = [PRESET_AWAY, PRESET_BOOST, PRESET_FROST_GUARD, PRESET_SCHEDULE] SUPPORT_PRESET = [PRESET_AWAY, PRESET_BOOST, PRESET_FROST_GUARD, PRESET_SCHEDULE]
STATE_NETATMO_SCHEDULE = "schedule" STATE_NETATMO_SCHEDULE = "schedule"
@ -175,7 +171,9 @@ class NetatmoThermostat(NetatmoBase, ClimateEntity):
_attr_hvac_mode = HVAC_MODE_AUTO _attr_hvac_mode = HVAC_MODE_AUTO
_attr_max_temp = DEFAULT_MAX_TEMP _attr_max_temp = DEFAULT_MAX_TEMP
_attr_preset_modes = SUPPORT_PRESET _attr_preset_modes = SUPPORT_PRESET
_attr_supported_features = SUPPORT_FLAGS _attr_supported_features = (
ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.PRESET_MODE
)
_attr_target_temperature_step = PRECISION_HALVES _attr_target_temperature_step = PRECISION_HALVES
_attr_temperature_unit = TEMP_CELSIUS _attr_temperature_unit = TEMP_CELSIUS

20
homeassistant/components/nexia/climate.py
View File

@ -18,7 +18,7 @@ from nexia.util import find_humidity_setpoint
from nexia.zone import NexiaThermostatZone from nexia.zone import NexiaThermostatZone
import voluptuous as vol import voluptuous as vol
from homeassistant.components.climate import ClimateEntity from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
ATTR_HUMIDITY, ATTR_HUMIDITY,
ATTR_HVAC_MODE, ATTR_HVAC_MODE,
@ -33,12 +33,6 @@ from homeassistant.components.climate.const import (
HVAC_MODE_HEAT, HVAC_MODE_HEAT,
HVAC_MODE_HEAT_COOL, HVAC_MODE_HEAT_COOL,
HVAC_MODE_OFF, HVAC_MODE_OFF,
SUPPORT_AUX_HEAT,
SUPPORT_FAN_MODE,
SUPPORT_PRESET_MODE,
SUPPORT_TARGET_HUMIDITY,
SUPPORT_TARGET_TEMPERATURE,
SUPPORT_TARGET_TEMPERATURE_RANGE,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
@ -116,10 +110,10 @@ HVAC_MODES = [
] ]
NEXIA_SUPPORTED = ( NEXIA_SUPPORTED = (
SUPPORT_TARGET_TEMPERATURE_RANGE ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
| SUPPORT_TARGET_TEMPERATURE | ClimateEntityFeature.TARGET_TEMPERATURE
| SUPPORT_FAN_MODE | ClimateEntityFeature.FAN_MODE
| SUPPORT_PRESET_MODE | ClimateEntityFeature.PRESET_MODE
) )
@ -177,9 +171,9 @@ class NexiaZone(NexiaThermostatZoneEntity, ClimateEntity):
self._has_dehumidify_support = self._thermostat.has_dehumidify_support() self._has_dehumidify_support = self._thermostat.has_dehumidify_support()
supported = NEXIA_SUPPORTED supported = NEXIA_SUPPORTED
if self._has_humidify_support or self._has_dehumidify_support: if self._has_humidify_support or self._has_dehumidify_support:
supported |= SUPPORT_TARGET_HUMIDITY supported |= ClimateEntityFeature.TARGET_HUMIDITY
if self._has_emergency_heat: if self._has_emergency_heat:
supported |= SUPPORT_AUX_HEAT supported |= ClimateEntityFeature.AUX_HEAT
self._attr_supported_features = supported self._attr_supported_features = supported
self._attr_preset_modes = self._zone.get_presets() self._attr_preset_modes = self._zone.get_presets()
self._attr_fan_modes = self._thermostat.get_fan_modes() self._attr_fan_modes = self._thermostat.get_fan_modes()

15
homeassistant/components/nuheat/climate.py
View File

@ -11,15 +11,13 @@ from nuheat.util import (
nuheat_to_fahrenheit, nuheat_to_fahrenheit,
) )
from homeassistant.components.climate import ClimateEntity from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
ATTR_HVAC_MODE, ATTR_HVAC_MODE,
CURRENT_HVAC_HEAT, CURRENT_HVAC_HEAT,
CURRENT_HVAC_IDLE, CURRENT_HVAC_IDLE,
HVAC_MODE_AUTO, HVAC_MODE_AUTO,
HVAC_MODE_HEAT, HVAC_MODE_HEAT,
SUPPORT_PRESET_MODE,
SUPPORT_TARGET_TEMPERATURE,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
@ -63,8 +61,6 @@ SCHEDULE_MODE_TO_PRESET_MODE_MAP = {
value: key for key, value in PRESET_MODE_TO_SCHEDULE_MODE_MAP.items() value: key for key, value in PRESET_MODE_TO_SCHEDULE_MODE_MAP.items()
} }
SUPPORT_FLAGS = SUPPORT_TARGET_TEMPERATURE | SUPPORT_PRESET_MODE
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
@ -86,6 +82,10 @@ async def async_setup_entry(
class NuHeatThermostat(CoordinatorEntity, ClimateEntity): class NuHeatThermostat(CoordinatorEntity, ClimateEntity):
"""Representation of a NuHeat Thermostat.""" """Representation of a NuHeat Thermostat."""
_attr_supported_features = (
ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.PRESET_MODE
)
def __init__(self, coordinator, thermostat, temperature_unit): def __init__(self, coordinator, thermostat, temperature_unit):
"""Initialize the thermostat.""" """Initialize the thermostat."""
super().__init__(coordinator) super().__init__(coordinator)
@ -99,11 +99,6 @@ class NuHeatThermostat(CoordinatorEntity, ClimateEntity):
"""Return the name of the thermostat.""" """Return the name of the thermostat."""
return self._thermostat.room return self._thermostat.room
@property
def supported_features(self):
"""Return the list of supported features."""
return SUPPORT_FLAGS
@property @property
def temperature_unit(self): def temperature_unit(self):
"""Return the unit of measurement.""" """Return the unit of measurement."""

9
homeassistant/components/nuki/lock.py
View File

@ -4,7 +4,7 @@ from abc import ABC, abstractmethod
from pynuki.constants import MODE_OPENER_CONTINUOUS from pynuki.constants import MODE_OPENER_CONTINUOUS
import voluptuous as vol import voluptuous as vol
from homeassistant.components.lock import SUPPORT_OPEN, LockEntity from homeassistant.components.lock import LockEntity, LockEntityFeature
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform from homeassistant.helpers import config_validation as cv, entity_platform
@ -60,6 +60,8 @@ async def async_setup_entry(
class NukiDeviceEntity(NukiEntity, LockEntity, ABC): class NukiDeviceEntity(NukiEntity, LockEntity, ABC):
"""Representation of a Nuki device.""" """Representation of a Nuki device."""
_attr_supported_features = LockEntityFeature.OPEN
@property @property
def name(self): def name(self):
"""Return the name of the lock.""" """Return the name of the lock."""
@ -84,11 +86,6 @@ class NukiDeviceEntity(NukiEntity, LockEntity, ABC):
} }
return data return data
@property
def supported_features(self):
"""Flag supported features."""
return SUPPORT_OPEN
@property @property
def available(self) -> bool: def available(self) -> bool:
"""Return True if entity is available.""" """Return True if entity is available."""

15
homeassistant/components/nx584/alarm_control_panel.py
View File

@ -11,10 +11,7 @@ import voluptuous as vol
import homeassistant.components.alarm_control_panel as alarm import homeassistant.components.alarm_control_panel as alarm
from homeassistant.components.alarm_control_panel import ( from homeassistant.components.alarm_control_panel import (
PLATFORM_SCHEMA as PARENT_PLATFORM_SCHEMA, PLATFORM_SCHEMA as PARENT_PLATFORM_SCHEMA,
) AlarmControlPanelEntityFeature,
from homeassistant.components.alarm_control_panel.const import (
SUPPORT_ALARM_ARM_AWAY,
SUPPORT_ALARM_ARM_HOME,
) )
from homeassistant.const import ( from homeassistant.const import (
CONF_HOST, CONF_HOST,
@ -95,6 +92,11 @@ async def async_setup_platform(
class NX584Alarm(alarm.AlarmControlPanelEntity): class NX584Alarm(alarm.AlarmControlPanelEntity):
"""Representation of a NX584-based alarm panel.""" """Representation of a NX584-based alarm panel."""
_attr_supported_features = (
AlarmControlPanelEntityFeature.ARM_HOME
| AlarmControlPanelEntityFeature.ARM_AWAY
)
def __init__(self, name, alarm_client, url): def __init__(self, name, alarm_client, url):
"""Init the nx584 alarm panel.""" """Init the nx584 alarm panel."""
self._name = name self._name = name
@ -117,11 +119,6 @@ class NX584Alarm(alarm.AlarmControlPanelEntity):
"""Return the state of the device.""" """Return the state of the device."""
return self._state return self._state
@property
def supported_features(self) -> int:
"""Return the list of supported features."""
return SUPPORT_ALARM_ARM_HOME | SUPPORT_ALARM_ARM_AWAY
def update(self): def update(self):
"""Process new events from panel.""" """Process new events from panel."""
try: try: